omnata-plugin-runtime 0.4.6a102__py3-none-any.whl → 0.4.6a103__py3-none-any.whl

omnata_plugin_runtime/rate_limiting.py

@@ -354,10 +354,15 @@ class RetryWithLogging(Retry):
     """
         Adding extra logs before making a retry request     
     """
-    def __init__(self, *args: Any, thread_cancellation_token:threading.Event, **kwargs: Any) -> Any:
-        self.thread_cancellation_token = thread_cancellation_token
+    def __init__(self, *args: Any, **kwargs: Any) -> Any:
+        self.thread_cancellation_token:Optional[threading.Event] = None
         return super().__init__(*args, **kwargs)
     
+    def new(self, **kw):
+        new_retry = super().new(**kw)
+        new_retry.thread_cancellation_token = self.thread_cancellation_token
+        return new_retry
+    
     def sleep_for_retry(self, response=None):
         retry_after = self.get_retry_after(response)
         if retry_after:
@@ -406,13 +411,13 @@ class RateLimitedSession(requests.Session):
         self.statuses_to_include = statuses_to_include
 
         retry_strategy = RetryWithLogging(
-            thread_cancellation_token=thread_cancellation_token,
             total=max_retries,
             backoff_factor=backoff_factor,
             status_forcelist=statuses_to_include,
             allowed_methods=["HEAD", "GET", "OPTIONS", "POST", "PUT", "DELETE"],
             respect_retry_after_header=respect_retry_after_header
         )
+        retry_strategy.thread_cancellation_token = thread_cancellation_token
         adapter = HTTPAdapter(max_retries=retry_strategy)
         self.mount("https://", adapter)
         self.mount("http://", adapter)

{omnata_plugin_runtime-0.4.6a102.dist-info → omnata_plugin_runtime-0.4.6a103.dist-info}/METADATA

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

@@ -5,8 +5,8 @@ omnata_plugin_runtime/forms.py,sha256=pw_aKVsXSz47EP8PFBI3VDwdSN5IjvZxp8JTjO1V13
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
 omnata_plugin_runtime/omnata_plugin.py,sha256=9ZXKoQTyRV_Yzz_e6wEw7rJtt8YDCNrGT0F0tg7Fecc,108571
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=JAGEdVcy9QEXv7TO5zt7co64LTP8nqGusOc0sJG9GtU,29149
-omnata_plugin_runtime/rate_limiting.py,sha256=1kflNdgTGP2F8zpfG9qUaQ7EQsezLIinvXti1sNGOrw,23310
-omnata_plugin_runtime-0.4.6a102.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.4.6a102.dist-info/METADATA,sha256=hsbRzMmfp9n4sCVEfUBGJWptbsuRArbGl7W4C97iog4,1642
-omnata_plugin_runtime-0.4.6a102.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.4.6a102.dist-info/RECORD,,
+omnata_plugin_runtime/rate_limiting.py,sha256=27_sgEkD7kmQlfSF3IaM09Hs8MA5tXuacVUOFR4zwC0,23454
+omnata_plugin_runtime-0.4.6a103.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.4.6a103.dist-info/METADATA,sha256=0nG3b5kY7pgFtBtoVR4fxLIzxlM12MJjiZ6vFD4HyIc,1642
+omnata_plugin_runtime-0.4.6a103.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.4.6a103.dist-info/RECORD,,