test get_doc_string.

Author: jakeichikawasalesforce

tabpy/tabpy_tools/client.py

@@ -380,10 +380,6 @@ def _gen_endpoint(self, name, obj, description, version=1, schema=None, is_publi
             description = obj.__doc__.strip() or "" if isinstance(obj.__doc__, str) else ""
 
         endpoint_object = CustomQueryObject(query=obj, description=description,)
-        
-        docstring = "-- no docstring found in query function --"
-        if sys.platform != "win32":
-            docstring = inspect.getdoc(obj) or "-- no docstring found in query function --"
 
         return {
             "name": name,
@@ -395,7 +391,7 @@ def _gen_endpoint(self, name, obj, description, version=1, schema=None, is_publi
             "methods": endpoint_object.get_methods(),
             "required_files": [],
             "required_packages": [],
-            "docstring": docstring,
+            "docstring": endpoint_object.get_doc_string(),
             "schema": copy.copy(schema),
             "is_public": is_public,
         }

tabpy/tabpy_tools/custom_query_object.py

@@ -1,4 +1,7 @@
+import inspect
 import logging
+import platform
+import sys
 from .query_object import QueryObject as _QueryObject
 
 
@@ -71,10 +74,13 @@ def query(self, *args, **kwargs):
 
     def get_doc_string(self):
         """Get doc string from customized query"""
-        if self.custom_query.__doc__ is not None:
-            return self.custom_query.__doc__
-        else:
-            return "-- no docstring found in query function --"
+        default_docstring = "-- no docstring found in query function --"
+        return default_docstring
+        # Docstring parsing not working on Windows.
+        # if platform.system() == "Windows":
+        #     return default_docstring
+        # else:
+        #     return inspect.getdoc(self.custom_query) or default_docstring
 
     def get_methods(self):
         return [self.get_query_method()]