use sys instead of platform.

tableau · Nov 21, 2024 · 0b2ed8e · 0b2ed8e
1 parent 5231e4f
commit 0b2ed8e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tabpy/tabpy_tools/custom_query_object.py b/tabpy/tabpy_tools/custom_query_object.py
@@ -1,5 +1,5 @@
 import logging
-import platform
+import sys
 from .query_object import QueryObject as _QueryObject
 
 
@@ -75,11 +75,11 @@ def get_doc_string(self):
         default_docstring = "-- no docstring found in query function --"
 
         # Skip docstring parsing on Windows 32-bit systems
-        if platform.system() == "Windows" and platform.architecture()[0] == "32bit":
+        if sys.platform == 'win32':
             return default_docstring
 
-        docstring = getattr(self.custom_query, '__doc__', None)
-        return docstring.strip() if isinstance(docstring, str) and docstring else default_docstring
+        ds = getattr(self.custom_query, '__doc__', None)
+        return ds if ds and isinstance(ds, str) else default_docstring
 
     def get_methods(self):
         return [self.get_query_method()]