kino-app/backend/sonarr.py

"""Sonarr v3 API client. https://sonarr.tv/docs/api/"""
import requests
from typing import List, Dict


def _h(api_key: str) -> dict:
    return {"X-Api-Key": api_key, "Content-Type": "application/json"}


def _poster(images):
    for img in images or []:
        if img.get("coverType") == "poster":
            return img.get("remoteUrl") or img.get("url") or ""
    return ""


def test_connection(base_url: str, api_key: str) -> bool:
    try:
        r = requests.get(f"{base_url.rstrip('/')}/api/v3/system/status", headers=_h(api_key), timeout=10)
        return r.ok
    except Exception:
        return False


def list_series(base_url: str, api_key: str) -> List[Dict]:
    r = requests.get(f"{base_url.rstrip('/')}/api/v3/series", headers=_h(api_key), timeout=30)
    r.raise_for_status()
    out = []
    for s in r.json() or []:
        stats = s.get("statistics") or {}
        out.append({
            "sonarr_id": s["id"],
            "title": s.get("title", ""),
            "year": s.get("year"),
            "overview": s.get("overview") or "",
            "poster_url": _poster(s.get("images")),
            "tvdb_id": s.get("tvdbId"),
            "tmdb_id": s.get("tmdbId"),
            "season_count": stats.get("seasonCount", 0),
            "episode_file_count": stats.get("episodeFileCount", 0),
        })
    return out


def list_episodes(base_url: str, api_key: str, series_id: int) -> List[Dict]:
    r = requests.get(
        f"{base_url.rstrip('/')}/api/v3/episode",
        params={"seriesId": series_id, "includeEpisodeFile": True},
        headers=_h(api_key), timeout=30,
    )
    r.raise_for_status()
    out = []
    for e in r.json() or []:
        ef = e.get("episodeFile") or {}
        out.append({
            "sonarr_episode_id": e["id"],
            "season_number": e.get("seasonNumber", 0),
            "episode_number": e.get("episodeNumber", 0),
            "title": e.get("title", ""),
            "description": e.get("overview") or "",
            "air_date": (e.get("airDate") or ""),
            "duration_minutes": int((ef.get("runTime") or 0) / 60) if ef.get("runTime") else 0,
            "has_file": bool(e.get("hasFile")),
            "file_path": ef.get("path") if ef else None,
        })
    return out