gitea-mirror/FastAnime
1
0
Fork 0
mirror of https://github.com/Benexl/FastAnime.git synced 2025-12-24 20:10:11 -08:00
Code Issues Packages Projects Releases Wiki Activity

feat:add basic error handling to allanime provider

Browse Source
This commit is contained in:
Benex254
2024-08-05 09:46:58 +03:00
parent ce3ee491e1
commit cc228cc38a
1 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
fastanime/libs/anime_provider/allanime/api.py
View File

@@ -29,17 +29,21 @@ class AllAnimeAPI:
api_endpoint = ALLANIME_API_ENDPOINT
def _fetch_gql(self, query: str, variables: dict) -> dict:
response = requests.get(
self.api_endpoint,
params={
"variables": json.dumps(variables),
"query": query,
},
headers={"Referer": ALLANIME_REFERER, "User-Agent": USER_AGENT},
)
if response.status_code != 200:
try:
response = requests.get(
self.api_endpoint,
params={
"variables": json.dumps(variables),
"query": query,
},
headers={"Referer": ALLANIME_REFERER, "User-Agent": USER_AGENT},
)
if response.status_code != 200:
return {}
return response.json().get("data", {})
except Exception as e:
Logger.error(f"allanime:Error: {e}")
return {}
return response.json().get("data", {})
def search_for_anime(
self, user_query: str, translation_type: str = "sub"
@@ -73,6 +77,11 @@ class AllAnimeAPI:
return anime_provider._fetch_gql(ALLANIME_EPISODES_GQL, variables)
def get_episode_streams(self, allanime_episode_embeds_data):
if (
not allanime_episode_embeds_data
or allanime_episode_embeds_data.get("episode") is None
):
return {}
embeds = allanime_episode_embeds_data["episode"]["sourceUrls"]
for embed in embeds:
# filter the working streams