Lazily instantiate the session

Author: eclipseo

pydepsdev/api.py

@@ -48,7 +48,7 @@
 
 
 class DepsdevAPI:
-    session: aiohttp.ClientSession
+    session: Optional[aiohttp.ClientSession]
     headers: Dict[str, str]
     timeout_duration: float
     max_retries: int
@@ -71,7 +71,7 @@ def __init__(
             base_backoff (float): Initial backoff interval in seconds.
             max_backoff (float): Maximum backoff interval in seconds.
         """
-        self.session = aiohttp.ClientSession()
+        self.session = None
         self.headers = {"Content-Type": "application/json"}
         self.timeout_duration = timeout_duration
         self.max_retries = max_retries
@@ -85,22 +85,37 @@ def __init__(
             max_backoff,
         )
 
+    async def _getsession(self) -> aiohttp.ClientSession:
+        """
+        Lazily instantiate and return an aiohttp.ClientSession
+        within a running event loop.
+
+        Returns:
+            aiohttp.ClientSession: The current aiohttp client session.
+        """
+        if self.session is None:
+            self.session = aiohttp.ClientSession()
+        return self.session
+
     async def close(self) -> None:
         """
         Close the underlying HTTP session.
 
         Returns:
             None
         """
-        await self.session.close()
+        if self.session is not None:
+            await self.session.close()
+            self.session = None
 
     async def __aenter__(self) -> "DepsdevAPI":
         """
-        Enter the async context.
+        Enter the async context. Ensures the session is created.
 
         Returns:
             DepsdevAPI: The current API client instance.
         """
+        await self._getsession()
         return self
 
     async def __aexit__(
@@ -145,6 +160,7 @@ async def fetch_data(
         Raises:
             APIError: On HTTP client/server error or network failure.
         """
+        session = await self._getsession()
         attempt = 0
         while attempt <= self.max_retries:
             logger.info(
@@ -156,7 +172,7 @@ async def fetch_data(
                 self.max_retries + 1,
             )
             try:
-                async with self.session.request(
+                async with session.request(
                     method,
                     request_url,
                     headers=self.headers,

pydepsdev/utils.py

@@ -24,6 +24,7 @@
     SUPPORTED_SYSTEMS_DEPENDENTS,
     SUPPORTED_SYSTEMS_CAPABILITIES,
     SUPPORTED_SYSTEMS_QUERY,
+    SUPPORTED_HASHES,
 )