From ef1fcd402614fe9130013a817c18a501050c19b3 Mon Sep 17 00:00:00 2001 From: Colin Powell Date: Mon, 27 Jan 2025 01:37:21 -0500 Subject: [PATCH] [videos] Fix video type reference --- vrobbler/apps/videos/sources/imdb.py | 4 ++-- vrobbler/apps/videos/sources/youtube.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/vrobbler/apps/videos/sources/imdb.py b/vrobbler/apps/videos/sources/imdb.py index bd89439..0c7ec02 100644 --- a/vrobbler/apps/videos/sources/imdb.py +++ b/vrobbler/apps/videos/sources/imdb.py @@ -69,14 +69,14 @@ def lookup_video_from_imdb( video_metadata.cover_url, width=800 ) - video_metadata.video_type = VideoType.MOVIE + video_metadata.video_type = VideoType.MOVIE.value series_name = None if imdb_result.get("kind") == "episode": series_name = imdb_result.get("episode of", None).data.get( "title", None ) series, _ = Series.objects.get_or_create(name=series_name) - video_metadata.video_type = VideoType.TV_EPISODE + video_metadata.video_type = VideoType.TV_EPISODE.value video_metadata.tv_series_id = series.id if imdb_result.get("runtimes"): diff --git a/vrobbler/apps/videos/sources/youtube.py b/vrobbler/apps/videos/sources/youtube.py index 8efb137..07afce6 100644 --- a/vrobbler/apps/videos/sources/youtube.py +++ b/vrobbler/apps/videos/sources/youtube.py @@ -32,7 +32,7 @@ def lookup_video_from_youtube(youtube_id: str) -> VideoMetadata: video_metadata.title = yt_metadata.title video_metadata.run_time_seconds = yt_metadata.duration - video_metadata.video_type = VideoType.YOUTUBE + video_metadata.video_type = VideoType.YOUTUBE.value video_metadata.youtube_id = yt_metadata.video_id video_metadata.cover_url = yt_metadata.thumbnail video_metadata.genres = yt_metadata.keywords