omnata-plugin-runtime 0.5.7a155__tar.gz → 0.5.8a158__tar.gz

{omnata_plugin_runtime-0.5.7a155 → omnata_plugin_runtime-0.5.8a158}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.7a155
+Version: 0.5.8a158
 Summary: Classes and common runtime components for building and running Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_runtime-0.5.7a155 → omnata_plugin_runtime-0.5.8a158}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.5.7-a155"
+version = "0.5.8-a158"
 description = "Classes and common runtime components for building and running Omnata Plugins"
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_runtime-0.5.7a155 → omnata_plugin_runtime-0.5.8a158}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -2425,7 +2425,10 @@ def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
                     expose_to_consumer=obj._omnata_udtf_expose_to_consumer,
                     handler=obj.__module__+'.'+obj.__name__
                 ))
-            elif isinstance(obj, UDTFDefinition) and cast(UDTFDefinition,obj).language == 'java':
+        # java doesn't use the python decorator, so we need to check for the class directly
+        for name, obj in inspect.getmembers(module, inspect.isclass):
+            # Check if the class has the specified attribute
+            if isinstance(obj, UDTFDefinition) and cast(UDTFDefinition,obj).language == 'java':
                 matching_classes.append(obj)
             
     return matching_classes
@@ -2529,7 +2532,10 @@ def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
                     packages=[],
                     handler=obj.__module__+'.'+obj.__name__
                 ))
-            elif isinstance(obj, UDFDefinition) and cast(UDFDefinition,obj).language == 'java':
+        # java doesn't use the python decorator, so we need to check for the class directly
+        for name, obj in inspect.getmembers(module, inspect.isclass):
+            # Check if the class has the specified attribute
+            if isinstance(obj, UDFDefinition) and cast(UDFDefinition,obj).language == 'java':
                 matching_classes.append(obj)
 
     return matching_classes