libentry 1.17__py3-none-any.whl → 1.18.1__py3-none-any.whl

libentry/api.py

@@ -166,6 +166,9 @@ class ServiceError(RuntimeError):
         return "".join(lines)
 
 
+ErrorCallback = Callable[[Exception], None]
+
+
 class APIClient:
 
     def __init__(
@@ -195,29 +198,41 @@ class APIClient:
             self,
             method: str,
             url: str,
-            num_trials: int = 5,
-            retry_factor: float = 2,
-            timeout: float = 5,
+            num_trials: int,
+            timeout: float,
+            interval: float,
+            retry_factor: float,
+            on_error: Optional[ErrorCallback] = None,
             **kwargs
     ):
         err = None
-        for _ in range(num_trials):
+        for i in range(num_trials):
             try:
-                return requests.request(method, url, timeout=timeout, **kwargs)
+                return requests.request(
+                    method=method,
+                    url=url,
+                    timeout=timeout * (1 + i * retry_factor),
+                    **kwargs
+                )
             except requests.Timeout as e:
                 err = e
+                if callable(on_error):
+                    on_error(e)
             except requests.ConnectionError as e:
                 err = e
-                sleep(timeout)
-            timeout *= retry_factor
+                if callable(on_error):
+                    on_error(e)
+            sleep(interval)
         raise err
 
     def get(
             self,
             path: Optional[str] = None, *,
             num_trials: int = 5,
-            retry_factor: float = 2,
-            timeout: float = 15
+            timeout: float = 15,
+            interval: float = 1,
+            retry_factor: float = 0.5,
+            on_error: Optional[ErrorCallback] = None
     ):
         full_url = urljoin(self.base_url, path)
         response = self._request(
@@ -226,8 +241,10 @@ class APIClient:
             headers=self.headers,
             verify=self.verify,
             num_trials=num_trials,
+            timeout=timeout,
+            interval=interval,
             retry_factor=retry_factor,
-            timeout=timeout
+            on_error=on_error
         )
 
         if response.status_code != 200:
@@ -247,8 +264,10 @@ class APIClient:
             stream: bool = False,
             exhaust_stream: bool = False,
             num_trials: int = 5,
-            retry_factor: float = 2,
             timeout: float = 15,
+            interval: float = 1,
+            retry_factor: float = 0.5,
+            on_error: Optional[ErrorCallback] = None,
             chunk_delimiter: str = "\n\n",
             chunk_prefix: str = None,
             chunk_suffix: str = None,
@@ -267,8 +286,10 @@ class APIClient:
             verify=self.verify,
             stream=stream,
             num_trials=num_trials,
+            timeout=timeout,
+            interval=interval,
             retry_factor=retry_factor,
-            timeout=timeout
+            on_error=on_error
         )
         if response.status_code != 200:
             text = response.text

{libentry-1.17.dist-info → libentry-1.18.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: libentry
-Version: 1.17
+Version: 1.18.1
 Summary: Entries for experimental utilities.
 Home-page: https://github.com/XoriieInpottn/libentry
 Author: xi
@@ -13,6 +13,7 @@ Requires-Dist: Flask
 Requires-Dist: PyYAML
 Requires-Dist: gunicorn
 Requires-Dist: json5
+Requires-Dist: numpy
 Requires-Dist: pydantic
 Requires-Dist: requests
 

{libentry-1.17.dist-info → libentry-1.18.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 libentry/__init__.py,sha256=rDBip9M1Xb1N4wMKE1ni_DldrQbkRjp8DxPkTp3K2qo,170
-libentry/api.py,sha256=zvW4YugZ7k0wmVVFTnOYHgNqWQPf8zTTPYOnED-iUJM,10097
+libentry/api.py,sha256=LqTa8T8Oh8z5BybzrgtdC0HhMXWL2-DZPdOdBHb6Vl0,10753
 libentry/argparse.py,sha256=NxzXV-jBN51ReZsNs5aeyOfzwYQ5A5nJ95rWoa-FYCs,10415
 libentry/dataclasses.py,sha256=AQV2PuxplJCwGZ5HKX72U-z-POUhTdy3XtpEK9KNIGQ,4541
 libentry/executor.py,sha256=cTV0WxJi0nU1TP-cOwmeodN8DD6L1691M2HIQsJtGrU,6582
@@ -15,9 +15,9 @@ libentry/service/list.py,sha256=ElHWhTgShGOhaxMUEwVbMXos0NQKjHsODboiQ-3AMwE,1397
 libentry/service/running.py,sha256=FrPJoJX6wYxcHIysoatAxhW3LajCCm0Gx6l7__6sULQ,5105
 libentry/service/start.py,sha256=mZT7b9rVULvzy9GTZwxWnciCHgv9dbGN2JbxM60OMn4,1270
 libentry/service/stop.py,sha256=wOpwZgrEJ7QirntfvibGq-XsTC6b3ELhzRW2zezh-0s,1187
-libentry-1.17.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-libentry-1.17.dist-info/METADATA,sha256=6kVrTOffjfB4RzkedsLggG0DVy8la9fIx2c3iP7cuAw,791
-libentry-1.17.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-libentry-1.17.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
-libentry-1.17.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-libentry-1.17.dist-info/RECORD,,
+libentry-1.18.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+libentry-1.18.1.dist-info/METADATA,sha256=TMT4FlCizP9Ca43z9oWAbHi57EKTNGha6Sd8ClXg4bA,814
+libentry-1.18.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+libentry-1.18.1.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
+libentry-1.18.1.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+libentry-1.18.1.dist-info/RECORD,,