refactor: conform all provider functions

This commit is contained in:
benex
2024-12-03 12:29:47 +03:00
parent acd937f8ab
commit 787eb0c9ca
3 changed files with 8 additions and 7 deletions

View File

@@ -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)\""

View File

@@ -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

View File

@@ -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 = [