omnata-plugin-runtime 0.5.7a138__py3-none-any.whl → 0.5.7a140__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2232,22 +2232,36 @@ def get_nested_value(nested_dict:Dict, keys:List[str]):
 
 @dataclass
 class SnowflakeFunctionParameter:
+    """
+    Represents a parameter for a Snowflake UDF or UDTF
+    """
     name: str
     description: str
     data_type: str
-    has_default_value: bool = False
     default_value_clause: Optional[str] = None
 
+    def __str__(self):
+        if self.default_value_clause:
+            return f"{self.name} {self.data_type} default {self.default_value_clause}"
+        return f"{self.name} {self.data_type}"
+
 @dataclass
 class SnowflakeUDTFResultColumn:
+    """
+    Represents a result column for a Snowflake UDTF
+    """
     name: str
     data_type: str
     description: str
 
 def consumer_udtf(
+        name:str,
         description: str,
         params: List[SnowflakeFunctionParameter],
         result_columns: List[SnowflakeUDTFResultColumn]):
+    """
+    A decorator for a class which should be automatically exposed to the consumer as a UDTF
+    """
     def class_decorator(cls):
         # Get the original 'process' method from the class
         original_process = getattr(cls, 'process')
@@ -2295,6 +2309,7 @@ def consumer_udtf(
         # Replace the original 'process' method with the wrapped version
         setattr(cls, 'process', wrapped_process)
         cls._is_omnata_consumer_udtf = True
+        cls._omnata_consumer_udtf_name = name
         cls._omnata_consumer_udtf_description = description
         cls._omnata_consumer_udtf_params = params
         cls._omnata_consumer_udtf_result_columns = result_columns
@@ -2303,6 +2318,9 @@ def consumer_udtf(
     return class_decorator
 
 def find_consumer_udtf_classes(path:str = '.'):
+    """
+    Finds all classes in the specified directory which have the 'consumer_udtf' decorator applied
+    """
     # Get the directory's absolute path
     current_dir = os.path.abspath(path)
 
@@ -2323,9 +2341,13 @@ def find_consumer_udtf_classes(path:str = '.'):
 
 
 def consumer_udf(
+        name: str,
         description: str,
         params: List[SnowflakeFunctionParameter],
         result_data_type: str):
+    """
+    A decorator for a function which should be automatically exposed to the consumer as a UDF
+    """
     def decorator(func):
         sig = signature(func)
         function_params = sig.parameters
@@ -2363,6 +2385,7 @@ def consumer_udf(
             return func(parameters, *args, **kwargs)
         
         wrapper._is_omnata_consumer_udf = True
+        wrapper._omnata_consumer_udf_name = name
         wrapper._omnata_consumer_udf_description = description
         wrapper._omnata_consumer_udf_params = params
         wrapper._omnata_consumer_udf_result_data_type = result_data_type
@@ -2371,6 +2394,9 @@ def consumer_udf(
     return decorator
 
 def find_consumer_udf_functions(path:str = '.'):
+    """
+    Finds all functions in the specified directory which have the 'consumer_udf' decorator applied
+    """
     # Get the current directory's absolute path
     current_dir = os.path.abspath(path)
 

{omnata_plugin_runtime-0.5.7a138.dist-info → omnata_plugin_runtime-0.5.7a140.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.7a138
+Version: 0.5.7a140
 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.7a138.dist-info → omnata_plugin_runtime-0.5.7a140.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=oMQDljO8zgE8k5HlSCZ4hGFRYcuE8QWjYyKttCsZNVs,119224
+omnata_plugin_runtime/omnata_plugin.py,sha256=nPCJ6b2cYFS-Gh_UmLUaJvfjgI8v1-DVZWFYWSuj2j8,120102
 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.7a138.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.7a138.dist-info/METADATA,sha256=CMc5dTHMT90TAlp6ZE6bRibqDN6hRwaxjClZFlYzBDs,1985
-omnata_plugin_runtime-0.5.7a138.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.7a138.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.7a140.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.7a140.dist-info/METADATA,sha256=-ohRghA5Z5Z1hblZCVo0qLLniQhM9DmC0N-NJWlwsXM,1985
+omnata_plugin_runtime-0.5.7a140.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.7a140.dist-info/RECORD,,