Update app/ht_noti.py

app/ht_noti.py

@@ -4,11 +4,8 @@ from datetime import datetime, timedelta
 from typing import Dict, List, Optional
 from fastapi import HTTPException
 
-
 class HentaiApp:
-
     def __init__(self):
-
         self.base_url = "https://hentaiz3.com/watch"
 
         self.api_url = (
@@ -221,6 +218,102 @@ class HentaiApp:
                 detail=f"Parse error: {str(e)}"
             )
 
+    async def fetch_info(self, slug: str):
+
+        url = f"https://hentaiz1.com/watch/{slug}/__data.json?x-sveltekit-invalidated=001"
+
+        headers = {
+            "User-Agent": "Mozilla/5.0"
+        }
+
+        res = self.scraper.get(url, headers=headers, timeout=15)
+
+        if res.status_code != 200:
+            raise HTTPException(status_code=404, detail="Episode not found")
+
+        raw = res.json()
+
+        data_pool = raw["nodes"][2]["data"]
+
+        info = data_pool[1]
+
+        title = data_pool[info["title"]]
+        description = data_pool[info["description"]]
+
+        episode = info.get("episodeNumber")
+        year = info.get("releaseYear")
+        views = info.get("viewsTotal")
+
+        embed_url = data_pool[info["embedUrl"]] if isinstance(info["embedUrl"], int) else info["embedUrl"]
+
+        published = ""
+
+        if isinstance(info["publishedAt"], list):
+            published = info["publishedAt"][1]
+
+        # genres
+        genres = []
+
+        genres_ref = info.get("genres")
+
+        if isinstance(genres_ref, list):
+
+            for g_idx in genres_ref:
+
+                g_obj = data_pool[g_idx]
+
+                if isinstance(g_obj, dict):
+
+                    genre_idx = g_obj.get("genre")
+
+                    if isinstance(genre_idx, int):
+
+                        genre_obj = data_pool[genre_idx]
+
+                        name_idx = genre_obj.get("name")
+
+                        if isinstance(name_idx, int):
+                            genres.append(data_pool[name_idx])
+
+
+        # studios
+        studios = []
+
+        studios_ref = info.get("studios")
+
+        if isinstance(studios_ref, list):
+
+            for s_idx in studios_ref:
+
+                s_obj = data_pool[s_idx]
+
+                if isinstance(s_obj, dict):
+
+                    studio_idx = s_obj.get("studio")
+
+                    if isinstance(studio_idx, int):
+
+                        studio_obj = data_pool[studio_idx]
+
+                        name_idx = studio_obj.get("name")
+
+                        if isinstance(name_idx, int):
+                            studios.append(data_pool[name_idx])
+
+
+        return {
+            "ok": True,
+            "title": title,
+            "slug": slug,
+            "episode": episode,
+            "year": year,
+            "views": views,
+            "published_at": published,
+            "description": description,
+            "embed_url": embed_url,
+            "genres": genres,
+            "studios": studios
+        }
 
     async def get_new(self, apikey: Optional[str], system_key: Optional[str]):