omnata-plugin-runtime 0.6.2a165__py3-none-any.whl → 0.6.2a167__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2397,40 +2397,39 @@ def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
     matching_classes = []
 
     # Iterate over all modules in the current directory
-    for root, _, _ in os.walk(current_dir):
-        for _, module_name, _ in pkgutil.iter_modules([root]):
-            # Import the module
-            module = importlib.import_module(module_name)
-
-            # Iterate over all members of the module
-            for name, obj in inspect.getmembers(module, inspect.isclass):
-                # Check if the class has the specified attribute
-                if hasattr(obj, '_is_omnata_udtf'):
-                    matching_classes.append(UDTFDefinition(
-                        name=obj._omnata_udtf_name,
-                        language='python',
-                        runtime_version='3.10',
-                        imports=['/app.zip'],
-                        description=obj._omnata_udtf_description,
-                        params=obj._omnata_udtf_params,
-                        result_columns=obj._omnata_udtf_result_columns,
-                        expose_to_consumer=obj._omnata_udtf_expose_to_consumer,
-                        handler=obj.__module__+'.'+obj.__name__
-                    ))
-            # java doesn't use the python decorator, so we need to check for the class directly
-            for name, obj in inspect.getmembers(module):
-                # Check if the class has the specified attribute
-                if isinstance(obj, UDTFDefinition) and cast(UDTFDefinition,obj).language == 'java':
-                    udtf_obj = cast(UDTFDefinition,obj)
-                    if udtf_obj.language == 'java':
-                        # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
-                        if udtf_obj.expose_to_consumer:
-                            if len(udtf_obj.params) == 0 or udtf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
-                                udtf_obj.params = [SnowflakeFunctionParameter(
-                                                    name='CONNECTION_PARAMETERS',
-                                                    data_type='OBJECT',
-                                                    description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udtf_obj.params
-                        matching_classes.append(udtf_obj)
+    for importer, module_name, ispkg in pkgutil.walk_packages(path=['.']):
+        # Import the module
+        module = importlib.import_module(module_name)
+
+        # Iterate over all members of the module
+        for name, obj in inspect.getmembers(module, inspect.isclass):
+            # Check if the class has the specified attribute
+            if hasattr(obj, '_is_omnata_udtf'):
+                matching_classes.append(UDTFDefinition(
+                    name=obj._omnata_udtf_name,
+                    language='python',
+                    runtime_version='3.10',
+                    imports=['/app.zip'],
+                    description=obj._omnata_udtf_description,
+                    params=obj._omnata_udtf_params,
+                    result_columns=obj._omnata_udtf_result_columns,
+                    expose_to_consumer=obj._omnata_udtf_expose_to_consumer,
+                    handler=obj.__module__+'.'+obj.__name__
+                ))
+        # java doesn't use the python decorator, so we need to check for the class directly
+        for name, obj in inspect.getmembers(module):
+            # Check if the class has the specified attribute
+            if isinstance(obj, UDTFDefinition) and cast(UDTFDefinition,obj).language == 'java':
+                udtf_obj = cast(UDTFDefinition,obj)
+                if udtf_obj.language == 'java':
+                    # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
+                    if udtf_obj.expose_to_consumer:
+                        if len(udtf_obj.params) == 0 or udtf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
+                            udtf_obj.params = [SnowflakeFunctionParameter(
+                                                name='CONNECTION_PARAMETERS',
+                                                data_type='OBJECT',
+                                                description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udtf_obj.params
+                    matching_classes.append(udtf_obj)
             
     return matching_classes
 
@@ -2513,41 +2512,39 @@ def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
     matching_classes = []
 
     # Iterate over all modules in the current directory
-    # Walk through the directory tree
-    for root, _, _ in os.walk(current_dir):
-        for _, module_name, _ in pkgutil.iter_modules([root]):
-            # Import the module
-            module = importlib.import_module(module_name)
-
-            # Iterate over all members of the module
-            for name, obj in inspect.getmembers(module, inspect.isfunction):
-                # Check if the class has the specified attribute
-                if hasattr(obj, '_is_omnata_udf'):
-                    matching_classes.append(UDFDefinition(
-                        name=obj._omnata_udf_name,
-                        language='python',
-                        runtime_version='3.10',
-                        imports=['/app.zip'],
-                        description=obj._omnata_udf_description,
-                        params=obj._omnata_udf_params,
-                        result_data_type=obj._omnata_udf_result_data_type,
-                        expose_to_consumer=obj._omnata_udf_expose_to_consumer,
-                        packages=[],
-                        handler=obj.__module__+'.'+obj.__name__
-                    ))
-            # java doesn't use the python decorator, so we need to check for the class directly
-            for name, obj in inspect.getmembers(module):
-                # Check if the class has the specified attribute
-                if isinstance(obj, UDFDefinition):
-                    udf_obj = cast(UDFDefinition,obj)
-                    if udf_obj.language == 'java':
-                        # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
-                        if udf_obj.expose_to_consumer:
-                            if len(udf_obj.params) == 0 or udf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
-                                udf_obj.params = [SnowflakeFunctionParameter(
-                                                    name='CONNECTION_PARAMETERS',
-                                                    data_type='OBJECT',
-                                                    description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udf_obj.params
-                        matching_classes.append(udf_obj)
+    for importer, module_name, ispkg in pkgutil.walk_packages(path=['.']):
+        # Import the module
+        module = importlib.import_module(module_name)
+
+        # Iterate over all members of the module
+        for name, obj in inspect.getmembers(module, inspect.isfunction):
+            # Check if the class has the specified attribute
+            if hasattr(obj, '_is_omnata_udf'):
+                matching_classes.append(UDFDefinition(
+                    name=obj._omnata_udf_name,
+                    language='python',
+                    runtime_version='3.10',
+                    imports=['/app.zip'],
+                    description=obj._omnata_udf_description,
+                    params=obj._omnata_udf_params,
+                    result_data_type=obj._omnata_udf_result_data_type,
+                    expose_to_consumer=obj._omnata_udf_expose_to_consumer,
+                    packages=[],
+                    handler=obj.__module__+'.'+obj.__name__
+                ))
+        # java doesn't use the python decorator, so we need to check for the class directly
+        for name, obj in inspect.getmembers(module):
+            # Check if the class has the specified attribute
+            if isinstance(obj, UDFDefinition):
+                udf_obj = cast(UDFDefinition,obj)
+                if udf_obj.language == 'java':
+                    # because the decorator wasn't used, we need to check if we need to add the connection_parameters parameter
+                    if udf_obj.expose_to_consumer:
+                        if len(udf_obj.params) == 0 or udf_obj.params[0].name.upper() != 'CONNECTION_PARAMETERS':
+                            udf_obj.params = [SnowflakeFunctionParameter(
+                                                name='CONNECTION_PARAMETERS',
+                                                data_type='OBJECT',
+                                                description='The connection object, obtained from calling PLUGIN.PLUGIN_CONNECTION.')] + udf_obj.params
+                    matching_classes.append(udf_obj)
 
     return matching_classes

{omnata_plugin_runtime-0.6.2a165.dist-info → omnata_plugin_runtime-0.6.2a167.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.6.2a165
+Version: 0.6.2a167
 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.6.2a165.dist-info → omnata_plugin_runtime-0.6.2a167.dist-info}/RECORD

@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=FxzTqri4no8ClkOm7vZADG8aD47jcGBCTTQDEORmOJM,
 omnata_plugin_runtime/configuration.py,sha256=TI6GaVFhewVawBCaYN34GujY57qEP6q2nik4YpSEk5s,38100
 omnata_plugin_runtime/forms.py,sha256=GzSPEwcijsoPCXEO1mHiE8ylvX_KSE5TkhwqkymA2Ss,19755
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
-omnata_plugin_runtime/omnata_plugin.py,sha256=ZxxrPEAnw9K6rEBlteuD99d-QkcfyC9brcqZ49hUFhs,128972
+omnata_plugin_runtime/omnata_plugin.py,sha256=0CF3f14KrerhkLWICIPR7f80x6JyVastGLZJb9AYsgY,128618
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=-mkIpfB_pTl1yDBMmTnioW44RPj-V8dvlhBiU7ekvkQ,27976
 omnata_plugin_runtime/rate_limiting.py,sha256=JukA0l7x7Klqz2b54mR-poP7NRxpUHgWSGp6h0B8u6Q,25612
-omnata_plugin_runtime-0.6.2a165.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.6.2a165.dist-info/METADATA,sha256=A5Z319HosHchAwvovZyTAu5a2gQyHpb9XF_FsPy2wsg,1985
-omnata_plugin_runtime-0.6.2a165.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.6.2a165.dist-info/RECORD,,
+omnata_plugin_runtime-0.6.2a167.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.6.2a167.dist-info/METADATA,sha256=yB2cZA_vrOSJhiQZFSoLDZZWBfrhOrb0Ly7y8QVUg58,1985
+omnata_plugin_runtime-0.6.2a167.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.6.2a167.dist-info/RECORD,,