api_request.py

from functools import lru_cache
from logging import getLogger

import requests
from requests.adapters import HTTPAdapter
from urllib3 import Retry

from config import API_KEY, MAX_RETRIES, RETRY_BACKOFF_FACTOR


__all__ = ["request"]


@lru_cache(1)
def _get_session():
    retry_strategy = Retry(
        total=MAX_RETRIES,
        backoff_factor=RETRY_BACKOFF_FACTOR,
    )

    session = requests.Session()
    session.headers.update(
        {
            "Authorization": "Bearer " + API_KEY,
            # https://dashflo.net/docs/api/pterodactyl/v1/
            "Content-Type": "application/json",
            "Accept": "application/json",
        }
    )

    adapter = HTTPAdapter(max_retries=retry_strategy)

    session.mount("https://", adapter)
    session.mount("http://", adapter)

    return session


def request(url, method: str = "GET", data=None) -> dict:
    for retry in range(MAX_RETRIES):
        try:
            response = _get_session().request(method=method, url=url, data=data)
            if not response:
                raise requests.exceptions.RequestException(
                    response.json()["errors"][0]["detail"]
                )

            return response.json()
        except (requests.exceptions.Timeout, requests.exceptions.ConnectionError) as e:
            getLogger(__name__).error(f"Network Error: {e}")
            exit(1)
        except requests.exceptions.RequestException as e:
            getLogger(__name__).error(f"Request Exception: {e}")
            exit(1)