diff --git a/python/kontor-cli/kontor/controllers/media.py b/python/kontor-cli/kontor/controllers/media.py index da4420c..0f5e610 100644 --- a/python/kontor-cli/kontor/controllers/media.py +++ b/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})