diff --git a/vrobbler/apps/videos/metadata.py b/vrobbler/apps/videos/metadata.py index 24799f6..487ffbc 100644 --- a/vrobbler/apps/videos/metadata.py +++ b/vrobbler/apps/videos/metadata.py @@ -71,6 +71,7 @@ class VideoMetadata: twitch_id: Optional[str] = "", base_run_time_seconds: int = 900, ): + self.title = "" self.imdb_id = imdb_id self.youtube_id = youtube_id self.twitch_id = twitch_id diff --git a/vrobbler/apps/videos/sources/omdb.py b/vrobbler/apps/videos/sources/omdb.py index eb661ae..1a6f68a 100644 --- a/vrobbler/apps/videos/sources/omdb.py +++ b/vrobbler/apps/videos/sources/omdb.py @@ -65,6 +65,17 @@ def lookup_video_from_omdb(imdb_id: str) -> Optional[VideoMetadata]: elif media_type in ("series", "episode"): metadata.video_type = VideoType.TV_EPISODE.value + if media_type == "episode": + raw_season = data.get("Season") + if raw_season and raw_season != "N/A": + metadata.season_number = int(raw_season) + raw_episode = data.get("Episode") + if raw_episode and raw_episode != "N/A": + metadata.episode_number = int(raw_episode) + series_imdb_id = data.get("seriesID") + if series_imdb_id and series_imdb_id != "N/A": + metadata.tv_series_imdb_id = series_imdb_id + raw_genres = data.get("Genre") if raw_genres: metadata.genres = [g.strip() for g in raw_genres.split(",") if g.strip()]