omnata-plugin-runtime 0.4.0a86__tar.gz → 0.4.0a88__tar.gz

{omnata_plugin_runtime-0.4.0a86 → omnata_plugin_runtime-0.4.0a88}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.4.0a86
+Version: 0.4.0a88
 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.4.0a86 → omnata_plugin_runtime-0.4.0a88}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.4.0-a86"
+version = "0.4.0-a88"
 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.4.0a86 → omnata_plugin_runtime-0.4.0a88}/src/omnata_plugin_runtime/omnata_plugin.py

@@ -350,12 +350,13 @@ class SyncRequest(ABC):
             update_rate_limit_result = self._plugin_message(
                 PluginMessageRateLimitState(rate_limit_state=self.rate_limit_state_this_sync_and_branch)
             )
-            sync_id:int = update_rate_limit_result["sync_id"]
-            sync_branch_name:str = update_rate_limit_result["sync_branch_name"]
-            latest_state:Dict[int,Dict[str,Dict[str,RateLimitState]]] = parse_obj_as(Dict[int,Dict[str,Dict[str,RateLimitState]]],update_rate_limit_result["latest_state"])
-            (rate_limit_state_all, rate_limit_state_this_branch) = RateLimitState.collapse(latest_state,sync_id, sync_branch_name)
-            self.rate_limit_state_all = rate_limit_state_all
-            self.rate_limit_state_this_sync_and_branch = rate_limit_state_this_branch
+            if update_rate_limit_result is not None:
+                sync_id:int = update_rate_limit_result["sync_id"]
+                sync_branch_name:str = update_rate_limit_result["sync_branch_name"]
+                latest_state:Dict[int,Dict[str,Dict[str,RateLimitState]]] = parse_obj_as(Dict[int,Dict[str,Dict[str,RateLimitState]]],update_rate_limit_result["latest_state"])
+                (rate_limit_state_all, rate_limit_state_this_branch) = RateLimitState.collapse(latest_state,sync_id, sync_branch_name)
+                self.rate_limit_state_all = rate_limit_state_all
+                self.rate_limit_state_this_sync_and_branch = rate_limit_state_this_branch
         else:
             logger.info("No rate limit state to update")
 
@@ -934,7 +935,6 @@ class InboundSyncRequest(SyncRequest):
                     ] = self._stream_record_counts[stream_name] + total_length
                     results.extend(non_empty_dfs)  # remove any None/empty dataframes
                     stream_names.append(stream_name)
-                self._session.sql("begin").collect()
                 if len(results) > 0:
                     logger.info(
                         f"Applying {len(results)} batches of queued results"
@@ -945,7 +945,6 @@ class InboundSyncRequest(SyncRequest):
                     self._apply_results_dataframe(stream_names, all_dfs)
                 # update the stream state object too
                 self._apply_latest_states()
-                self._session.sql("commit").collect()
                 for stream_name in stream_names:
                     self._apply_results[stream_name] = None
                 self._apply_results = {}