omnata-plugin-runtime 0.4.2a96__tar.gz → 0.4.3__tar.gz

{omnata_plugin_runtime-0.4.2a96 → omnata_plugin_runtime-0.4.3}/PKG-INFO

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

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

@@ -1126,6 +1126,16 @@ class InboundSyncRequest(SyncRequest):
         Sets the record count for a stream, used to provide progress updates.
         """
         self._stream_record_counts[stream_name] = count
+    
+    def set_stream_total_records_estimate(self, stream_name: str, count: int):
+        """
+        Sets the total record count for a stream, used to provide progress updates.
+        This should be a best estimate of the number of changes which will be fetched from the source.
+        In other words, as results are enqueued they should build toward reaching this number as the sync progresses.
+        This does not have to be exact, it's just to give the user some feedback while the sync is running.
+        Totals will always be replaced by the actual count when the stream is completed.
+        """
+        self._total_records_estimate[stream_name] = count
 
     def _enqueue_state(self, stream_name: str, new_state: Any):
         """

{omnata_plugin_runtime-0.4.2a96 → omnata_plugin_runtime-0.4.3}/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