use as context managers

cent/client/async_client.py

@@ -269,3 +269,9 @@ async def __call__(
         :return: Centrifugo response
         """
         return await self._session(self._api_key, request, timeout=request_timeout)
+
+    async def __aenter__(self) -> "AsyncClient":
+        return self
+
+    async def __aexit__(self, *kwargs: Any) -> None:
+        await self.close()

cent/client/grpc_client.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, TypeVar, List
+from typing import Optional, Dict, TypeVar, List, Any
 
 from cent.client.session import GrpcSession
 from cent.base import CentRequest
@@ -244,3 +244,9 @@ async def __call__(
         :return: Centrifugo response
         """
         return await self._session(request, request_timeout)
+
+    async def __aenter__(self) -> "GrpcClient":
+        return self
+
+    async def __aexit__(self, *kwargs: Any) -> None:
+        await self.close()

cent/client/session/aiohttp.py

@@ -45,7 +45,8 @@ async def make_request(
         timeout: Optional[float] = None,
     ) -> CentType:
         session = self._session
-        session.headers["X-API-Key"] = api_key
+        if api_key:
+            session.headers["X-API-Key"] = api_key
 
         if isinstance(request, BatchRequest):
             json_data = self.get_batch_json_data(request)

cent/client/session/base_http_async.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, Optional, cast
+from typing import Optional, cast
 
 from cent.client.session.base_http import BaseHttpSession
 from cent.base import CentType, CentRequest
@@ -36,9 +36,3 @@ async def __call__(
         timeout: Optional[float] = None,
     ) -> CentType:
         return cast(CentType, await self.make_request(api_key, request, timeout))
-
-    async def __aenter__(self) -> "BaseHttpAsyncSession":
-        return self
-
-    async def __aexit__(self, *kwargs: Any) -> None:
-        await self.close()

cent/client/session/requests.py

@@ -40,7 +40,8 @@ def make_request(
         request: CentRequest[CentType],
         timeout: Optional[float] = None,
     ) -> CentType:
-        self._session.headers["X-API-Key"] = api_key
+        if api_key:
+            self._session.headers["X-API-Key"] = api_key
         if isinstance(request, BatchRequest):
             json_data = self.get_batch_json_data(request)
         else:

cent/client/sync_client.py

@@ -269,3 +269,9 @@ def __call__(self, request: CentRequest[T], request_timeout: Optional[float] = N
         :return: Centrifugo response
         """
         return self._session(self._api_key, request, timeout=request_timeout)
+
+    def __enter__(self) -> "Client":
+        return self
+
+    def __exit__(self, *kwargs: Any) -> None:
+        self.close()