api_client: wrap methods

Flared · Aug 15, 2024 · 3b32b58 · 3b32b58
1 parent c975ccd
commit 3b32b58
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/flareio/api_client.py b/flareio/api_client.py
@@ -7,6 +7,13 @@
 import typing as t
 
 
+T = t.TypeVar("T")
+
+
+def _type_wraps(_: T) -> t.Callable[..., T]:
+    return lambda x: x # type: ignore
+
+
 class FlareApiClient:
     def __init__(
         self,
@@ -79,27 +86,31 @@ def _request(
             headers=headers,
         )
 
+    @_type_wraps(requests.post)
     def post(
         self,
         *args: t.Any,
         **kwargs: t.Any,
     ) -> requests.Response:
         return self._request("POST", *args, **kwargs)
 
+    @_type_wraps(requests.get)
     def get(
         self,
         *args: t.Any,
         **kwargs: t.Any,
     ) -> requests.Response:
         return self._request("GET", *args, **kwargs)
 
+    @_type_wraps(requests.put)
     def put(
         self,
         *args: t.Any,
         **kwargs: t.Any,
     ) -> requests.Response:
         return self._request("PUT", *args, **kwargs)
 
+    @_type_wraps(requests.delete)
     def delete(
         self,
         *args: t.Any,