libentry 1.21.2__py3-none-any.whl → 1.21.3__py3-none-any.whl

libentry/api.py

@@ -10,6 +10,7 @@ __all__ = [
     "APIClient",
 ]
 
+import asyncio
 from dataclasses import dataclass, field
 from time import sleep
 from typing import Any, AsyncIterable, Callable, Iterable, List, Literal, Mapping, Optional, Tuple, Union
@@ -216,6 +217,10 @@ class BaseClient:
             stream: bool,
             verify: bool,
     ) -> Union[bytes, Iterable[bytes]]:
+        if headers is None:
+            headers = self.headers
+        else:
+            headers = {**self.headers, **headers}
         response = self.URLLIB3_POOL[int(verify)].request(
             method=method,
             url=url,
@@ -304,6 +309,10 @@ class BaseClient:
             stream: bool,
             verify: bool,
     ):
+        if headers is None:
+            headers = self.headers
+        else:
+            headers = {**self.headers, **headers}
         if not stream:
             async with httpx.AsyncClient(headers=headers, verify=verify) as client:
                 response = await client.request(
@@ -377,7 +386,7 @@ class BaseClient:
                 err = e
                 if callable(on_error):
                     on_error(e)
-            sleep(interval)
+            await asyncio.sleep(interval)
         raise err
 
 
@@ -404,8 +413,10 @@ class APIClient(BaseClient):
             verify: Optional[bool] = None,
     ):
         full_url = urljoin(self.base_url, path)
-        headers = {**self.headers}
-        headers["Accept"] = headers["Accept"] + f"; stream={int(stream)}"
+        if headers is None:
+            headers = {}
+        accept = headers["Accept"] if "Accept" in headers else self.headers["Accept"]
+        headers["Accept"] = accept + f"; stream={int(stream)}"
         body = json.dumps(json_data) if json_data is not None else None
         content = super().request(
             method,
@@ -481,6 +492,7 @@ class APIClient(BaseClient):
             self,
             path: Optional[str] = None,
             *,
+            headers: Optional[Mapping[str, str]] = None,
             timeout: float = 15,
             num_trials: int = 5,
             interval: float = 1,
@@ -490,6 +502,7 @@ class APIClient(BaseClient):
         return self.request(
             "GET",
             path,
+            headers=headers,
             timeout=timeout,
             num_trials=num_trials,
             interval=interval,
@@ -502,6 +515,7 @@ class APIClient(BaseClient):
             path: Optional[str] = None,
             json_data: Optional[Mapping] = None,
             *,
+            headers: Optional[Mapping[str, str]] = None,
             timeout: float = 15,
             num_trials: int = 5,
             interval: float = 1,
@@ -518,6 +532,7 @@ class APIClient(BaseClient):
             "POST",
             path,
             json_data=json_data,
+            headers=headers,
             timeout=timeout,
             num_trials=num_trials,
             interval=interval,
@@ -552,8 +567,10 @@ class APIClient(BaseClient):
             verify: Optional[bool] = None,
     ):
         full_url = urljoin(self.base_url, path)
-        headers = {**self.headers}
-        headers["Accept"] = headers["Accept"] + f"; stream={int(stream)}"
+        if headers is None:
+            headers = {}
+        accept = headers["Accept"] if "Accept" in headers else self.headers["Accept"]
+        headers["Accept"] = accept + f"; stream={int(stream)}"
         body = json.dumps(json_data) if json_data is not None else None
         content = await super().request_async(
             method,
@@ -629,6 +646,7 @@ class APIClient(BaseClient):
             self,
             path: Optional[str] = None,
             *,
+            headers: Optional[Mapping[str, str]] = None,
             timeout: float = 15,
             num_trials: int = 5,
             interval: float = 1,
@@ -638,6 +656,7 @@ class APIClient(BaseClient):
         return await self.request_async(
             "GET",
             path,
+            headers=headers,
             timeout=timeout,
             num_trials=num_trials,
             interval=interval,
@@ -650,6 +669,7 @@ class APIClient(BaseClient):
             path: Optional[str] = None,
             json_data: Optional[Mapping] = None,
             *,
+            headers: Optional[Mapping[str, str]] = None,
             timeout: float = 15,
             num_trials: int = 5,
             interval: float = 1,
@@ -666,6 +686,7 @@ class APIClient(BaseClient):
             "POST",
             path,
             json_data=json_data,
+            headers=headers,
             timeout=timeout,
             num_trials=num_trials,
             interval=interval,

{libentry-1.21.2.dist-info → libentry-1.21.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: libentry
-Version: 1.21.2
+Version: 1.21.3
 Summary: Entries for experimental utilities.
 Home-page: https://github.com/XoriieInpottn/libentry
 Author: xi

{libentry-1.21.2.dist-info → libentry-1.21.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 libentry/__init__.py,sha256=ko2YBIIx5H3dD0tedBkialzJGEDczFaP_PZmT1cIlak,148
-libentry/api.py,sha256=zlZF0IWyQu8GPEVAM0J6K8BOSgqI06p8JJVoCNaTCBo,22110
+libentry/api.py,sha256=ei67QL_PEMs1J_utCuh5MifPbAw17BOYnaYtc8d35UY,22923
 libentry/argparse.py,sha256=NxzXV-jBN51ReZsNs5aeyOfzwYQ5A5nJ95rWoa-FYCs,10415
 libentry/dataclasses.py,sha256=AQV2PuxplJCwGZ5HKX72U-z-POUhTdy3XtpEK9KNIGQ,4541
 libentry/executor.py,sha256=cTV0WxJi0nU1TP-cOwmeodN8DD6L1691M2HIQsJtGrU,6582
@@ -15,10 +15,10 @@ 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.21.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-libentry-1.21.2.dist-info/METADATA,sha256=M2-N8UEjYrgrXC-85vZ2-ts0SeNjOTx9_Vjclgdu8Xg,813
-libentry-1.21.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-libentry-1.21.2.dist-info/entry_points.txt,sha256=vgHmJZhM-kqM7U9S179UwDD3pM232tpzJ5NntncXi_8,62
-libentry-1.21.2.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
-libentry-1.21.2.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-libentry-1.21.2.dist-info/RECORD,,
+libentry-1.21.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+libentry-1.21.3.dist-info/METADATA,sha256=43wuxNjrLdOaLoW5E55_JEMhGAPAAVTFe5WltenN7L0,813
+libentry-1.21.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+libentry-1.21.3.dist-info/entry_points.txt,sha256=vgHmJZhM-kqM7U9S179UwDD3pM232tpzJ5NntncXi_8,62
+libentry-1.21.3.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
+libentry-1.21.3.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+libentry-1.21.3.dist-info/RECORD,,