base

arr_api/services.py

@@ -0,0 +1,133 @@
+# arr_api/services.py
+import os
+import requests
+from datetime import datetime, timedelta, timezone
+from dateutil.parser import isoparse
+
+# ENV-Fallbacks
+ENV_SONARR_URL = os.getenv("SONARR_URL", "")
+ENV_SONARR_KEY = os.getenv("SONARR_API_KEY", "")
+ENV_RADARR_URL = os.getenv("RADARR_URL", "")
+ENV_RADARR_KEY = os.getenv("RADARR_API_KEY", "")
+DEFAULT_DAYS = int(os.getenv("ARR_DEFAULT_DAYS", "30"))
+
+class ArrServiceError(Exception):
+    pass
+
+def _get(url, headers, params=None, timeout=5):
+    try:
+        r = requests.get(url, headers=headers, params=params or {}, timeout=timeout)
+        r.raise_for_status()
+        return r.json()
+    except requests.exceptions.RequestException as e:
+        raise ArrServiceError(str(e))
+
+def _abs_url(base: str, p: str | None) -> str | None:
+    if not p:
+        return None
+    return f"{base.rstrip('/')}{p}" if p.startswith("/") else p
+
+def sonarr_calendar(days: int | None = None, base_url: str | None = None, api_key: str | None = None):
+    base = (base_url or ENV_SONARR_URL).strip()
+    key  = (api_key  or ENV_SONARR_KEY).strip()
+    if not base or not key:
+        return []
+    d = days or DEFAULT_DAYS
+    start = datetime.now(timezone.utc)
+    end = start + timedelta(days=d)
+
+    url = f"{base.rstrip('/')}/api/v3/calendar"
+    headers = {"X-Api-Key": key}
+    data = _get(url, headers, params={
+        "start": start.date().isoformat(),
+        "end": end.date().isoformat(),
+        "unmonitored": "false",
+        "includeSeries": "true",
+    })
+
+    out = []
+    for ep in data:
+        series = ep.get("series") or {}
+        # Poster finden
+        poster = None
+        for img in (series.get("images") or []):
+            if (img.get("coverType") or "").lower() == "poster":
+                poster = img.get("remoteUrl") or _abs_url(base, img.get("url"))
+                if poster:
+                    break
+
+        aired = isoparse(ep["airDateUtc"]).isoformat() if ep.get("airDateUtc") else None
+        out.append({
+            "seriesId": series.get("id"),
+            "seriesTitle": series.get("title"),
+            "seriesStatus": (series.get("status") or "").lower(),
+            "seriesPoster": poster,
+            "seriesOverview": series.get("overview") or "",
+            "seriesGenres": series.get("genres") or [],
+            "episodeId": ep.get("id"),
+            "seasonNumber": ep.get("seasonNumber"),
+            "episodeNumber": ep.get("episodeNumber"),
+            "title": ep.get("title"),
+            "airDateUtc": aired,
+            "tvdbId": series.get("tvdbId"),
+            "imdbId": series.get("imdbId"),
+            "network": series.get("network"),
+        })
+    return [x for x in out if x["seriesStatus"] == "continuing"]
+
+def radarr_calendar(days: int | None = None, base_url: str | None = None, api_key: str | None = None):
+    base = (base_url or ENV_RADARR_URL).strip()
+    key  = (api_key  or ENV_RADARR_KEY).strip()
+    if not base or not key:
+        return []
+    d = days or DEFAULT_DAYS
+    start = datetime.now(timezone.utc)
+    end = start + timedelta(days=d)
+
+    url = f"{base.rstrip('/')}/api/v3/calendar"
+    headers = {"X-Api-Key": key}
+    data = _get(url, headers, params={
+        "start": start.date().isoformat(),
+        "end": end.date().isoformat(),
+        "unmonitored": "false",
+        "includeMovie": "true",
+    })
+
+    out = []
+    for it in data:
+        movie = it.get("movie") or it
+        # Poster finden
+        poster = None
+        for img in (movie.get("images") or []):
+            if (img.get("coverType") or "").lower() == "poster":
+                poster = img.get("remoteUrl") or _abs_url(base, img.get("url"))
+                if poster:
+                    break
+
+        out.append({
+            "movieId": movie.get("id"),
+            "title": movie.get("title"),
+            "year": movie.get("year"),
+            "tmdbId": movie.get("tmdbId"),
+            "imdbId": movie.get("imdbId"),
+            "posterUrl": poster,
+            "overview": movie.get("overview") or "",
+            "inCinemas": movie.get("inCinemas"),
+            "physicalRelease": movie.get("physicalRelease"),
+            "digitalRelease": movie.get("digitalRelease"),
+            "hasFile": movie.get("hasFile"),
+            "isAvailable": movie.get("isAvailable"),
+        })
+
+    def is_upcoming(m):
+        for k in ("inCinemas", "physicalRelease", "digitalRelease"):
+            v = m.get(k)
+            if v:
+                try:
+                    if isoparse(v) > datetime.now(timezone.utc):
+                        return True
+                except Exception:
+                    pass
+        return False
+
+    return [m for m in out if is_upcoming(m)]