Add series/season/episode ingestion pipeline
CI / lint (push) Canceled after 0s
CI / typecheck (push) Canceled after 0s
CI / unit-tests (push) Canceled after 0s
CI / integration-tests (push) Canceled after 0s
CI / container-build (push) Canceled after 0s
CI / dependency-audit (push) Canceled after 0s

Movie and person ingestion existed, but the series/season/episode API
endpoints had no way to actually populate them from TMDb -- the
biggest remaining gap in Core 0.1's television scope.

- normalize_series/normalize_season in the metadata normalizer,
  including two TMDb quirks that would otherwise silently produce
  empty data: /tv nests keywords and alternative_titles under
  "results" (not "keywords"/"titles" like /movie), and series-level
  creators come from a separate created_by array, not the credits
  cast/crew lists.
- ingest_series upserts the Series row plus season *stubs* (number,
  title, air date, poster -- everything TMDb inlines on /tv/{id});
  ingest_season fetches and upserts one season's full episode list,
  and requires the series to already exist (TMDb has no single
  endpoint for a whole series' episodes, so this has to be a separate
  per-season call).
- Wired through the same job-lifecycle path as movie/person ingestion
  (runner.py, arq tasks, admin endpoints), with the shared boilerplate
  between them factored into runner._run_ingestion and
  admin.py:_enqueue rather than copy-pasted a third and fourth time.
- Unit tests for both normalizers (runnable without a DB) and
  integration tests for the full ingest_series -> ingest_season flow,
  including the series-must-exist-first error and idempotency on
  re-ingestion.

