Unify the core request structure

app/drb_cdb_api.py

@@ -2,8 +2,9 @@ import httpx
 import json
 import asyncio # Import asyncio for running the example usage
 from drb_cdb_types import ConfigGenerator
+from base_api import BaseAPI
 
-class DRBCDBAPI:
+class DRBCDBAPI(BaseAPI):
     """
     An asynchronous Python wrapper for interacting with the FastAPI application.
     Uses httpx for asynchronous HTTP requests.
@@ -15,79 +16,11 @@ class DRBCDBAPI:
         Args:
             base_url: The base URL of the FastAPI application (e.g., "http://localhost:8000").
         """
+        super().__init__()
         self.base_url = base_url
         # Use an AsyncClient for making asynchronous requests
         self._client = httpx.AsyncClient()
 
-    async def __aenter__(self):
-        """Allows using the client with async with."""
-        return self
-
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
-        """Ensures the client is closed when exiting async with."""
-        await self.close()
-
-    async def close(self):
-        """Closes the underlying asynchronous HTTP client."""
-        await self._client.close()
-
-    async def _post(self, endpoint: str, data: dict = None):
-        """
-        Asynchronous helper method for making POST requests.
-
-        Args:
-            endpoint: The API endpoint (e.g., "/op25/start").
-            data: The data to send in the request body (as a dictionary).
-
-        Returns:
-            The JSON response from the API.
-
-        Raises:
-            httpx.RequestError: If the request fails.
-            httpx.HTTPStatusError: If the API returns an error status code (4xx or 5xx).
-        """
-        url = f"{self.base_url}{endpoint}"
-        try:
-            # Use await with the asynchronous httpx client
-            response = await self._client.post(url, json=data)
-            response.raise_for_status() # Raise HTTPStatusError for bad responses (4xx or 5xx)
-            return response.json()
-        except httpx.HTTPStatusError as e:
-            print(f"HTTP error occurred: {e}")
-            print(f"Response body: {e.response.text}") # Access response text from the exception
-            raise
-        except httpx.RequestError as e:
-            print(f"Request to '{url}' failed: {e}")
-            raise
-
-    async def _get(self, endpoint: str):
-        """
-        Asynchronous helper method for making GET requests.
-
-        Args:
-            endpoint: The API endpoint (e.g., "/op25/status").
-
-        Returns:
-            The JSON response from the API.
-
-        Raises:
-            httpx.RequestError: If the request fails.
-            httpx.HTTPStatusError: If the API returns an error status code (4xx or 5xx).
-        """
-        url = f"{self.base_url}{endpoint}"
-        try:
-            # Use await with the asynchronous httpx client
-            response = await self._client.get(url)
-            response.raise_for_status() # Raise HTTPStatusError for bad responses (4xx or 5xx)
-            return response.json()
-        except httpx.HTTPStatusError as e:
-            print(f"HTTP error occurred: {e}")
-            print(f"Response body: {e.response.text}") # Access response text from the exception
-            raise
-        except httpx.RequestError as e:
-            print(f"Request to '{url}' failed: {e}")
-            raise
-
     # --- OP25 Endpoints ---
 
     async def start_op25(self):