Unify the core request structure

This commit is contained in:
Logan Cusano
2025-04-27 03:06:53 -04:00
parent 19c0bd2ef8
commit de8570d014
3 changed files with 106 additions and 112 deletions

View File

@@ -1,7 +1,8 @@
import httpx
import json
from base_api import BaseAPI
class RadioAPIClient:
class RadioAPIClient(BaseAPI):
"""
A client wrapper for interacting with the Radio App Server API.
Uses httpx for asynchronous HTTP requests.
@@ -13,51 +14,11 @@ class RadioAPIClient:
Args:
base_url (str): The base URL of the server's API (default is http://localhost:5000).
"""
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 HTTP client."""
await self._client.close()
async def _request(self, method, endpoint, **kwargs):
"""
Helper method to make an asynchronous HTTP request.
Args:
method (str): The HTTP method (e.g., 'GET', 'POST').
endpoint (str): The API endpoint path (e.g., '/systems').
**kwargs: Additional arguments for httpx.AsyncClient.request.
Returns:
dict: The JSON response from the API.
Raises:
httpx.HTTPStatusError: If the request returns a non-2xx status code.
httpx.RequestError: For other request-related errors.
"""
url = f"{self.base_url}{endpoint}"
try:
response = await self._client.request(method, url, **kwargs)
response.raise_for_status() # Raise an exception for bad status codes (4xx or 5xx)
return response.json()
except httpx.HTTPStatusError as e:
print(f"HTTP error occurred: {e}")
# You might want to return the error response body or raise the exception
raise
except httpx.RequestError as e:
print(f"An error occurred while requesting {e.request.url!r}: {e}")
raise
async def get_systems(self):
"""
Retrieves a summary list of all available radio systems.