omnata-plugin-runtime 0.5.4a121__py3-none-any.whl → 0.5.4a122__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -2042,9 +2042,13 @@ def __managed_inbound_processing_worker(
             # restore the first argument, was originally the dataframe/generator but now it's the appropriately sized dataframe
             try:
                 logger.info(f"worker {worker_index} processing stream {stream.stream_name}, invoking plugin class method {method.__name__}")
-                method(plugin_class_obj, *(stream, *method_args), **method_kwargs)
-                logger.info(f"worker {worker_index} completed processing stream {stream.stream_name}, marking as complete")
-                plugin_class_obj._sync_request.mark_stream_complete(stream.stream_name)
+                result = method(plugin_class_obj, *(stream, *method_args), **method_kwargs)
+                logger.info(f"worker {worker_index} completed processing stream {stream.stream_name}")
+                if result is not None and result is False:
+                    logger.info(f"worker {worker_index} requested that {stream.stream_name} be not marked as complete")
+                else:
+                    logger.info(f"worker {worker_index} marking stream {stream.stream_name} as complete")
+                    plugin_class_obj._sync_request.mark_stream_complete(stream.stream_name)
             except InterruptedWhileWaitingException:
                 # If an inbound run is cancelled while waiting for rate limiting, this should mean that
                 # the cancellation is handled elsewhere, so we don't need to do anything special here other than stop waiting

{omnata_plugin_runtime-0.5.4a121.dist-info → omnata_plugin_runtime-0.5.4a122.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.4a121
+Version: 0.5.4a122
 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.5.4a121.dist-info → omnata_plugin_runtime-0.5.4a122.dist-info}/RECORD

@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=FxzTqri4no8ClkOm7vZADG8aD47jcGBCTTQDEORmOJM,
 omnata_plugin_runtime/configuration.py,sha256=4eKYhw6_l-RUVUwNU5NlTpPwL7xZRs6VLaCKf8OIKuY,36420
 omnata_plugin_runtime/forms.py,sha256=GzSPEwcijsoPCXEO1mHiE8ylvX_KSE5TkhwqkymA2Ss,19755
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
-omnata_plugin_runtime/omnata_plugin.py,sha256=zIk8dS5m4pXsVy_2im-Cd-t1uQV9hf1seosb1T2jGGs,110880
+omnata_plugin_runtime/omnata_plugin.py,sha256=_13zy630P_fPyrqCuRStXOy9iIEZnUSZNBSeHYxkujQ,111179
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=JirYUbPBaN0UMh9t_uAHDdhaQZ7NUhdMJ11eHRKOoNY,29302
 omnata_plugin_runtime/rate_limiting.py,sha256=DVQ_bc-mVLBkrU1PTns1MWXhHiLpSB5HkWCcdePtJ2A,25611
-omnata_plugin_runtime-0.5.4a121.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.4a121.dist-info/METADATA,sha256=kEkHqK291jhDAXEY-5BYmP2m-6RxTt3YPjv7-o7mFlY,1951
-omnata_plugin_runtime-0.5.4a121.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.4a121.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.4a122.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.4a122.dist-info/METADATA,sha256=nhK1zNYD6bVZfgYkcH6aqxs-EfutxlfD2vq1W4Sc8SY,1951
+omnata_plugin_runtime-0.5.4a122.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.4a122.dist-info/RECORD,,