From 787eb0c9ca53d6e1c1c0ac85c7f00c6952ef87bd Mon Sep 17 00:00:00 2001 From: benex Date: Tue, 3 Dec 2024 12:29:47 +0300 Subject: [PATCH] refactor: conform all provider functions --- fastanime/libs/anime_provider/allanime/constants.py | 3 ++- fastanime/libs/anime_provider/animepahe/api.py | 6 +++--- fastanime/libs/anime_provider/hianime/api.py | 6 +++--- 3 files changed, 8 insertions(+), 7 deletions(-) diff --git a/fastanime/libs/anime_provider/allanime/constants.py b/fastanime/libs/anime_provider/allanime/constants.py index 9b6a063..080a5cf 100644 --- a/fastanime/libs/anime_provider/allanime/constants.py +++ b/fastanime/libs/anime_provider/allanime/constants.py @@ -1,3 +1,5 @@ +import re + SERVERS_AVAILABLE = [ "sharepoint", "dropbox", @@ -19,7 +21,6 @@ DEFAULT_PER_PAGE = 40 DEFAULT_PAGE = 1 # regex stuff -import re MP4_SERVER_JUICY_STREAM_REGEX = re.compile( r"video/mp4\",src:\"(https?://.*/video\.mp4)\"" diff --git a/fastanime/libs/anime_provider/animepahe/api.py b/fastanime/libs/anime_provider/animepahe/api.py index 3d7c7be..bb268d6 100644 --- a/fastanime/libs/anime_provider/animepahe/api.py +++ b/fastanime/libs/anime_provider/animepahe/api.py @@ -32,7 +32,7 @@ class AnimePahe(AnimeProvider): HEADERS = REQUEST_HEADERS @debug_provider - def search_for_anime(self, search_keywords: str, *args, **kwargs): + def search_for_anime(self, search_keywords: str, translation_type, **kwargs): response = self.session.get( ANIMEPAHE_ENDPOINT, params={"m": "search", "q": search_keywords} ) @@ -111,7 +111,7 @@ class AnimePahe(AnimeProvider): return data @debug_provider - def get_anime(self, session_id: str, *args): + def get_anime(self, session_id: str, **kwargs): page = 1 if d := self.store.get(str(session_id), "search_result"): anime_result: "AnimePaheSearchResult" = d @@ -211,7 +211,7 @@ class AnimePahe(AnimeProvider): @debug_provider def get_episode_streams( - self, anime_id, episode_number: str, translation_type, *args + self, anime_id, episode_number: str, translation_type, **kwargs ): anime_title = "" # extract episode details from memory diff --git a/fastanime/libs/anime_provider/hianime/api.py b/fastanime/libs/anime_provider/hianime/api.py index f206959..7d10bc9 100644 --- a/fastanime/libs/anime_provider/hianime/api.py +++ b/fastanime/libs/anime_provider/hianime/api.py @@ -43,7 +43,7 @@ class HiAnime(AnimeProvider): # HEADERS = {"Referer": "https://hianime.to/home"} @debug_provider - def search_for_anime(self, anime_title: str, *args): + def search_for_anime(self, anime_title: str, translation_type, **kwargs): query = quote_plus(anime_title) url = f"https://hianime.to/search?keyword={query}" response = self.session.get(url) @@ -91,7 +91,7 @@ class HiAnime(AnimeProvider): return {"pageInfo": {}, "results": results} @debug_provider - def get_anime(self, hianime_id, *args): + def get_anime(self, hianime_id, **kwargs): anime_result = {} if d := self.store.get(str(hianime_id), "search_result"): anime_result = d @@ -144,7 +144,7 @@ class HiAnime(AnimeProvider): } @debug_provider - def get_episode_streams(self, anime_id, episode, translation_type, *args): + def get_episode_streams(self, anime_id, episode, translation_type, **kwargs): if d := self.store.get(str(anime_id), "anime_info"): episodes_info = d episode_details = [