Co-Authored-By: Claude Sonnet 5 <[email protected]>
This commit is contained in:
NixReaper
2026-09-04 19:43:56 -05:00
co-authored by Claude Sonnet 5
parent 669034896a
commit 3f6332559a
11 changed files with 1288 additions and 49 deletions
+348
View File
@@ -339,3 +339,351 @@ def normalize_person(raw: dict) -> NormalizedPerson:
artwork=artwork,
external_ids=external_ids,
)
_SERIES_STATUS_MAP = {
"Planned": "planned",
"Pilot": "pilot",
"In Production": "in_production",
"Returning Series": "returning_series",
"Ended": "ended",
"Canceled": "canceled",
}
@dataclass
class NormalizedSeasonStub:
"""A season summary as it appears embedded in a series payload — enough
to create/update the Season row, but without episodes (those require a
separate per-season fetch — see `normalize_season`)."""
season_number: int
title: str | None
overview: str | None
air_date: date | None
episode_count: int | None
poster: NormalizedArtwork | None
@dataclass
class NormalizedSeries:
external_id: str
title: str
original_title: str | None
overview: str | None
tagline: str | None
status: str
original_language: str | None
first_air_date: date | None
last_air_date: date | None
number_of_seasons: int | None
number_of_episodes: int | None
popularity: float | None
vote_average: float | None
vote_count: int | None
genres: list[str] = field(default_factory=list)
keywords: list[str] = field(default_factory=list)
alternate_titles: list[str] = field(default_factory=list)
networks: list[NormalizedCompanyRef] = field(default_factory=list)
companies: list[NormalizedCompanyRef] = field(default_factory=list)
creators: list[NormalizedPersonRef] = field(default_factory=list)
cast: list[NormalizedPersonRef] = field(default_factory=list)
crew: list[NormalizedPersonRef] = field(default_factory=list)
artwork: list[NormalizedArtwork] = field(default_factory=list)
videos: list[NormalizedVideo] = field(default_factory=list)
external_ids: dict[str, str] = field(default_factory=dict)
seasons: list[NormalizedSeasonStub] = field(default_factory=list)
def normalize_series(raw: dict) -> NormalizedSeries:
credits_ = raw.get("credits") or {}
images = raw.get("images") or {}
videos = (raw.get("videos") or {}).get("results") or []
# Unlike /movie, TMDb's /tv keywords and alternative_titles nest under
# "results", not "keywords"/"titles".
keywords = (raw.get("keywords") or {}).get("results") or []
alt_titles = (raw.get("alternative_titles") or {}).get("results") or []
ext_ids = raw.get("external_ids") or {}
external_ids: dict[str, str] = {}
if ext_ids.get("imdb_id"):
external_ids["imdb"] = ext_ids["imdb_id"]
if ext_ids.get("tvdb_id"):
external_ids["tvdb"] = str(ext_ids["tvdb_id"])
if ext_ids.get("wikidata_id"):
external_ids["wikidata"] = ext_ids["wikidata_id"]
cast = [
NormalizedPersonRef(
external_id=str(c["id"]),
name=c["name"],
profile_path=_image_url(c.get("profile_path")),
department="Acting",
character=c.get("character"),
billing_order=c.get("order"),
gender=_gender_label(c.get("gender")),
)
for c in credits_.get("cast", [])
]
crew = [
NormalizedPersonRef(
external_id=str(c["id"]),
name=c["name"],
profile_path=_image_url(c.get("profile_path")),
department=c.get("department") or "Crew",
job=c.get("job"),
gender=_gender_label(c.get("gender")),
)
for c in credits_.get("crew", [])
]
creators = [
NormalizedPersonRef(
external_id=str(c["id"]),
name=c["name"],
profile_path=_image_url(c.get("profile_path")),
department="Creator",
gender=_gender_label(c.get("gender")),
)
for c in raw.get("created_by", [])
]
networks = [
NormalizedCompanyRef(
external_id=str(n["id"]), name=n["name"], country_code=n.get("origin_country") or None
)
for n in raw.get("networks", [])
]
companies = [
NormalizedCompanyRef(
external_id=str(pc["id"]),
name=pc["name"],
country_code=pc.get("origin_country") or None,
)
for pc in raw.get("production_companies", [])
]
artwork: list[NormalizedArtwork] = []
if raw.get("poster_path"):
artwork.append(
NormalizedArtwork(
"poster",
_image_url_required(raw["poster_path"]),
raw["poster_path"],
None,
None,
None,
None,
)
)
if raw.get("backdrop_path"):
artwork.append(
NormalizedArtwork(
"backdrop",
_image_url_required(raw["backdrop_path"]),
raw["backdrop_path"],
None,
None,
None,
None,
)
)
for kind, tmdb_key in (("poster", "posters"), ("backdrop", "backdrops"), ("logo", "logos")):
for img in images.get(tmdb_key, []):
artwork.append(
NormalizedArtwork(
kind,
_image_url_required(img["file_path"]),
img["file_path"],
img.get("width"),
img.get("height"),
img.get("aspect_ratio"),
img.get("iso_639_1"),
img.get("vote_average"),
)
)
normalized_videos = [
NormalizedVideo(
video_type=(v.get("type") or "other").lower().replace(" ", "_"),
provider_video_id=v["id"],
site=v.get("site"),
url=_video_url(v),
language=v.get("iso_639_1"),
country=v.get("iso_3166_1"),
is_official=bool(v.get("official")),
published_at=_parse_date(v.get("published_at", "")[:10] if v.get("published_at") else None),
)
for v in videos
]
seasons = [
NormalizedSeasonStub(
season_number=s["season_number"],
title=s.get("name"),
overview=s.get("overview") or None,
air_date=_parse_date(s.get("air_date")),
episode_count=s.get("episode_count"),
poster=(
NormalizedArtwork(
"poster", _image_url_required(s["poster_path"]), s["poster_path"],
None, None, None, None,
)
if s.get("poster_path")
else None
),
)
for s in raw.get("seasons", [])
# TMDb includes a "season 0" for specials; keep it (it's a real,
# commonly-referenced season) but drop entries missing a number.
if s.get("season_number") is not None
]
return NormalizedSeries(
external_id=str(raw["id"]),
title=raw.get("name") or raw.get("original_name") or "",
original_title=raw.get("original_name"),
overview=raw.get("overview") or None,
tagline=raw.get("tagline") or None,
status=_SERIES_STATUS_MAP.get(raw.get("status", "Returning Series"), "returning_series"),
original_language=raw.get("original_language"),
first_air_date=_parse_date(raw.get("first_air_date")),
last_air_date=_parse_date(raw.get("last_air_date")),
number_of_seasons=raw.get("number_of_seasons"),
number_of_episodes=raw.get("number_of_episodes"),
popularity=raw.get("popularity"),
vote_average=raw.get("vote_average"),
vote_count=raw.get("vote_count"),
genres=[g["name"] for g in raw.get("genres", [])],
keywords=[k["name"] for k in keywords],
alternate_titles=[t["title"] for t in alt_titles if t.get("title")],
networks=networks,
companies=companies,
creators=creators,
cast=cast,
crew=crew,
artwork=artwork,
videos=normalized_videos,
external_ids=external_ids,
seasons=seasons,
)
@dataclass
class NormalizedEpisode:
external_id: str
episode_number: int
title: str | None
overview: str | None
air_date: date | None
runtime_minutes: int | None
cast: list[NormalizedPersonRef] = field(default_factory=list)
guest_stars: list[NormalizedPersonRef] = field(default_factory=list)
crew: list[NormalizedPersonRef] = field(default_factory=list)
stills: list[NormalizedArtwork] = field(default_factory=list)
@dataclass
class NormalizedSeason:
season_number: int
title: str | None
overview: str | None
air_date: date | None
episode_count: int | None
artwork: list[NormalizedArtwork] = field(default_factory=list)
episodes: list[NormalizedEpisode] = field(default_factory=list)
external_ids: dict[str, str] = field(default_factory=dict)
def _episode_person_refs(entries: list[dict]) -> list[NormalizedPersonRef]:
return [
NormalizedPersonRef(
external_id=str(e["id"]),
name=e["name"],
profile_path=_image_url(e.get("profile_path")),
department="Acting",
character=e.get("character"),
billing_order=e.get("order"),
gender=_gender_label(e.get("gender")),
)
for e in entries
]
def normalize_season(raw: dict) -> NormalizedSeason:
images = raw.get("images") or {}
ext_ids = raw.get("external_ids") or {}
external_ids: dict[str, str] = {}
if ext_ids.get("tvdb_id"):
external_ids["tvdb"] = str(ext_ids["tvdb_id"])
artwork: list[NormalizedArtwork] = []
if raw.get("poster_path"):
artwork.append(
NormalizedArtwork(
"poster", _image_url_required(raw["poster_path"]), raw["poster_path"],
None, None, None, None,
)
)
for img in images.get("posters", []):
artwork.append(
NormalizedArtwork(
"poster",
_image_url_required(img["file_path"]),
img["file_path"],
img.get("width"),
img.get("height"),
img.get("aspect_ratio"),
img.get("iso_639_1"),
img.get("vote_average"),
)
)
episodes = []
for ep in raw.get("episodes", []):
crew = [
NormalizedPersonRef(
external_id=str(c["id"]),
name=c["name"],
profile_path=_image_url(c.get("profile_path")),
department=c.get("department") or "Crew",
job=c.get("job"),
gender=_gender_label(c.get("gender")),
)
for c in ep.get("crew", [])
]
stills: list[NormalizedArtwork] = []
if ep.get("still_path"):
stills.append(
NormalizedArtwork(
"still", _image_url_required(ep["still_path"]), ep["still_path"],
None, None, None, None,
)
)
episodes.append(
NormalizedEpisode(
external_id=str(ep["id"]),
episode_number=ep["episode_number"],
title=ep.get("name"),
overview=ep.get("overview") or None,
air_date=_parse_date(ep.get("air_date")),
runtime_minutes=ep.get("runtime"),
# TMDb's per-episode payload has no separate "main cast"
# list distinct from guest_stars/crew — the series' regular
# cast is assumed, not repeated per episode.
cast=[],
guest_stars=_episode_person_refs(ep.get("guest_stars", [])),
crew=crew,
stills=stills,
)
)
return NormalizedSeason(
season_number=raw["season_number"],
title=raw.get("name"),
overview=raw.get("overview") or None,
air_date=_parse_date(raw.get("air_date")),
episode_count=len(episodes) or None,
artwork=artwork,
episodes=episodes,
external_ids=external_ids,
)