omnata-plugin-runtime 0.4.1a92__tar.gz → 0.4.2__tar.gz

{omnata_plugin_runtime-0.4.1a92 → omnata_plugin_runtime-0.4.2}/PKG-INFO

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

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

@@ -875,6 +875,28 @@ class InboundSyncRequest(SyncRequest):
         :param bool test_replay_mode: When enabled, it is safe to assume that HTTP requests are hitting a re-recorded log, so there is no need to wait in between polling
         :return: nothing
         """
+        self.streams = streams
+        self._streams_dict: Dict[str, StoredStreamConfiguration] = {
+            s.stream_name: s for s in streams
+        }
+        self._stream_record_counts: Dict[str, int] = {
+            stream_name: 0 for stream_name in self._streams_dict.keys()
+        }
+        
+        # These are similar to the results, but represent requests to delete records by some criteria
+        self._apply_results_criteria_deletes: Dict[str, List[pandas.DataFrame]] = {}
+        self._latest_states: Dict[str, Any] = {}
+        self._temp_tables = {}
+        self._temp_table_lock = threading.Lock()
+        self._results_exist: Dict[
+            str, bool
+        ] = {}  # track whether or not results exist for stream
+        self._total_records_estimate: Optional[Dict[str,int]] = {}
+        self._stream_change_counts: Dict[str, int] = {
+            stream_name: 0 for stream_name in self._streams_dict.keys()
+        }
+        self._completed_streams: List[str] = []
+        self._omnata_log_handler = omnata_log_handler
         SyncRequest.__init__(
             self,
             run_id=run_id,
@@ -890,32 +912,11 @@ class InboundSyncRequest(SyncRequest):
             development_mode=development_mode,
             test_replay_mode=test_replay_mode
         )
-        self.streams = streams
-        self._streams_dict: Dict[str, StoredStreamConfiguration] = {
-            s.stream_name: s for s in streams
-        }
-        self._apply_results: Dict[str, List[pandas.DataFrame]] = {}
         # named by convention, see SyncRunProcessor.enqueue
         self._criteria_deletes_table_name = (
             f"{self._source_app_name}.{self._results_schema_name}.{self._results_table_name}_CRITERIA_DELETES"
         )
-        # These are similar to the results, but represent requests to delete records by some criteria
-        self._apply_results_criteria_deletes: Dict[str, List[pandas.DataFrame]] = {}
-        self._latest_states: Dict[str, Any] = {}
-        self._temp_tables = {}
-        self._temp_table_lock = threading.Lock()
-        self._results_exist: Dict[
-            str, bool
-        ] = {}  # track whether or not results exist for stream
-        self._stream_record_counts: Dict[str, int] = {
-            stream_name: 0 for stream_name in self._streams_dict.keys()
-        }
-        self._total_records_estimate: Optional[Dict[str,int]] = {}
-        self._stream_change_counts: Dict[str, int] = {
-            stream_name: 0 for stream_name in self._streams_dict.keys()
-        }
-        self._completed_streams: List[str] = []
-        self._omnata_log_handler = omnata_log_handler
+        self._apply_results: Dict[str, List[pandas.DataFrame]] = {}
 
     def apply_results_queue(self):
         """

{omnata_plugin_runtime-0.4.1a92 → omnata_plugin_runtime-0.4.2}/src/omnata_plugin_runtime/plugin_entrypoints.py

@@ -218,7 +218,9 @@ class PluginEntrypoint:
                     inbound_sync_request.update_activity("Staging remaining records")
                     logger.info("Calling apply_results_queue")
                     inbound_sync_request.apply_results_queue()
+                    logger.info("Calling apply_rate_limit_state")
                     inbound_sync_request.apply_rate_limit_state()
+                    logger.info("Calling apply_progress_updates")
                     inbound_sync_request.apply_progress_updates()
                 if inbound_sync_request.deadline_reached:
                     # if we actually hit the deadline, this is flagged by the cancellation checking worker and the cancellation