diff --git a/hyko_sdk/io.py b/hyko_sdk/io.py index 6095bf45..08b50cbc 100644 --- a/hyko_sdk/io.py +++ b/hyko_sdk/io.py @@ -255,6 +255,10 @@ def validate_object(val: "Video"): assert obj_ext.value in [ Ext.MP4, Ext.WEBM, + Ext.AVI, + Ext.MKV, + Ext.MOV, + Ext.WMV, ], "Invalid file extension for Video error" return Video(obj_ext=obj_ext, file_name=file_name) @@ -266,6 +270,10 @@ def validate_file_name(file_name: str): assert obj_ext.value in [ Ext.MP4, Ext.WEBM, + Ext.AVI, + Ext.MKV, + Ext.MOV, + Ext.WMV, ], "Invalid file extension for Video error" return Video(obj_ext=obj_ext, file_name=file_name) diff --git a/hyko_sdk/types.py b/hyko_sdk/types.py index 80be2920..9f28702c 100644 --- a/hyko_sdk/types.py +++ b/hyko_sdk/types.py @@ -12,3 +12,7 @@ class Ext(str, Enum): WAV = "wav" MP4 = "mp4" MP3 = "mp3" + AVI = "avi" + MKV = "mkv" + MOV = "mov" + WMV = "wmv" diff --git a/hyko_toolkit/functions/utils/converters/video_converter/main.py b/hyko_toolkit/functions/utils/converters/video_converter/main.py index f26c5ff8..b0f5c5f3 100644 --- a/hyko_toolkit/functions/utils/converters/video_converter/main.py +++ b/hyko_toolkit/functions/utils/converters/video_converter/main.py @@ -37,10 +37,10 @@ async def main(inputs: Inputs, params: Params) -> Outputs: type_to_object = { "webm": Ext.WEBM, "mp4": Ext.MP4, - # "avi": Ext.AVI, - # "mkv": Ext.MKV, - # "mov": Ext.MOV, - # "wmv": Ext.WMV, + "avi": Ext.AVI, + "mkv": Ext.MKV, + "mov": Ext.MOV, + "wmv": Ext.WMV, } with open(f"/app/video{ext}", "wb") as f: diff --git a/hyko_toolkit/functions/utils/converters/video_converter/metadata.py b/hyko_toolkit/functions/utils/converters/video_converter/metadata.py index d0d73845..4973a5ad 100644 --- a/hyko_toolkit/functions/utils/converters/video_converter/metadata.py +++ b/hyko_toolkit/functions/utils/converters/video_converter/metadata.py @@ -12,11 +12,11 @@ class SupportedTypes(str, Enum): - # mkv = "mkv" - # mov = "mov" - # avi = "avi" + mkv = "mkv" + mov = "mov" + avi = "avi" mp4 = "mp4" - # wmv = "wmv" + wmv = "wmv" webm = "webm"