Add docstring updates.

Author: jakeichikawasalesforce

tabpy/tabpy_tools/client.py

@@ -1,4 +1,5 @@
 import copy
+import inspect
 from re import compile
 import time
 import requests
@@ -379,6 +380,7 @@ 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 = inspect.getdoc(obj) or "-- no docstring found in query function --"
 
         return {
             "name": name,
@@ -390,6 +392,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,
             "schema": copy.copy(schema),
             "is_public": is_public,
         }

tabpy/tabpy_tools/rest_client.py

@@ -41,6 +41,7 @@ class Endpoint(RESTObject):
     version = RESTProperty(int)
     description = RESTProperty(str)
     dependencies = RESTProperty(list)
+    docstring = RESTProperty(str)
     methods = RESTProperty(list)
     creation_time = RESTProperty(datetime, from_epoch, to_epoch)
     last_modified_time = RESTProperty(datetime, from_epoch, to_epoch)
@@ -64,6 +65,7 @@ def __eq__(self, other):
             and self.version == other.version
             and self.description == other.description
             and self.dependencies == other.dependencies
+            and self.docstring == other.docstring
             and self.methods == other.methods
             and self.evaluator == other.evaluator
             and self.schema_version == other.schema_version