omnata-plugin-runtime 0.10.15a269__tar.gz → 0.10.16__tar.gz

{omnata_plugin_runtime-0.10.15a269 → omnata_plugin_runtime-0.10.16}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.10.15-a269"
+version = "0.10.16"
 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.15a269 → omnata_plugin_runtime-0.10.16}/src/omnata_plugin_runtime/json_schema.py

@@ -218,18 +218,19 @@ class SnowflakeViewColumn(BaseModel):
     
     @classmethod
     def from_json_schema_property(cls,
-                                  column_name:str,
-                                  comment:str,
-                                  variant_path:str,
-                                  json_schema_property:JsonSchemaProperty,
-                                  column_name_environment:Environment,
-                                  column_name_expression:str) -> Self:
+                                column_name:str,
+                                comment:str,
+                                variant_path:str,
+                                json_schema_property:JsonSchemaProperty,
+                                column_name_environment:Environment,
+                                column_name_expression:str,
+                                plugin_app_database: Optional[str] = None,) -> Self:
         """
         Takes a JSON schema property (which may be nested via variant_path), along with its final name and comment,
         and returns a SnowflakeViewColumn object which is ready to use in a select statement.
         It does this by applying overarching type conversion rules, and evaluating the final column name using Jinja.
         """
-        jinja_vars = {"column_name": column_name}
+        jinja_vars = {"column_name": column_name,"plugin_app_database": plugin_app_database}
         final_column_name = column_name_environment.from_string(column_name_expression).render(**jinja_vars)
         expression = f"""RECORD_DATA:{variant_path}"""
         if json_schema_property.snowflakeColumnExpression:
@@ -529,7 +530,8 @@ class SnowflakeViewParts(BaseModel):
         stream_name: str,
         include_default_columns: bool = True,
         column_name_environment: Environment = Environment(),
-        column_name_expression: str = "{{column_name}}"
+        column_name_expression: str = "{{column_name}}",
+        plugin_app_database: Optional[str] = None,
     ) -> Self:
         """
         Returns the building blocks required to create a normalized view from a stream.
@@ -796,7 +798,8 @@ class JsonSchemaTopLevel(BaseModel):
 
     def build_view_columns(self,
             column_name_environment: Environment,                
-            column_name_expression: str
+            column_name_expression: str,
+            plugin_app_database: Optional[str] = None,
         ) -> List[SnowflakeViewColumn]:
         """
         Returns a list of column definitions from a json schema
@@ -809,6 +812,7 @@ class JsonSchemaTopLevel(BaseModel):
                 property_value=property_value,
                 column_name_environment=column_name_environment,
                 column_name_expression=column_name_expression,
+                plugin_app_database=plugin_app_database,
             )
             for property_name, property_value in self.properties.items()
         ]
@@ -822,7 +826,8 @@ class JsonSchemaTopLevel(BaseModel):
         column_name_environment: Environment,                
         column_name_expression: str,
         current_field_name_path: List[str] = [],
-        current_comment_path: List[str] = []
+        current_comment_path: List[str] = [],
+        plugin_app_database: Optional[str] = None,
     ) -> List[SnowflakeViewColumn]:
         """
         Recursive function which returns a list of column definitions.
@@ -845,6 +850,7 @@ class JsonSchemaTopLevel(BaseModel):
                             column_name_expression=column_name_expression,
                             current_field_name_path=current_field_name_path + [property_name],
                             current_comment_path=current_comment_path + [json_property.description or ""],
+                            plugin_app_database=plugin_app_database,
                         )
                         for child_property_name, child_property_value in json_property.properties.items()
                     ]
@@ -862,7 +868,8 @@ class JsonSchemaTopLevel(BaseModel):
             variant_path=":".join([f'"{p}"' for p in current_field_name_path if p]),
             json_schema_property=json_property,
             column_name_environment=column_name_environment,
-            column_name_expression=column_name_expression
+            column_name_expression=column_name_expression,
+            plugin_app_database=plugin_app_database,
         )]
         columns.extend(children)
         return columns
@@ -875,6 +882,7 @@ def normalized_view_part(
     column_name_environment: Environment,                
     column_name_expression: str,
     stream_schema: Optional[Dict] = None,
+    plugin_app_database: Optional[str] = None,
 ) -> SnowflakeViewPart:
     """
     Returns an object containing:
@@ -937,7 +945,8 @@ def normalized_view_part(
         json_schema = JsonSchemaTopLevel.model_validate(stream_schema)
         view_columns += json_schema.build_view_columns(
             column_name_environment=column_name_environment,
-            column_name_expression=column_name_expression
+            column_name_expression=column_name_expression,
+            plugin_app_database=plugin_app_database,
         )
         if json_schema.joins:
             joins = json_schema.joins