omnata-plugin-runtime 0.5.9a161__py3-none-any.whl → 0.5.10__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -112,6 +112,90 @@ class PluginManifest(SubscriptableBaseModel):
     supported_outbound_strategies: List[OutboundSyncStrategy]
 
 
+class SnowflakeFunctionParameter(BaseModel):
+    """
+    Represents a parameter for a Snowflake UDF or UDTF
+    """
+    name: str
+    description: str
+    data_type: str
+    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}"
+
+class SnowflakeUDTFResultColumn(BaseModel):
+    """
+    Represents a result column for a Snowflake UDTF
+    """
+    name: str
+    data_type: str
+    def __str__(self):
+        return f"{self.name} {self.data_type}"
+
+class UDTFDefinition(BaseModel):
+    """
+    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 omnata_udtf decorator.
+    """
+    name: str = Field(..., title="The name of the UDTF")
+    language: Literal['python','java'] = Field(..., title="The language of the UDF")
+    runtime_version: str = Field(..., title="The runtime version of the UDF (language dependent)")
+    description: str = Field(..., title="A description of the UDTF")
+    params: List[SnowflakeFunctionParameter] = Field(..., title="The parameters of the UDTF")
+    result_columns: List[SnowflakeUDTFResultColumn] = Field(..., title="The result columns of the UDTF")
+    handler: str = Field(..., title="The handler class/function for the UDTF")
+    expose_to_consumer: bool = Field(..., title="Whether the UDTF should be exposed to consumers")
+    imports: Optional[List[str]] = Field(None, title="A list of imports required by the UDF")
+    packages: Optional[List[str]] = Field(None, title="A list of packages required by the UDTF")
+
+    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 {self.language.upper()}
+RUNTIME_VERSION={self.runtime_version}
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ({imports_str})
+HANDLER='{self.handler}';
+"""
+
+class UDFDefinition(BaseModel):
+    """
+    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 omnata_udf decorator.
+    """
+    name: str = Field(..., title="The name of the UDF")
+    language: Literal['python','java'] = Field(..., title="The language of the UDF")
+    runtime_version: str = Field(..., title="The runtime version of the UDF (language dependent)")
+    description: str = Field(..., title="A description of the UDF")
+    params: List[SnowflakeFunctionParameter] = Field(..., title="The parameters of the UDF")
+    result_data_type: str = Field(..., title="The data type returned by the UDF")
+    handler: str = Field(..., title="The handler class/function for the UDF")
+    expose_to_consumer: bool = Field(..., title="Whether the UDF should be exposed to consumers")
+    imports: Optional[List[str]] = Field(None, title="A list of imports required by the UDF")
+    packages: Optional[List[str]] = Field(None, title="A list of packages required by the UDF")
+
+    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 {self.language.upper()}
+RUNTIME_VERSION={self.runtime_version}
+COMMENT = $${self.description}$$
+PACKAGES = ({packages_str})
+IMPORTS = ({imports_str})
+HANDLER='{self.handler}';
+"""
+
 class PluginInfo(BaseModel):
     """
     Manifest plus other derived information about a plugin which is determined during upload.
@@ -2234,90 +2318,6 @@ def get_nested_value(nested_dict:Dict, keys:List[str]):
     return reduce(lambda d, key: d.get(key) if isinstance(d, dict) else None, keys, nested_dict)
 
 
-class SnowflakeFunctionParameter(BaseModel):
-    """
-    Represents a parameter for a Snowflake UDF or UDTF
-    """
-    name: str
-    description: str
-    data_type: str
-    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}"
-
-class SnowflakeUDTFResultColumn(BaseModel):
-    """
-    Represents a result column for a Snowflake UDTF
-    """
-    name: str
-    data_type: str
-    def __str__(self):
-        return f"{self.name} {self.data_type}"
-
-class UDTFDefinition(BaseModel):
-    """
-    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 omnata_udtf decorator.
-    """
-    name: str = Field(..., title="The name of the UDTF")
-    language: Literal['python','java'] = Field(..., title="The language of the UDF")
-    runtime_version: str = Field(..., title="The runtime version of the UDF (language dependent)")
-    description: str = Field(..., title="A description of the UDTF")
-    params: List[SnowflakeFunctionParameter] = Field(..., title="The parameters of the UDTF")
-    result_columns: List[SnowflakeUDTFResultColumn] = Field(..., title="The result columns of the UDTF")
-    handler: str = Field(..., title="The handler class/function for the UDTF")
-    expose_to_consumer: bool = Field(..., title="Whether the UDTF should be exposed to consumers")
-    imports: Optional[List[str]] = Field(None, title="A list of imports required by the UDF")
-    packages: Optional[List[str]] = Field(None, title="A list of packages required by the UDTF")
-
-    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 {self.language.upper()}
-RUNTIME_VERSION={self.runtime_version}
-COMMENT = $${self.description}$$
-PACKAGES = ({packages_str})
-IMPORTS = ({imports_str})
-HANDLER='{self.handler}';
-"""
-
-class UDFDefinition(BaseModel):
-    """
-    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 omnata_udf decorator.
-    """
-    name: str = Field(..., title="The name of the UDF")
-    language: Literal['python','java'] = Field(..., title="The language of the UDF")
-    runtime_version: str = Field(..., title="The runtime version of the UDF (language dependent)")
-    description: str = Field(..., title="A description of the UDF")
-    params: List[SnowflakeFunctionParameter] = Field(..., title="The parameters of the UDF")
-    result_data_type: str = Field(..., title="The data type returned by the UDF")
-    handler: str = Field(..., title="The handler class/function for the UDF")
-    expose_to_consumer: bool = Field(..., title="Whether the UDF should be exposed to consumers")
-    imports: Optional[List[str]] = Field(None, title="A list of imports required by the UDF")
-    packages: Optional[List[str]] = Field(None, title="A list of packages required by the UDF")
-
-    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 {self.language.upper()}
-RUNTIME_VERSION={self.runtime_version}
-COMMENT = $${self.description}$$
-PACKAGES = ({packages_str})
-IMPORTS = ({imports_str})
-HANDLER='{self.handler}';
-"""
-
 def omnata_udtf(
         name:str,
         description: str,

{omnata_plugin_runtime-0.5.9a161.dist-info → omnata_plugin_runtime-0.5.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.9a161
+Version: 0.5.10
 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.9a161.dist-info → omnata_plugin_runtime-0.5.10.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=FkJXOxU6n63DTgiGB-rD1vk2nHjr6qoWnb28RxSlE_w,128957
+omnata_plugin_runtime/omnata_plugin.py,sha256=PDPaUHbYuhbWfY3xj0g__EejO9bOGv7xt3weuUDM_bo,128957
 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.9a161.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.9a161.dist-info/METADATA,sha256=lmBdPKUP3HfLI8gv115ZsT5CiNlmTUxU72HGYDstENA,1985
-omnata_plugin_runtime-0.5.9a161.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.9a161.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.10.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.10.dist-info/METADATA,sha256=kNuFZBhkStkg8wsq21vu-qPsc5_cYcWn7E1V_vrODjg,1982
+omnata_plugin_runtime-0.5.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.10.dist-info/RECORD,,