omnata-plugin-runtime 0.4.6a102__tar.gz → 0.4.6a103__tar.gz

{omnata_plugin_runtime-0.4.6a102 → omnata_plugin_runtime-0.4.6a103}/PKG-INFO

@@ -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 → omnata_plugin_runtime-0.4.6a103}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.4.6-a102"
+version = "0.4.6-a103"
 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.6a102 → omnata_plugin_runtime-0.4.6a103}/src/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)