omnata-plugin-runtime 0.10.2__tar.gz → 0.10.3__tar.gz

{omnata_plugin_runtime-0.10.2 → omnata_plugin_runtime-0.10.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-runtime
-Version: 0.10.2
+Version: 0.10.3
 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.10.2 → omnata_plugin_runtime-0.10.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.10.2"
+version = "0.10.3"
 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.10.2 → omnata_plugin_runtime-0.10.3}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -2427,7 +2427,7 @@ def omnata_udtf(
     
     return class_decorator
 
-def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
+def find_udtf_classes(path:str = '.',top_level_modules:Optional[List[str]] = None) -> List[UDTFDefinition]:
     """
     Finds all classes in the specified directory which have the 'omnata_udtf' decorator applied
     """
@@ -2440,6 +2440,9 @@ def find_udtf_classes(path:str = '.') -> List[UDTFDefinition]:
     # Iterate over all modules in the current directory
     for importer, module_name, ispkg in pkgutil.walk_packages(path=[current_dir]):
         # Import the module
+        if top_level_modules is not None:
+            if len([x for x in top_level_modules if module_name.startswith(x)]) == 0:
+                continue
         module = importlib.import_module(module_name)
 
         # Iterate over all members of the module
@@ -2544,7 +2547,7 @@ def omnata_udf(
     
     return decorator
 
-def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
+def find_udf_functions(path:str = '.',top_level_modules:Optional[List[str]] = None) -> List[UDFDefinition]:
     """
     Finds all functions in the specified directory which have the 'omnata_udf' decorator applied
     """
@@ -2557,6 +2560,9 @@ def find_udf_functions(path:str = '.') -> List[UDFDefinition]:
     # Iterate over all modules in the current directory
     for importer, module_name, ispkg in pkgutil.walk_packages(path=[current_dir]):
         # Import the module
+        if top_level_modules is not None:
+            if len([x for x in top_level_modules if module_name.startswith(x)]) == 0:
+                continue
         module = importlib.import_module(module_name)
 
         # Iterate over all members of the module