omnata-plugin-runtime 0.5.7a141__py3-none-any.whl → 0.5.7a145__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2255,6 +2255,117 @@ class SnowflakeUDTFResultColumn:
     data_type: str
     description: str
 
+@dataclass
+class PythonUDTFDefinition:
+    """
+    The information needed by the plugin uploader to put a Python UDTF definition into the setup script.
+    Do not use this class directly in plugins, instead use the consumer_udtf decorator.
+    """
+    name: str
+    description: str
+    params: List[SnowflakeFunctionParameter]
+    result_columns: List[SnowflakeUDTFResultColumn]
+    handler: str
+    packages: Optional[List[str]] = None
+
+    def __str__(self):
+        param_str = ', '.join([str(param) for param in self.params])
+        table_result_columns = ', '.join([f"{col.name} {col.data_type}" for col in self.result_columns])
+        packages_str = ', '.join([f"'{p}'" for p in self.packages])
+        return f"""CREATE OR REPLACE FUNCTION UDFS.{self.name}({param_str})
+RETURNS TABLE({table_result_columns})
+LANGUAGE PYTHON
+RUNTIME_VERSION=3.10
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ('/app.zip')
+HANDLER='{self.handler}';
+"""
+    
+@dataclass
+class JavaUDTFDefinition:
+    """
+    The information needed by the plugin uploader to put a Java UDTF definition into the setup script.
+    These are created directly in python, since we don't want the omnata cli to need to do any Java reflection.
+    Any .jar files placed in the udf_direct_imports directory will be uploaded to the stage and can be
+    added to the imports parameter as (e.g.) '/my_jar.jar'.
+    """
+    name: str
+    description: str
+    params: List[SnowflakeFunctionParameter]
+    result_columns: List[SnowflakeUDTFResultColumn]
+    packages: List[str]
+    imports: List[str]
+    handler: str
+
+    def __str__(self):
+        param_str = ', '.join([str(param) for param in self.params])
+        table_result_columns = ', '.join([f"{col.name} {col.data_type}" for col in self.result_columns])
+        packages_str = ', '.join([f"'{p}'" for p in self.packages])
+        imports_str = ', '.join([f"'{i}'" for i in self.imports])
+        return f"""CREATE OR REPLACE FUNCTION UDFS.{self.name}({param_str})
+RETURNS TABLE({table_result_columns})
+LANGUAGE PYTHON
+RUNTIME_VERSION=3.10
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ({imports_str})
+HANDLER='{self.handler}';
+"""
+
+@dataclass
+class PythonUDFDefinition:
+    """
+    The information needed by the plugin uploader to put a Python UDF definition into the setup script.
+    Do not use this class directly in plugins, instead use the consumer_udf decorator.
+    """
+    name: str
+    description: str
+    params: List[SnowflakeFunctionParameter]
+    result_data_type: str
+    handler: str
+    packages: Optional[List[str]] = None
+
+    def __str__(self):
+        param_str = ', '.join([str(param) for param in self.params])
+        packages_str = ', '.join([f"'{p}'" for p in self.packages])
+        return f"""CREATE OR REPLACE FUNCTION UDFS.{self.name}({param_str})
+RETURNS {self.result_data_type}
+LANGUAGE PYTHON
+RUNTIME_VERSION=3.10
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ('/app.zip')
+HANDLER='{self.handler}';
+"""
+
+@dataclass
+class JavaUDFDefinition:
+    """
+    The information needed by the plugin uploader to put a java UDF definition into the setup script.
+    """
+    name: str
+    description: str
+    params: List[SnowflakeFunctionParameter]
+    result_data_type: str
+    packages: List[str]
+    imports: List[str]
+    handler: str
+
+    def __str__(self):
+        param_str = ', '.join([str(param) for param in self.params])
+        packages_str = ', '.join([f"'{p}'" for p in self.packages])
+        imports_str = ', '.join([f"'{i}'" for i in self.imports])
+        return f"""CREATE OR REPLACE FUNCTION UDFS.{self.name}({param_str})
+RETURNS {self.result_data_type}
+LANGUAGE PYTHON
+RUNTIME_VERSION=3.10
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ({imports_str})
+HANDLER='{self.handler}';
+"""
+
 def consumer_udtf(
         name:str,
         description: str,
@@ -2318,7 +2429,7 @@ def consumer_udtf(
     
     return class_decorator
 
-def find_consumer_udtf_classes(path:str = '.') -> List[type]:
+def find_consumer_udtf_classes(path:str = '.') -> List[PythonUDTFDefinition | JavaUDTFDefinition]:
     """
     Finds all classes in the specified directory which have the 'consumer_udtf' decorator applied
     """
@@ -2337,7 +2448,16 @@ def find_consumer_udtf_classes(path:str = '.') -> List[type]:
         for name, obj in inspect.getmembers(module, inspect.isclass):
             # Check if the class has the specified attribute
             if hasattr(obj, '_is_omnata_consumer_udtf'):
+                matching_classes.append(PythonUDTFDefinition(
+                    name=obj._omnata_consumer_udtf_name,
+                    description=obj._omnata_consumer_udtf_description,
+                    params=obj._omnata_consumer_udtf_params,
+                    result_columns=obj._omnata_consumer_udtf_result_columns,
+                    handler=obj.__module__+'.'+obj.__name__
+                ))
+            elif isinstance(obj, JavaUDTFDefinition):
                 matching_classes.append(obj)
+            
     return matching_classes
 
 
@@ -2394,7 +2514,7 @@ def consumer_udf(
     
     return decorator
 
-def find_consumer_udf_functions(path:str = '.') -> List[function]:
+def find_consumer_udf_functions(path:str = '.') -> List[PythonUDFDefinition | JavaUDFDefinition]:
     """
     Finds all functions in the specified directory which have the 'consumer_udf' decorator applied
     """
@@ -2413,6 +2533,15 @@ def find_consumer_udf_functions(path:str = '.') -> List[function]:
         for name, obj in inspect.getmembers(module, inspect.isfunction):
             # Check if the class has the specified attribute
             if hasattr(obj, '_is_omnata_consumer_udf'):
+                matching_classes.append(PythonUDFDefinition(
+                    name=obj._omnata_consumer_udf_name,
+                    description=obj._omnata_consumer_udf_description,
+                    params=obj._omnata_consumer_udf_params,
+                    result_data_type=obj._omnata_consumer_udf_result_data_type,
+                    packages=[],
+                    handler=obj.__module__+'.'+obj.__name__
+                ))
+            elif isinstance(obj, JavaUDFDefinition):
                 matching_classes.append(obj)
 
     return matching_classes

{omnata_plugin_runtime-0.5.7a141.dist-info → omnata_plugin_runtime-0.5.7a145.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.7a141
+Version: 0.5.7a145
 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.7a141.dist-info → omnata_plugin_runtime-0.5.7a145.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=TRD_t9QBXSGNHqn3u2ZjRhnxz2_-VBHDZUCRO0BvcQo,120165
+omnata_plugin_runtime/omnata_plugin.py,sha256=PVgyiqHCXXqsUIbLnOFe9H2FRZB7o_4FiI-9KHzRsfg,125045
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=PFSLsYEVnWHVvSoOYTtTK2JY6pp6_8_eYP53WqLRiPE,27975
 omnata_plugin_runtime/rate_limiting.py,sha256=DVQ_bc-mVLBkrU1PTns1MWXhHiLpSB5HkWCcdePtJ2A,25611
-omnata_plugin_runtime-0.5.7a141.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.7a141.dist-info/METADATA,sha256=GxlnEUYJq_OY85r6o6WB0tQUGHg1MktSjdIPpBpRg9k,1985
-omnata_plugin_runtime-0.5.7a141.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.7a141.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.7a145.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.7a145.dist-info/METADATA,sha256=CjSBdW5cKSIrQ1iKBJ3krdVJy6RHqzAAD4b2g44nik4,1985
+omnata_plugin_runtime-0.5.7a145.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.7a145.dist-info/RECORD,,