src/model/wiki/SupabaseRest.py

import logging
import random
from enum import StrEnum

import requests

import resources.Environment as Env
from src.model.enums.AssetPath import AssetPath
from src.model.error.CustomException import WikiException
from src.model.game.GameDifficulty import GameDifficulty
from src.model.wiki.Character import Character
from src.model.wiki.Terminology import Terminology
from src.utils.file_utils import get_list_from_json


class SupabaseTableName(StrEnum):
    CHARACTER = "character"
    TERMINOLOGY = "terminology"

    def get_asset_path(self) -> AssetPath:
        return TABLE_TO_LOCAL_ASSET_PATH[self]


TABLE_TO_LOCAL_ASSET_PATH = {
    SupabaseTableName.CHARACTER: AssetPath.CHARACTERS,
    SupabaseTableName.TERMINOLOGY: AssetPath.TERMINOLOGIES,
}


class SupabaseRest:
    def __init__(self):
        """
        Creates a wiki object
        """
        self.rest_url = Env.SUPABASE_REST_URL.get()
        self.api_key = Env.SUPABASE_API_KEY.get()

    def make_get_request(self, table: SupabaseTableName) -> list[dict]:
        """
        Make a request to the Supabase REST API
        :param table: The table of the resource
        :return: The response
        """

        # If Supabase is not set up, use local file resource instead
        if not self.rest_url or not self.api_key:
            return get_list_from_json(table.get_asset_path())

        full_path = f"{self.rest_url}{table}"
        # Make request
        response = requests.get(
            url=full_path, headers={"apikey": self.api_key, "Content-Type": "application/json"}
        )

        # Check response
        if response.status_code != 200:
            logging.error(f"Error making request to {full_path}, using local resource")
            return get_list_from_json(table.get_asset_path())

        # Return response
        return response.json()

    @staticmethod
    def get_random_character(difficulty: GameDifficulty) -> Character:
        """
        Get a random character
        :param difficulty: The difficulty level
        :return: A random character
        """

        # Get a random character
        response: list[dict] = SupabaseRest().make_get_request(SupabaseTableName.CHARACTER)

        if len(response) == 0:
            raise WikiException("No characters found")

        random.shuffle(response)

        if difficulty is None:
            return Character(**response[0])

        for char in response:
            if char["difficulty"] == difficulty.value:
                return Character(**char)

        raise WikiException(f"No characters found with difficulty {difficulty} or lower")

    @staticmethod
    def get_random_terminology(
        max_len: int = None,
        only_letters: bool = False,
        consider_len_without_space: bool = False,
        allow_spaces: bool = True,
        min_unique_characters: int = None,
        max_unique_characters: int = None,
    ) -> Terminology:
        """
        Get a random terminology
        :param max_len: The maximum length of the terminology
        :param only_letters: Whether the terminology should only contain letters
        :param consider_len_without_space: Whether the length of the terminology should be
        considered without spaces
        :param allow_spaces: Whether the terminology can contain spaces if only_letters is True
        :param min_unique_characters: The minimum amount of unique characters in the terminology
        :param max_unique_characters: The maximum amount of unique characters in the terminology
        :return: A random terminology
        """

        # Get a random terminology
        response: list[dict] = SupabaseRest().make_get_request(SupabaseTableName.TERMINOLOGY)
        random.shuffle(response)

        for term in response:
            if max_len is not None and len(term["name"]) > max_len:
                if not (
                    consider_len_without_space
                    and len(str(term["name"]).replace(" ", "")) <= max_len
                ):
                    continue

            if only_letters and not str(term["name"]).isalpha():
                if not (allow_spaces and str(term["name"]).replace(" ", "").isalpha()):
                    continue

            if (
                min_unique_characters is not None
                and len(set(term["name"])) < min_unique_characters
            ):
                continue

            if (
                max_unique_characters is not None
                and len(set(term["name"])) > max_unique_characters
            ):
                continue

            return Terminology(**term)

        raise WikiException("No terminologies found")