Browse Source

[videos] Fix video type reference

Colin Powell 3 months ago
parent
commit
ef1fcd4026
2 changed files with 3 additions and 3 deletions
  1. 2 2
      vrobbler/apps/videos/sources/imdb.py
  2. 1 1
      vrobbler/apps/videos/sources/youtube.py

+ 2 - 2
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"):

+ 1 - 1
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