test get_doc_string.

tableau · Nov 21, 2024 · 4ed2768 · 4ed2768
1 parent a0ca62b
commit 4ed2768
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tabpy/tabpy_tools/custom_query_object.py b/tabpy/tabpy_tools/custom_query_object.py
@@ -1,7 +1,6 @@
 import inspect
 import logging
 import platform
-import sys
 from .query_object import QueryObject as _QueryObject
 
 
@@ -75,7 +74,8 @@ def query(self, *args, **kwargs):
     def get_doc_string(self):
         """Get doc string from customized query"""
         default_docstring = "-- no docstring found in query function --"
-        return default_docstring
+        obj = self.custom_query
+        return obj.__doc__.strip() or default_docstring if isinstance(obj.__doc__, str) else default_docstring
         # Docstring parsing not working on Windows.
         # if platform.system() == "Windows":
         #     return default_docstring