add Enum for Videotype

python/kontor-cli/kontor/controllers/media.py

@@ -1,3 +1,4 @@
+from enum import Enum
 from pathlib import Path
 
 from cement import Controller, ex
@@ -5,6 +6,11 @@ from kontor_schema import KontorDB
 from kontor_video import VideoLink, MediaVideo
 
 
+class VideoType(Enum):
+    MEDIA_FILE = "media_file"
+    MEDIA_VIDEO = "media_video"
+
+
 class Media(Controller):
     class Meta:
         label = 'media'
@@ -45,7 +51,7 @@ class Media(Controller):
         downloads = db.get_download_list()
         self.app.log.info(f"found {len(downloads)} links for download")
         for file_id, url in downloads.items():
-            link = VideoLink(url)
+            link = VideoLink(url, VideoType.MEDIA_FILE)
             file_name = link.download(download_dir=data['media_dir'])
             if file_name is None:
                 db.update_entry('media_file', file_id, {'file_name': None, 'should_download': 1})