unit3dprep 1.1.2__tar.gz → 1.1.4__tar.gz

{unit3dprep-1.1.2/unit3dprep.egg-info → unit3dprep-1.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unit3dprep
-Version: 1.1.2
+Version: 1.1.4
 Summary: Web UI + CLI di pre-flight per tracker Unit3D, companion di Unit3DWebUp (audio ITA, nomenclatura ItaTorrents, hardlink, upload)
 Author: Davide Sidoti
 License: GNU GENERAL PUBLIC LICENSE

{unit3dprep-1.1.2 → unit3dprep-1.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "unit3dprep"
-version = "1.1.2"
+version = "1.1.4"
 description = "Web UI + CLI di pre-flight per tracker Unit3D, companion di Unit3DWebUp (audio ITA, nomenclatura ItaTorrents, hardlink, upload)"
 readme = "README.md"
 license = { file = "LICENSE" }

{unit3dprep-1.1.2 → unit3dprep-1.1.4}/unit3dprep/core.py

@@ -434,6 +434,44 @@ def tmdb_search(kind: str, query: str, year: str, api_key: str, language: str |
     return normalized
 
 
+def tv_season_status(data: dict) -> dict:
+    """Compute per-season completion from a TMDB ``/tv/{id}`` object.
+
+    Returns::
+
+        {
+          "show_status": str,        # raw TMDB `status` (e.g. "Ended", "Returning Series")
+          "seasons": {               # keyed by season_number as a string
+            "1": {"episode_count": int, "complete": bool, "next_air_date": str | None},
+            ...
+          },
+        }
+
+    A season is ``complete`` (no further episodes expected) when the show has
+    ended/been canceled, when no next episode is scheduled, or when the next
+    episode to air belongs to a different season. The single season still
+    receiving episodes (``next_episode_to_air.season_number``) is the only one
+    marked not complete.
+    """
+    show_status = (data.get("status") or "").strip()
+    ended = show_status in ("Ended", "Canceled")
+    next_ep = data.get("next_episode_to_air") or {}
+    next_season = next_ep.get("season_number")
+    next_air_date = next_ep.get("air_date")
+    seasons: dict[str, dict] = {}
+    for s in data.get("seasons") or []:
+        num = s.get("season_number")
+        if num is None:
+            continue
+        ongoing = (not ended) and next_season is not None and next_season == num
+        seasons[str(num)] = {
+            "episode_count": s.get("episode_count") or 0,
+            "complete": not ongoing,
+            "next_air_date": next_air_date if ongoing else None,
+        }
+    return {"show_status": show_status, "seasons": seasons}
+
+
 # ---------------------------------------------------------------------------
 # Name builder
 # ---------------------------------------------------------------------------

{unit3dprep-1.1.2 → unit3dprep-1.1.4}/unit3dprep/web/api/library.py

@@ -7,6 +7,7 @@ from __future__ import annotations
 import asyncio
 import json
 import os
+import time
 from pathlib import Path
 from typing import Any, AsyncGenerator
 
@@ -17,10 +18,12 @@ from sse_starlette.sse import EventSourceResponse
 
 from ...core import (
     audio_languages,
+    tmdb_fetch,
     tmdb_fetch_bilingual,
     tmdb_poster_url,
     tmdb_search,
     tmdb_year,
+    tv_season_status,
 )
 from ...media import (
     MediaItem,
@@ -33,7 +36,7 @@ from ...media import (
 from ...i18n import get_request_lang, t as _i18n_t
 from ..db import list_uploads, record_upload
 from ..lang_cache import get_many_langs, set_lang
-from ..tmdb_cache import get_cache, get_many, set_cache
+from ..tmdb_cache import get_cache, get_many, set_cache, set_series_status
 
 TMDB_API_KEY = os.environ.get("TMDB_API_KEY", "")
 
@@ -242,6 +245,67 @@ async def library_categories():
     })
 
 
+_STATUS_TTL_SECONDS = 12 * 3600
+
+
+def _series_status_fresh(fetched_at: Any, show_status: str) -> bool:
+    """Cached season status is fresh when the show is finished (immutable) or
+    the last fetch is within the TTL."""
+    if show_status in ("Ended", "Canceled"):
+        return True
+    if not isinstance(fetched_at, (int, float)):
+        return False
+    return (time.time() - fetched_at) < _STATUS_TTL_SECONDS
+
+
+@router.get("/library/series-status")
+async def library_series_status(path: str):
+    """Per-season completion (complete vs ongoing) + TMDB episode count for a
+    series. Resolves the TMDB id from cache, refetches ``/tv/{id}`` when the
+    cached status is missing/stale, then caches it. On-demand only — never hit
+    during the library scan, so it costs at most one TMDB call per panel open.
+    """
+    record = await get_cache(path)
+    tmdb_id = (record or {}).get("tmdb_id", "")
+    if not tmdb_id:
+        return JSONResponse({"tmdb_id": "", "show_status": "", "seasons": {}})
+
+    cached_meta = (record or {}).get("seasons_meta")
+    cached_status = (record or {}).get("show_status", "")
+    if cached_meta is not None and _series_status_fresh(
+        (record or {}).get("status_fetched_at"), cached_status
+    ):
+        return JSONResponse(
+            {"tmdb_id": tmdb_id, "show_status": cached_status, "seasons": cached_meta}
+        )
+
+    if not TMDB_API_KEY:
+        return JSONResponse(
+            {"tmdb_id": tmdb_id, "show_status": cached_status, "seasons": cached_meta or {}}
+        )
+
+    loop = asyncio.get_event_loop()
+    try:
+        data = await loop.run_in_executor(None, tmdb_fetch, "tv", tmdb_id, TMDB_API_KEY)
+    except Exception:
+        # TMDB hiccup: serve stale cache if present, else report unknown.
+        if cached_meta is not None:
+            return JSONResponse(
+                {"tmdb_id": tmdb_id, "show_status": cached_status, "seasons": cached_meta}
+            )
+        return JSONResponse({"tmdb_id": tmdb_id, "show_status": "", "seasons": {}})
+
+    status_obj = tv_season_status(data)
+    await set_series_status(path, status_obj["show_status"], status_obj["seasons"])
+    return JSONResponse(
+        {
+            "tmdb_id": tmdb_id,
+            "show_status": status_obj["show_status"],
+            "seasons": status_obj["seasons"],
+        }
+    )
+
+
 @router.get("/library/{category}")
 async def library_list(request: Request, category: str):
     if category not in discover_categories():

{unit3dprep-1.1.2 → unit3dprep-1.1.4}/unit3dprep/web/api/tmdb.py

@@ -8,9 +8,9 @@ from fastapi import APIRouter, HTTPException, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 
-from ...core import tmdb_fetch_bilingual, tmdb_poster_url, tmdb_search, tmdb_year
+from ...core import tmdb_fetch_bilingual, tmdb_poster_url, tmdb_search, tmdb_year, tv_season_status
 from ...i18n import get_request_lang, t
-from ..tmdb_cache import delete_cache, set_cache
+from ..tmdb_cache import delete_cache, set_cache, set_series_status
 
 TMDB_API_KEY = os.environ.get("TMDB_API_KEY", "")
 
@@ -64,6 +64,14 @@ async def set_manual(request: Request, body: SetBody):
         overview=(data.get("overview") or "")[:300],
         overview_en=(data.get("overview_en") or "")[:300],
     )
+    if body.tmdb_kind == "tv":
+        # `data` already holds the full /tv/{id} object → derive season status
+        # with no extra TMDB call. set_cache above replaced the record, so this
+        # merge must run after it.
+        status_obj = tv_season_status(data)
+        await set_series_status(
+            body.source_path, status_obj["show_status"], status_obj["seasons"]
+        )
     return JSONResponse({"ok": True, "title": title, "year": year, "poster": poster})