omnata-plugin-devkit 0.2.32__tar.gz → 0.2.34__tar.gz

{omnata_plugin_devkit-0.2.32 → omnata_plugin_devkit-0.2.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-devkit
-Version: 0.2.32
+Version: 0.2.34
 Summary: 
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_devkit-0.2.32 → omnata_plugin_devkit-0.2.34}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-devkit"
-version = "0.2.32"
+version = "0.2.34"
 description = ""
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_devkit-0.2.32 → omnata_plugin_devkit-0.2.34}/src/omnata_plugin_devkit/development_session.py

@@ -82,6 +82,7 @@ class SyncScenario:
         configuration_parameters: SyncConfigurationParameters,
         sync_request: SyncRequest,
         scenario_records: Optional[pandas.DataFrame],
+        initial_stream_state: Optional[Dict] = None,
     ):
         self.scenario_name = scenario_name
         self.sync_direction = sync_direction
@@ -94,6 +95,7 @@ class SyncScenario:
         )
         self.scenario_records = scenario_records
         self.results = None
+        self.initial_stream_state = initial_stream_state
 
 
 class DevelopmentSession:
@@ -425,6 +427,7 @@ class DevelopmentSession:
             configuration_parameters=configuration_parameters,
             sync_request=inbound_sync_request,
             scenario_records=None,
+            initial_stream_state=inbound_payload.latest_stream_state.copy()
         )
         self.run_id = inbound_payload.run_id
         self.plugin_instance._sync_request = inbound_sync_request
@@ -778,9 +781,9 @@ class DevelopmentSession:
                     recorded_scenario.sync_request
                 )
                 for stream in inbound_sync_request.streams:
-                    if stream.latest_state is not None:
+                    if recorded_scenario.initial_stream_state is not None and stream.stream_name in recorded_scenario.initial_stream_state:
                         f.write(
-                            f"\t\t\t| {stream.stream_name} | {json.dumps(stream.latest_state,separators=(',', ':'),default=str)} | \n"
+                            f"\t\t\t| {stream.stream_name} | {json.dumps(recorded_scenario.initial_stream_state[stream.stream_name],separators=(',', ':'),default=str)} | \n"
                         )
                     else:
                         f.write(f"\t\t\t| {stream.stream_name} | {{}} | \n")

{omnata_plugin_devkit-0.2.32 → omnata_plugin_devkit-0.2.34}/src/omnata_plugin_devkit/jinja_templates/CONFIGURE_APIS.sql.jinja

@@ -155,6 +155,29 @@ try{
             binds:[]
         } ).execute();
     });
+    // if there are any custom Java or Python UDFS that ship with the native app, we'll apply the settings to those too
+    var sqlText = `select FUNCTION_NAME||
+                            '('||array_to_string(REGEXP_EXTRACT_ALL(replace(ARGUMENT_SIGNATURE,', ',','),'\\\\s\\\\w+'),',')||')' as         
+                            FUNCTION_SIGNATURE
+                    from INFORMATION_SCHEMA.FUNCTIONS
+                    where FUNCTION_SCHEMA='UDFS'
+                    and FUNCTION_LANGUAGE in ('JAVA','PYTHON')`;
+    snowflake.log("info", `Executing SQL: ${sqlText}`);
+    var udfResults = snowflake.createStatement( {
+        sqlText: sqlText,
+        binds:[]
+    } ).execute();
+    while (udfResults.next()) {
+        var udfSig = udfResults.getColumnValue(1);
+        var sqlText = `alter function UDFS.${udfSig}
+                    set EXTERNAL_ACCESS_INTEGRATIONS = (${integrationNames.join(',')}),
+                        SECRETS = (${secrets.join(',')})`
+        snowflake.log("info", `Executing SQL: ${sqlText}`);
+        snowflake.createStatement( {
+            sqlText:sqlText,
+            binds:[]
+        } ).execute();
+    }
     // write the latest list of applied external access integrations back to the registration table
     var appliedIntegrationsResults = snowflake.createStatement( {
         sqlText: `update DATA.OMNATA_REGISTRATION