main.py

from requests_cache import CachedSession
from hsr_client.datamodels.searchItem import *
from hsr_client.datamodels.character import Character
from hsr_client.routes import *
from typing import Union, List, LiteralString
from hsr_client.utils import generate_t, base36encode
from hsr_client.constants import *
from hsr_client.errors import *

from json import dump

class SRSClient:
    '''
    StarRailStation Website Client

    : initializes the client



    '''
    def __init__(self) -> None:
        
        self.__session = CachedSession(cache_name='srs.cache', backend='sqlite', expire_after=3600)

        self.__session.headers.update(
            {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36 Edg/112.0.1722.68',
             'referer': 'https://starrailstation.com/'}
        )

    def generate_hash_route(self, language: Language, route: Routes, goto: bool = False, item_id : Union[int, str]=''):
        '''
        
        :generates hashed route for fetching data

        --
        params
        --

        - language: Languages Enum
             Languages.EN, Languages.RU etc
        - route: a Routes object
        - goto: if you want to search in a specific route [True] 
             defaults to False
        
        - item_id : id of the item you want to search in a route
        
        '''

        if not isinstance(language, Language):
            raise InvalidLanguage
        
        url = route.generate_main_lang_path(language)
        if goto:
            if route.path is not None:
                url = f"{route.generate_goto_lang_path(language)}{item_id}.json"            

        hashed_path = base36encode(generate_t(url))

        return  f"{MAIN_ROUTE}{hashed_path}"




        
    
    def fetch(self, language: Language , route: Routes, goto: bool = False, item_id : Union[int, str] = '') -> List[dict] | dict | None:
        '''
        
        :fetches data from the api route
        --
        params
        --

        - language: Languages Enum
             Languages.EN, Languages.RU etc

        - route: a Routes object

        - goto: if you want to search in a specific route [True] 
             defaults to False
        
        - item_id : id of the item you want to search in a route
        
        '''

        if not isinstance(language, Language):
            raise InvalidLanguage

        response = self.__session.get(self.generate_hash_route(language, route, goto, item_id))
    

        if response.status_code  < 300:
            data = response.json()
            if 'entries' in data:
                return data['entries']
            else:
                return data
            

    def get_all_items(self,  type: Optional[Item], language: Language = Language.EN) -> list[SearchItem]:
        '''
        
        :fetches all items from api route
        --
        params
        --

        - language: Languages Enum
             Languages.EN, Languages.RU etc
        - type : a type object 
             Item.MATERIALS, Item.PLAYERCARDS, Item.CHARACTERS etc
        
        
        '''

        if not isinstance(language, Language):
            raise InvalidLanguage

        response = self.fetch(language, SEARCH, False)

        if response is not None:
            all_items = [SearchItem(**{ **d, **{'id': d['url'].split("/")[1]}}) for d in response]
            if type is not None:
                return list(filter(lambda x: x.type == type, all_items))
           
      
            return all_items
        
        raise Exception('Not enough arguments provided, or nothing is returned from api call!')