restfly

import os from restfly.session import APISession from restfly.endpoint import APIEndpoint # Define your base API session class MyAPI(APISession): _url = 'https://httpbin.org' # Example of adding an API key, if needed def __init__(self, api_key: str = os.environ.get('MY_API_KEY', ''), **kwargs): self._api_key = api_key super().__init__(**kwargs) def _build_session(self, **kwargs): super()._build_session(**kwargs) if self._api_key: self._session.headers.update({ 'X-API-Key': self._api_key }) # Define an endpoint for a specific resource class StatusAPI(APIEndpoint): _path = 'status' def get_status_code(self, code: int) -> dict: # _req is a protected method for making requests, leveraging _path return self._req('GET', path=str(code)).json() # Instantiate and use the API my_api_client = MyAPI(api_key='your_optional_api_key_here') # Link the endpoint to the API session my_api_client.status = StatusAPI(my_api_client) try: # Call a simple GET request directly from the session response_get = my_api_client.get('get').json() print(f"GET /get Response: {response_get.get('url')}") # Use the defined endpoint status_200 = my_api_client.status.get_status_code(200) print(f"GET /status/200 Response: {status_200}") status_404 = my_api_client.status.get_status_code(404) print(f"GET /status/404 Response: {status_404}") # This will raise an APIError except Exception as e: print(f"An error occurred: {e}") # Example of handling specific errors from restfly.errors import NotFoundError, APIError try: my_api_client.status.get_status_code(404) except NotFoundError as e: print(f"Caught specific NotFoundError: {e}") except APIError as e: print(f"Caught generic APIError: {e}")