cinnamon-cli 0.2.39__tar.gz → 0.3.0__tar.gz

{cinnamon_cli-0.2.39 → cinnamon_cli-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cinnamon-cli
-Version: 0.2.39
+Version: 0.3.0
 Summary: Watch Movies, TV shows and anime from your terminal.
 Author: pizza-droid
 License: CC BY-NC 4.0

cinnamon_cli-0.3.0/cinnamon/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.0"

cinnamon_cli-0.3.0/cinnamon/anilist.py

@@ -0,0 +1,196 @@
+import time as _time
+
+import requests
+
+from .errors import CinnamonError
+
+API = "https://graphql.anilist.co"
+UA = "cinnamon/0.1.0"
+
+_SEARCH_QUERY = """
+query ($search: String, $page: Int) {
+  Page(page: $page, perPage: 20) {
+    media(search: $search, type: ANIME) {
+      id
+      idMal
+      title { romaji english native }
+      episodes
+      status
+      format
+      startDate { year }
+      genres
+    }
+  }
+}
+"""
+
+
+_RELATION_QUERY = """
+query ($search: String) {
+  Page(page: 1, perPage: 5) {
+    media(search: $search, type: ANIME) {
+      id
+      title { romaji english }
+      relations {
+        edges {
+          relationType
+          node {
+            id
+            title { romaji english }
+            episodes
+            status
+            format
+          }
+        }
+      }
+    }
+  }
+}
+"""
+
+
+def search_anime(query):
+    try:
+        resp = requests.post(
+            API,
+            json={"query": _SEARCH_QUERY, "variables": {"search": query, "page": 1}},
+            timeout=15,
+            headers={"User-Agent": UA, "Content-Type": "application/json", "Accept": "application/json"},
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        return data.get("data", {}).get("Page", {}).get("media", [])
+    except requests.ConnectionError:
+        raise CinnamonError("Could not reach AniList. Check your internet connection.")
+    except requests.Timeout:
+        raise CinnamonError("AniList request timed out.")
+    except requests.HTTPError as e:
+        raise CinnamonError(f"AniList returned an error (HTTP {e.response.status_code}).")
+
+
+def find_sequel(anime_name):
+    """Search AniList for an anime and return its first SEQUEL relation, or None."""
+    try:
+        resp = requests.post(
+            API,
+            json={"query": _RELATION_QUERY, "variables": {"search": anime_name}},
+            timeout=15,
+            headers={"User-Agent": UA, "Content-Type": "application/json", "Accept": "application/json"},
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        media_list = data.get("data", {}).get("Page", {}).get("media", [])
+        for m in media_list:
+            edges = m.get("relations", {}).get("edges", [])
+            for e in edges:
+                if e.get("relationType") == "SEQUEL":
+                    node = e.get("node", {})
+                    title = node.get("title", {})
+                    name = title.get("romaji") or title.get("english") or ""
+                    if name:
+                        return {"id": node["id"], "name": name}
+        return None
+    except (requests.RequestException, CinnamonError):
+        return None
+
+
+_SCHEDULE_QUERY = """
+query ($page: Int, $perPage: Int, $airingAtGreater: Int, $airingAtLesser: Int) {
+  Page(page: $page, perPage: $perPage) {
+    airingSchedules(
+      airingAt_greater: $airingAtGreater
+      airingAt_lesser: $airingAtLesser
+      sort: TIME
+    ) {
+      id
+      airingAt
+      episode
+      media {
+        id
+        title { romaji english }
+        format
+        status
+        episodes
+        popularity
+        averageScore
+      }
+    }
+  }
+}
+"""
+
+
+_MEDIA_AIRING_QUERY = """
+query ($mediaId: Int) {
+  Page(page: 1, perPage: 100) {
+    airingSchedules(mediaId: $mediaId, sort: EPISODE) {
+      airingAt
+      episode
+      timeUntilAiring
+    }
+  }
+}
+"""
+
+
+def get_media_airing(media_id):
+    """Return list of {airingAt, episode, timeUntilAiring} for a given media ID."""
+    try:
+        resp = requests.post(
+            API,
+            json={"query": _MEDIA_AIRING_QUERY, "variables": {"mediaId": media_id}},
+            timeout=15,
+            headers={"User-Agent": UA, "Content-Type": "application/json", "Accept": "application/json"},
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        return data.get("data", {}).get("Page", {}).get("airingSchedules", [])
+    except requests.ConnectionError:
+        return []
+    except requests.Timeout:
+        return []
+    except requests.HTTPError:
+        return []
+
+
+def get_schedule(days=7):
+    now = int(_time.time())
+    end = now + (days * 86400)
+    try:
+        resp = requests.post(
+            API,
+            json={
+                "query": _SCHEDULE_QUERY,
+                "variables": {
+                    "page": 1,
+                    "perPage": 50,
+                    "airingAtGreater": now,
+                    "airingAtLesser": end,
+                },
+            },
+            timeout=15,
+            headers={"User-Agent": UA, "Content-Type": "application/json", "Accept": "application/json"},
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        schedules = data.get("data", {}).get("Page", {}).get("airingSchedules", [])
+        results = []
+        for s in schedules:
+            media = s.get("media", {})
+            title = media.get("title", {}) or {}
+            results.append({
+                "airing_at": s.get("airingAt", 0),
+                "episode": s.get("episode"),
+                "title_romaji": title.get("romaji", "") or "",
+                "title_english": title.get("english", "") or "",
+                "format": media.get("format", ""),
+                "popularity": media.get("popularity", 0) or 0,
+                "score": media.get("averageScore", 0) or 0,
+            })
+        return results
+    except requests.ConnectionError:
+        raise CinnamonError("Could not reach AniList. Check your internet connection.")
+    except requests.Timeout:
+        raise CinnamonError("AniList request timed out.")
+    except requests.HTTPError as e:
+        raise CinnamonError(f"AniList returned an error (HTTP {e.response.status_code}).")