feat: add new video types (MKV, MOV, AVI, MP4, WMV, and WebM) to hyko…

…sdk io
BIGmama-technology · Feb 25, 2024 · 73ae807 · 73ae807
1 parent d205b75
commit 73ae807
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 8 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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"