omnata-plugin-runtime 0.10.30a292__tar.gz → 0.10.31a295__tar.gz

{omnata_plugin_runtime-0.10.30a292 → omnata_plugin_runtime-0.10.31a295}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-runtime
-Version: 0.10.30a292
+Version: 0.10.31a295
 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.30a292 → omnata_plugin_runtime-0.10.31a295}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.10.30-a292"
+version = "0.10.31-a295"
 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.30a292 → omnata_plugin_runtime-0.10.31a295}/src/omnata_plugin_runtime/json_schema.py

@@ -191,11 +191,17 @@ class SnowflakeViewColumn(BaseModel):
             self.referenced_columns
         )
 
-    def definition(self,original_name:bool = False) -> str:
+    def definition(self,original_name:bool = False, remove_stream_prefix:Optional[str] = None) -> str:
         """
         Returns the column definition for a normalized view.
         If original_name is True, the original name will be used instead of the transformed name.
         """
+        if remove_stream_prefix is not None and self.referenced_columns is not None:
+            if remove_stream_prefix in self.referenced_columns:
+                for referenced_column in self.referenced_columns[remove_stream_prefix]:
+                    replace_source = f'"{remove_stream_prefix}"."{referenced_column}"'
+                    replace_target = f'"{referenced_column}"'
+                    self.expression = self.expression.replace(replace_source, replace_target)
         if original_name:
             return f'{self.expression} as "{self.original_name}"'
         return f'{self.expression} as "{self.name}"'
@@ -456,11 +462,11 @@ class SnowflakeViewPart(BaseModel):
         """
         if include_only_columns is None:
             return f""" "{self.stream_name}" as (
-    select {', '.join([c.definition(original_name=original_name) for c in self.direct_columns()])} 
+    select {', '.join([c.definition(original_name=original_name,remove_stream_prefix=self.stream_name) for c in self.direct_columns()])} 
     from {self.raw_table_location.get_fully_qualified_name()}
 ) """
         return f""" "{self.stream_name}" as (
-    select {', '.join([c.definition(original_name=original_name) for c in self.columns
+    select {', '.join([c.definition(original_name=original_name,remove_stream_prefix=self.stream_name) for c in self.columns
                        if c.original_name in include_only_columns])} 
     from {self.raw_table_location.get_fully_qualified_name()}
 ) """