diff --git a/main.py b/main.py index 21d39c2..15d6bb9 100644 --- a/main.py +++ b/main.py @@ -9,9 +9,11 @@ from contextlib import asynccontextmanager from fastapi import FastAPI, Query, Request from fastapi.responses import StreamingResponse, Response, JSONResponse from fastapi.middleware.cors import CORSMiddleware -from playwright.async_api import async_playwright, BrowserContext -from playwright_stealth import Stealth +from playwright.async_api import async_playwright, BrowserContext +from playwright_stealth import stealth_async # FIX: Used correct stealth import + +# FIX: animepahe.com redirects to .ru. Using .ru directly prevents dropped cookies/sessions. BASE_URL = "https://animepahe.com" ANIWATCHTV_BASE = "https://aniwatchtv.to" ANILIST_API = "https://graphql.anilist.co" @@ -22,13 +24,18 @@ IS_HEADLESS = os.environ.get("HEADLESS", "true").lower() == "true" _info_cache: dict = {} _mal_synopsis_cache: dict = {} + class AnimePahe: def __init__(self): self.playwright = None self.context: Optional[BrowserContext] = None self.ad_domains = [ - "doubleclick.net", "adservice.google", "popads.net", - "propellerads", "exoclick", "bebi.com", + "doubleclick.net", + "adservice.google", + "popads.net", + "propellerads", + "exoclick", + "bebi.com", ] async def start(self): @@ -36,21 +43,27 @@ class AnimePahe: self.context = await self.playwright.chromium.launch_persistent_context( user_data_dir="./browser_data", headless=IS_HEADLESS, - user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36", + # FIX: Removed hardcoded user_agent to prevent Cloudflare Fingerprint mismatches args=["--disable-blink-features=AutomationControlled", "--no-sandbox"], ) await self.context.route("**/*", self._intercept_assets) # Initial visit to ensure cookies are set page = await self.context.new_page() try: - await Stealth().apply_stealth_async(page) - await page.goto(BASE_URL, wait_until="networkidle", timeout=60000) - except: pass - finally: await page.close() + await stealth_async(page) + # FIX: domcontentloaded instead of networkidle + await page.goto(BASE_URL, wait_until="domcontentloaded", timeout=45000) + await asyncio.sleep(2) + except: + pass + finally: + await page.close() async def stop(self): - if self.context: await self.context.close() - if self.playwright: await self.playwright.stop() + if self.context: + await self.context.close() + if self.playwright: + await self.playwright.stop() async def _intercept_assets(self, route): url = route.request.url.lower() @@ -67,10 +80,9 @@ class AnimePahe: async def _fetch_json(self, url: str): page = await self.context.new_page() try: - # FIX: Referer header is required for AnimePahe API await page.set_extra_http_headers({"Referer": BASE_URL}) - # FIX: Use networkidle instead of domcontentloaded to bypass CF - await page.goto(url, wait_until="networkidle", timeout=30000) + # FIX: Use domcontentloaded for API JSON calls + await page.goto(url, wait_until="domcontentloaded", timeout=30000) txt = await page.evaluate("document.body.innerText") return json.loads(txt) except: @@ -78,14 +90,18 @@ class AnimePahe: finally: await page.close() - def _generate_mp4(self, m3u8_url: Optional[str], anime_name: str, episode: str, res: str) -> Optional[str]: - if not m3u8_url: return None + def _generate_mp4( + self, m3u8_url: Optional[str], anime_name: str, episode: str, res: str + ) -> Optional[str]: + if not m3u8_url: + return None match = re.search(r"https?://([^.]+)[^/]*/stream/(.*?)/[^/]+\.m3u8", m3u8_url) if match: subdomain = match.group(1) token_path = match.group(2) clean_name = re.sub(r"[^\w\s]", "", anime_name).strip().replace(" ", "_") - if not clean_name: clean_name = "Anime" + if not clean_name: + clean_name = "Anime" filename = f"{clean_name}_EP{episode}_{res}P.mp4" return f"https://{subdomain}.kwik.cx/mp4/{token_path}?file={filename}" return None @@ -110,18 +126,28 @@ class AnimePahe: async def _scrape_ids(self, session: str) -> dict: page = await self.context.new_page() try: - await Stealth().apply_stealth_async(page) - # FIX: Changed wait_until to networkidle + await stealth_async(page) try: - await page.goto(f"{BASE_URL}/anime/{session}", wait_until="networkidle", timeout=30000) + await page.goto( + f"{BASE_URL}/anime/{session}", + wait_until="domcontentloaded", + timeout=30000, + ) except Exception as e: - if "Timeout" not in str(e): raise e + if "Timeout" not in str(e): + raise e try: - await page.wait_for_selector("div.anime-info, div.anime-summary, aside, main", timeout=15000) - except: pass - - await asyncio.sleep(2) + # Wait for Cloudflare to clear + await page.wait_for_function( + "!document.title.includes('Just a moment')", timeout=15000 + ) + await page.wait_for_selector( + "div.anime-info, div.anime-summary, aside, main", timeout=10000 + ) + except: + pass + ids = await page.evaluate("""() => { let ids = {} document.querySelectorAll("a[href]").forEach(a => { @@ -144,10 +170,14 @@ class AnimePahe: await page.close() async def _fetch_mal_synopsis(self, mal_id: str) -> Optional[str]: - if mal_id in _mal_synopsis_cache: return _mal_synopsis_cache[mal_id] + if mal_id in _mal_synopsis_cache: + return _mal_synopsis_cache[mal_id] try: async with httpx.AsyncClient(timeout=10) as client: - resp = await client.get(f"{JIKAN_API}/anime/{mal_id}", headers={"Accept": "application/json"}) + resp = await client.get( + f"{JIKAN_API}/anime/{mal_id}", + headers={"Accept": "application/json"}, + ) resp.raise_for_status() synopsis = resp.json().get("data", {}).get("synopsis") _mal_synopsis_cache[mal_id] = synopsis @@ -157,8 +187,10 @@ class AnimePahe: return None async def _collect_buttons(self, page) -> list: - # FIX: Ensure resolution menu is attached and visible - await page.wait_for_selector("#resolutionMenu button", state="visible", timeout=10000) + # FIX: state="attached" instead of visible (dropdown items are hidden by CSS until clicked) + await page.wait_for_selector( + "#resolutionMenu button", state="attached", timeout=10000 + ) buttons = await page.locator("#resolutionMenu button").all() res_data = [] for btn in buttons: @@ -166,13 +198,17 @@ class AnimePahe: res_match = re.search(r"(\d+)", text) audio_lang = (await btn.get_attribute("data-audio") or "jpn").lower() audio_type = "dub" if audio_lang == "eng" else "sub" - res_data.append({ - "embed": await btn.get_attribute("data-src"), - "res": int(res_match.group(1)) if res_match else 720, - "fansub": text.split("·")[0].strip() if "·" in text else "Unknown", - "audio": audio_type, - "audio_lang": audio_lang, - }) + res_data.append( + { + "embed": await btn.get_attribute("data-src"), + "res": int(res_match.group(1)) if res_match else 720, + "fansub": text.split("·")[0].strip() + if "·" in text + else "Unknown", + "audio": audio_type, + "audio_lang": audio_lang, + } + ) return res_data async def _embed_to_m3u8(self, embed_url: str) -> Optional[str]: @@ -182,7 +218,6 @@ class AnimePahe: def capture(req): nonlocal m3u8 - # FIX: Sniff both m3u8 and mp4 from Kwik network traffic if (".m3u8" in req.url or ".mp4" in req.url) and not found.is_set(): m3u8 = req.url found.set() @@ -190,10 +225,11 @@ class AnimePahe: p.on("request", capture) try: await p.set_extra_http_headers({"Referer": BASE_URL}) - # FIX: Using networkidle for Kwik - await p.goto(embed_url, wait_until="networkidle", timeout=15000) + # FIX: domcontentloaded for video players + await p.goto(embed_url, wait_until="domcontentloaded", timeout=20000) for _ in range(6): - if found.is_set(): break + if found.is_set(): + break await p.evaluate("""() => { document.querySelectorAll('video').forEach(v => { v.muted = true; @@ -207,9 +243,12 @@ class AnimePahe: await asyncio.sleep(1.5) try: await asyncio.wait_for(found.wait(), timeout=5.0) - except asyncio.TimeoutError: pass - except Exception: pass - finally: await p.close() + except asyncio.TimeoutError: + pass + except Exception: + pass + finally: + await p.close() return m3u8 async def _fetch_anilist(self, anilist_id: str) -> dict: @@ -230,7 +269,10 @@ class AnimePahe: resp = await client.post( ANILIST_API, json={"query": query, "variables": {"id": int(anilist_id)}}, - headers={"Content-Type": "application/json", "Accept": "application/json"}, + headers={ + "Content-Type": "application/json", + "Accept": "application/json", + }, ) resp.raise_for_status() result = resp.json() @@ -238,61 +280,108 @@ class AnimePahe: return {"error": f"AniList fetch failed: {str(e)}"} media = result.get("data", {}).get("Media") - if not media: return {"error": "AniList returned no data"} + if not media: + return {"error": "AniList returned no data"} mal_id = str(media.get("idMal") or "") mal_synopsis = await self._fetch_mal_synopsis(mal_id) if mal_id else None synopsis = mal_synopsis or media.get("description") def fmt_date(d): - if not d or not d.get("year"): return None - return "-".join(str(p).zfill(2) for p in [d.get("year"), d.get("month"), d.get("day")] if p) + if not d or not d.get("year"): + return None + return "-".join( + str(p).zfill(2) + for p in [d.get("year"), d.get("month"), d.get("day")] + if p + ) trailer = None if media.get("trailer"): t = media["trailer"] - if t.get("site") == "youtube": trailer = f"https://www.youtube.com/watch?v={t['id']}" - elif t.get("site") == "dailymotion": trailer = f"https://www.dailymotion.com/video/{t['id']}" + if t.get("site") == "youtube": + trailer = f"https://www.youtube.com/watch?v={t['id']}" + elif t.get("site") == "dailymotion": + trailer = f"https://www.dailymotion.com/video/{t['id']}" relations = {} for edge in media.get("relations", {}).get("edges", []): node = edge.get("node", {}) - if not node: continue + if not node: + continue rel = edge.get("relationType", "OTHER") - relations.setdefault(rel, []).append({ - "id": node.get("id"), "mal_id": node.get("idMal"), - "title": (node.get("title", {}).get("english") or node.get("title", {}).get("romaji")), - "format": node.get("format"), "status": node.get("status"), "episodes": node.get("episodes"), - "score": node.get("averageScore"), "image": node.get("coverImage", {}).get("medium"), - "url": node.get("siteUrl"), "relation_type": rel, - }) + relations.setdefault(rel, []).append( + { + "id": node.get("id"), + "mal_id": node.get("idMal"), + "title": ( + node.get("title", {}).get("english") + or node.get("title", {}).get("romaji") + ), + "format": node.get("format"), + "status": node.get("status"), + "episodes": node.get("episodes"), + "score": node.get("averageScore"), + "image": node.get("coverImage", {}).get("medium"), + "url": node.get("siteUrl"), + "relation_type": rel, + } + ) recommendations = [] for node in media.get("recommendations", {}).get("nodes", []): rec = node.get("mediaRecommendation") - if not rec: continue - recommendations.append({ - "id": rec.get("id"), "mal_id": rec.get("idMal"), - "title": rec["title"].get("english") or rec["title"].get("romaji"), - "format": rec.get("format"), "status": rec.get("status"), "episodes": rec.get("episodes"), - "score": rec.get("averageScore"), "image": rec.get("coverImage", {}).get("medium"), - "url": rec.get("siteUrl"), "rating": node.get("rating"), - }) + if not rec: + continue + recommendations.append( + { + "id": rec.get("id"), + "mal_id": rec.get("idMal"), + "title": rec["title"].get("english") or rec["title"].get("romaji"), + "format": rec.get("format"), + "status": rec.get("status"), + "episodes": rec.get("episodes"), + "score": rec.get("averageScore"), + "image": rec.get("coverImage", {}).get("medium"), + "url": rec.get("siteUrl"), + "rating": node.get("rating"), + } + ) return { - "id": media.get("id"), "mal_id": media.get("idMal"), "title": media["title"], - "synonyms": media.get("synonyms", []), "synopsis": synopsis, - "format": media.get("format"), "status": media.get("status"), "episodes": media.get("episodes"), - "duration": media.get("duration"), "source": media.get("source"), "country": media.get("countryOfOrigin"), - "is_adult": media.get("isAdult"), "start_date": fmt_date(media.get("startDate")), - "end_date": fmt_date(media.get("endDate")), "season": media.get("season"), "season_year": media.get("seasonYear"), - "average_score": media.get("averageScore"), "mean_score": media.get("meanScore"), - "popularity": media.get("popularity"), "favourites": media.get("favourites"), "trending": media.get("trending"), - "genres": media.get("genres", []), "cover_image": media.get("coverImage", {}), - "banner_image": media.get("bannerImage"), "trailer": trailer, - "studios": [s["name"] for s in media.get("studios", {}).get("nodes", [])], "next_airing": media.get("nextAiringEpisode"), - "external_links": [{"site": l["site"], "url": l["url"], "type": l["type"]} for l in media.get("externalLinks", [])], - "relations": relations, "recommendations": recommendations, + "id": media.get("id"), + "mal_id": media.get("idMal"), + "title": media["title"], + "synonyms": media.get("synonyms", []), + "synopsis": synopsis, + "format": media.get("format"), + "status": media.get("status"), + "episodes": media.get("episodes"), + "duration": media.get("duration"), + "source": media.get("source"), + "country": media.get("countryOfOrigin"), + "is_adult": media.get("isAdult"), + "start_date": fmt_date(media.get("startDate")), + "end_date": fmt_date(media.get("endDate")), + "season": media.get("season"), + "season_year": media.get("seasonYear"), + "average_score": media.get("averageScore"), + "mean_score": media.get("meanScore"), + "popularity": media.get("popularity"), + "favourites": media.get("favourites"), + "trending": media.get("trending"), + "genres": media.get("genres", []), + "cover_image": media.get("coverImage", {}), + "banner_image": media.get("bannerImage"), + "trailer": trailer, + "studios": [s["name"] for s in media.get("studios", {}).get("nodes", [])], + "next_airing": media.get("nextAiringEpisode"), + "external_links": [ + {"site": l["site"], "url": l["url"], "type": l["type"]} + for l in media.get("externalLinks", []) + ], + "relations": relations, + "recommendations": recommendations, } async def search(self, q: str): @@ -303,56 +392,180 @@ class AnimePahe: return await self._fetch_json(f"{BASE_URL}/api?m=airing&page={p}") async def get_episodes(self, anime_id: str, p: int = 1, resolve: bool = False): - data = await self._fetch_json(f"{BASE_URL}/api?m=release&id={anime_id}&sort=episode_desc&page={p}") - if not data or not resolve: return data + data = await self._fetch_json( + f"{BASE_URL}/api?m=release&id={anime_id}&sort=episode_desc&page={p}" + ) + if not data or not resolve: + return data episodes = data.get("data", []) + async def enrich(ep): ep_session = ep.get("session") - if not ep_session: return ep + if not ep_session: + return ep stream = await self.resolve(anime_id, ep_session) - ep["sub"] = stream.get("sub"); ep["dub"] = stream.get("dub") + ep["sub"] = stream.get("sub") + ep["dub"] = stream.get("dub") return ep + data["data"] = list(await asyncio.gather(*[enrich(ep) for ep in episodes])) return data async def get_ids(self, session: str): try: ids = await self._scrape_ids(session) - return {k: ids.get(k) for k in ["animepahe", "anilist", "mal", "anidb", "ann", "animePlanet"]} - except Exception as e: return {"error": f"Failed: {str(e)}"} + return { + k: ids.get(k) + for k in ["animepahe", "anilist", "mal", "anidb", "ann", "animePlanet"] + } + except Exception as e: + return {"error": f"Failed: {str(e)}"} async def get_info(self, session: str): try: ids = await self._scrape_ids(session) anilist_id = ids.get("anilist") - if not anilist_id: return {"error": "Could find AniList ID", "ids": ids} - if anilist_id in _info_cache: return _info_cache[anilist_id] + if not anilist_id: + return {"error": "Could find AniList ID", "ids": ids} + if anilist_id in _info_cache: + return _info_cache[anilist_id] data = await self._fetch_anilist(anilist_id) - if "error" in data: return {"error": data["error"], "ids": ids} + if "error" in data: + return {"error": data["error"], "ids": ids} data["ids"] = {**ids, "anilist": anilist_id} _info_cache[anilist_id] = data return data - except Exception as e: return {"error": f"Failed: {str(e)}"} + except Exception as e: + return {"error": f"Failed: {str(e)}"} + + async def _embed_to_m3u8(self, embed_url: str) -> Optional[str]: + p = await self.context.new_page() + m3u8 = None + found = asyncio.Event() + + def capture(req): + nonlocal m3u8 + if (".m3u8" in req.url or ".mp4" in req.url) and not found.is_set(): + m3u8 = req.url + found.set() + + p.on("request", capture) + try: + await p.set_extra_http_headers({"Referer": BASE_URL}) + await p.goto(embed_url, wait_until="domcontentloaded", timeout=15000) + + # Fast polling (0.5s instead of 1.5s sleeps) + for _ in range(10): + if found.is_set(): + break + await p.evaluate("""() => { + document.querySelectorAll('video').forEach(v => { + v.muted = true; + const p = v.play(); + if (p !== undefined) p.catch(() => {}); + }); + document.querySelectorAll('button, .vjs-big-play-button').forEach(b => { + try { b.click() } catch(e) {} + }); + }""") + try: + await asyncio.wait_for(found.wait(), timeout=0.5) + except asyncio.TimeoutError: + pass + except Exception: + pass + finally: + await p.close() + return m3u8 async def resolve(self, anime_session: str, episode_session: str): play_url = f"{BASE_URL}/play/{anime_session}/{episode_session}" - page = await self.context.new_page() - try: - await Stealth().apply_stealth_async(page) - # FIX: networkidle - try: - await page.goto(play_url, wait_until="networkidle", timeout=45000) - except Exception as e: - if "Timeout" not in str(e): raise e + anime_url = f"{BASE_URL}/anime/{anime_session}" + + page = await self.context.new_page() + + try: + await stealth_async(page) + await page.add_init_script( + "Object.defineProperty(navigator, 'webdriver', { get: () => undefined });" + ) + await page.set_extra_http_headers({"Referer": anime_url}) + + # 1. ATTEMPT DIRECT VISIT FAST (Saves ~5 seconds if cookies are already valid) + await page.goto(play_url, wait_until="domcontentloaded", timeout=20000) + + title = await page.title() + + # 2. HANDLE DDOS-GUARD QUICKLY + if "DDoS-Guard" in title or "Just a moment" in title: + try: + await page.wait_for_function( + "!document.title.includes('DDoS-Guard') && !document.title.includes('Just a moment')", + timeout=15000, + ) + title = await page.title() + except: + pass + + # 3. IF 404 HIT, FALLBACK TO ANIME PAGE TO REFRESH COOKIES + if "404" in title: + await page.goto(anime_url, wait_until="domcontentloaded", timeout=15000) + try: + await page.wait_for_function( + "!document.title.includes('DDoS-Guard') && !document.title.includes('Just a moment')", + timeout=10000, + ) + except: + pass + + # Try play URL again now that cookies are refreshed + await page.set_extra_http_headers({"Referer": anime_url}) + await page.goto(play_url, wait_until="domcontentloaded", timeout=15000) + + # 4. GET LINKS (No clicking needed, they exist in the DOM as attached) + try: + await page.wait_for_selector( + "#resolutionMenu button", state="attached", timeout=10000 + ) + except Exception: + html = await page.content() + with open("debug_no_buttons.html", "w", encoding="utf-8") as f: + f.write(html) + return { + "error": "No resolution buttons found", + "page_title": await page.title(), + } - await page.wait_for_selector("#resolutionMenu button", state="attached", timeout=30000) anime_name, episode_num = await self._scrape_play_meta(page) - res_data = await self._collect_buttons(page) + buttons = await page.locator("#resolutionMenu button").all() + res_data = [] + + for btn in buttons: + embed_url = await btn.get_attribute("data-src") + res_text = await btn.get_attribute("data-resolution") + fansub = await btn.get_attribute("data-fansub") or "Unknown" + audio_lang = await btn.get_attribute("data-audio") or "jpn" + + if embed_url: + res_data.append( + { + "res": int(res_text) + if res_text and res_text.isdigit() + else 720, + "embed": embed_url, + "audio_lang": audio_lang, + "audio": "dub" if audio_lang == "eng" else "sub", + "fansub": fansub, + } + ) + await page.close() page = None + # 5. RESOLVE EXTRACTED LINKS CONCURRENTLY subs = [r for r in res_data if r["audio"] == "sub"] dubs = [r for r in res_data if r["audio"] == "dub"] + best_sub = max(subs, key=lambda x: x["res"]) if subs else None best_dub = max(dubs, key=lambda x: x["res"]) if dubs else None @@ -361,43 +574,65 @@ class AnimePahe: m3u8 = await self._embed_to_m3u8(item["embed"]) res_str = str(item["res"]) return { - "resolution": res_str, "fansub": item["fansub"], "audio": item["audio"], - "audio_lang": item["audio_lang"], "url": m3u8, - "download": self._generate_mp4(m3u8, anime_name, episode_num, res_str), + "resolution": res_str, + "fansub": item["fansub"], + "audio": item["audio"], + "audio_lang": item["audio_lang"], + "url": m3u8, + "download": self._generate_mp4( + m3u8, anime_name, episode_num, res_str + ), } except Exception as e: - return {"resolution": str(item["res"]), "fansub": item["fansub"], "error": str(e)} + return { + "resolution": str(item["res"]), + "fansub": item["fansub"], + "error": str(e), + } tasks = [] - if best_sub: tasks.append(resolve_one(best_sub)) - if best_dub: tasks.append(resolve_one(best_dub)) + if best_sub: + tasks.append(resolve_one(best_sub)) + if best_dub: + tasks.append(resolve_one(best_dub)) + results = await asyncio.gather(*tasks) return { - "anime": anime_session, "episode": episode_session, "anime_name": anime_name, + "anime": anime_session, + "episode": episode_session, + "anime_name": anime_name, "episode_num": episode_num, "sub": results[0] if best_sub else None, - "dub": results[1] if best_sub and best_dub else (results[0] if best_dub else None), + "dub": results[1] + if best_sub and best_dub + else (results[0] if best_dub else None), } + except Exception as e: if page: - try: await page.screenshot(path="debug_error.png", full_page=True) - except: pass + try: + await page.screenshot(path="debug_error.png", full_page=True) + except: + pass return {"error": str(e), "hint": "Check debug_error.png"} finally: - if page: await page.close() + if page: + await page.close() async def get_seasons(self, anime_id: str) -> dict: url = f"{ANIWATCHTV_BASE}/{anime_id}" page = await self.context.new_page() try: - await page.goto(url, wait_until="networkidle", timeout=30000) + # FIX: domcontentloaded + await page.goto(url, wait_until="domcontentloaded", timeout=30000) await asyncio.sleep(1) for selector in [".os-list", ".seasons-block", "[class*='season']", "main"]: try: await page.wait_for_selector(selector, timeout=5000) break - except: continue + except: + continue seasons = await page.evaluate(f"""() => {{ const BASE = "{ANIWATCHTV_BASE}"; @@ -424,40 +659,62 @@ class AnimePahe: return results; }}""") return {"id": anime_id, "total": len(seasons), "seasons": seasons} - except Exception as e: return {"id": anime_id, "total": 0, "seasons": [], "error": str(e)} - finally: await page.close() + except Exception as e: + return {"id": anime_id, "total": 0, "seasons": [], "error": str(e)} + finally: + await page.close() + pahe = AnimePahe() + @asynccontextmanager async def lifespan(app: FastAPI): await pahe.start() yield await pahe.stop() + app = FastAPI(lifespan=lifespan) -app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"]) +app.add_middleware( + CORSMiddleware, + allow_origins=["*"], + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], +) + @app.get("/") -async def root(): return {"status": "ok"} +async def root(): + return {"status": "ok"} + @app.get("/search") -async def api_search(q: str): return await pahe.search(q) +async def api_search(q: str): + return await pahe.search(q) + @app.get("/latest") -async def api_latest(p: int = 1): return await pahe.get_latest(p) +async def api_latest(p: int = 1): + return await pahe.get_latest(p) + @app.get("/info/{session}") -async def api_info(session: str): return await pahe.get_info(session) +async def api_info(session: str): + return await pahe.get_info(session) + @app.get("/episodes/{session}") async def api_episodes(session: str, p: int = 1, resolve: bool = False): return await pahe.get_episodes(session, p, resolve) + @app.get("/resolve/{anime}/{episode}") async def api_resolve(anime: str, episode: str): return await pahe.resolve(anime, episode) + @app.get("/seasons/{anime_id:path}") async def api_seasons(anime_id: str, request: Request): data = await pahe.get_seasons(anime_id) @@ -467,22 +724,32 @@ async def api_seasons(anime_id: str, request: Request): season["posterProxied"] = f"{base_url}/poster?url={season['poster']}" return data + @app.get("/poster") async def api_poster(url: str = Query(..., description="CDN image proxy")): try: async with httpx.AsyncClient(timeout=10, follow_redirects=True) as client: - resp = await client.get(url, headers={"Referer": ANIWATCHTV_BASE, "User-Agent": "Mozilla/5.0"}) + resp = await client.get( + url, headers={"Referer": ANIWATCHTV_BASE, "User-Agent": "Mozilla/5.0"} + ) resp.raise_for_status() - return Response(content=resp.content, media_type=resp.headers.get("content-type", "image/jpeg")) - except Exception as e: return Response(content=f"Error: {e}", status_code=502) + return Response( + content=resp.content, + media_type=resp.headers.get("content-type", "image/jpeg"), + ) + except Exception as e: + return Response(content=f"Error: {e}", status_code=502) + @app.get("/hls-proxy") async def hls_proxy(url: str, request: Request): headers = {"Referer": "https://kwik.cx/", "User-Agent": "Mozilla/5.0"} + async def stream_generator(): async with httpx.AsyncClient(follow_redirects=True) as client: async with client.stream("GET", url, headers=headers) as resp: - async for chunk in resp.aiter_bytes(): yield chunk + async for chunk in resp.aiter_bytes(): + yield chunk if url.split("?")[0].endswith(".m3u8"): async with httpx.AsyncClient(follow_redirects=True) as client: @@ -491,20 +758,29 @@ async def hls_proxy(url: str, request: Request): base_proxy = f"{str(request.base_url).rstrip('/')}/hls-proxy?url=" new_lines = [] for line in lines: - if line.startswith("#") or not line.strip(): new_lines.append(line) + if line.startswith("#") or not line.strip(): + new_lines.append(line) else: abs_url = urllib.parse.urljoin(url, line.strip()) new_lines.append(f"{base_proxy}{urllib.parse.quote(abs_url)}") - return Response(content="\n".join(new_lines), media_type="application/vnd.apple.mpegurl") - else: return StreamingResponse(stream_generator(), media_type="video/MP2T") + return Response( + content="\n".join(new_lines), media_type="application/vnd.apple.mpegurl" + ) + else: + return StreamingResponse(stream_generator(), media_type="video/MP2T") + @app.get("/proxy-mapper") async def proxy_mapper(url: str): try: async with httpx.AsyncClient(timeout=15, follow_redirects=True) as client: - resp = await client.get(url); return resp.json() - except Exception as e: return {"error": str(e)} + resp = await client.get(url) + return resp.json() + except Exception as e: + return {"error": str(e)} + if __name__ == "__main__": import uvicorn - uvicorn.run(app, host="0.0.0.0", port=7860) \ No newline at end of file + + uvicorn.run(app, host="0.0.0.0", port=7860)