use contextlib abstract base classes

Author: ElectricalBoy

src/lpdb_python/async_session/async_session.py

@@ -1,3 +1,4 @@
+from contextlib import AbstractAsyncContextManager
 from datetime import date
 from types import TracebackType
 from typing import Any, Literal, Optional, override
@@ -9,7 +10,7 @@
 __all__ = ["AsyncLpdbSession"]
 
 
-class AsyncLpdbSession(AbstractLpdbSession):
+class AsyncLpdbSession(AbstractLpdbSession, AbstractAsyncContextManager):
     """
     Asynchronous implementation of a LPDB session
     """
@@ -33,9 +34,6 @@ def __exit__(
     ) -> None:
         pass
 
-    async def __aenter__(self) -> "AsyncLpdbSession":
-        return self
-
     async def __aexit__(
         self,
         exc_type: Optional[type[BaseException]],

src/lpdb_python/session.py

@@ -1,4 +1,5 @@
 from abc import abstractmethod, ABC
+from contextlib import AbstractContextManager
 from datetime import date
 from functools import cache
 from http import HTTPStatus
@@ -318,7 +319,7 @@ def _parse_results(
         return result
 
 
-class LpdbSession(AbstractLpdbSession):
+class LpdbSession(AbstractLpdbSession, AbstractContextManager):
     """
     Implementation of a LPDB session
     """
@@ -330,9 +331,6 @@ def __init__(self, api_key: str, base_url=AbstractLpdbSession.BASE_URL):
         self.__session = requests.Session()
         self.__session.headers.update(self._get_header())
 
-    def __enter__(self) -> "LpdbSession":
-        return self
-
     def __exit__(
         self,
         exc_type: Optional[type[BaseException]],