feat: add transcode

ankit-v2-3 · ankit-v2-3 · commit 552ed166a8f6 · 2025-04-29T10:01:21.000+05:30
diff --git a/videodb/__about__.py b/videodb/__about__.py
@@ -1,7 +1,7 @@
 """ About information for videodb sdk"""
 
 
-__version__ = "0.2.13"
+__version__ = "0.2.14"
 __title__ = "videodb"
 __author__ = "videodb"
 __email__ = "contact@videodb.io"
diff --git a/videodb/__init__.py b/videodb/__init__.py
@@ -16,6 +16,10 @@
     SubtitleBorderStyle,
     SubtitleStyle,
     TextStyle,
+    TranscodeMode,
+    ResizeMode,
+    VideoConfig,
+    AudioConfig,
 )
 from videodb.client import Connection
 from videodb.exceptions import (
@@ -43,6 +47,10 @@
     "TextStyle",
     "SceneExtractionType",
     "Segmenter",
+    "TranscodeMode",
+    "ResizeMode",
+    "VideoConfig",
+    "AudioConfig",
 ]
 
 
diff --git a/videodb/_constants.py b/videodb/_constants.py
@@ -76,6 +76,7 @@ class ApiPath:
     web = "web"
     translate = "translate"
     dub = "dub"
+    transcode = "transcode"
 
 
 class Status:
@@ -164,3 +165,28 @@ class TextStyle:
     tabsize: int = 4
     x: Union[str, int] = "(main_w-text_w)/2"
     y: Union[str, int] = "(main_h-text_h)/2"
+
+
+class TranscodeMode:
+    lightning = "lightning"
+    economy = "economy"
+
+
+class ResizeMode:
+    crop = "crop"
+    fit = "fit"
+    pad = "pad"
+
+
+@dataclass
+class VideoConfig:
+    resolution: int = 720
+    quality: int = 23
+    framerate: int = None
+    aspect_ratio: str = None
+    resize_mode: str = ResizeMode.crop
+
+
+@dataclass
+class AudioConfig:
+    mute: bool = False
diff --git a/videodb/client.py b/videodb/client.py
@@ -8,6 +8,9 @@
 from videodb.__about__ import __version__
 from videodb._constants import (
     ApiPath,
+    TranscodeMode,
+    VideoConfig,
+    AudioConfig,
 )
 
 from videodb.collection import Collection
@@ -188,6 +191,41 @@ def youtube_search(
         )
         return search_data.get("results")
 
+    def transcode(
+        self,
+        source: str,
+        callback_url: str,
+        mode: TranscodeMode = TranscodeMode.economy,
+        start_ts: int = None,
+        end_ts: int = None,
+        video_config: VideoConfig = VideoConfig(),
+        audio_config: AudioConfig = AudioConfig(),
+    ) -> None:
+        """Transcode the video
+
+        :param str source: URL of the video to transcode, preferably a downloadable URL
+        :param str callback_url: URL to receive the callback
+        :param TranscodeMode mode: Mode of the transcoding
+        :param int start_ts: Start timestamp of the video to transcode (optional)
+        :param int end_ts: End timestamp of the video to transcode (optional)
+        :param VideoConfig video_config: Video configuration (optional)
+        :param AudioConfig audio_config: Audio configuration (optional)
+        :return: None
+        :rtype: None
+        """
+        self.post(
+            path=f"{ApiPath.transcode}",
+            data={
+                "source": source,
+                "callback_url": callback_url,
+                "mode": mode,
+                "start_ts": start_ts,
+                "end_ts": end_ts,
+                "video_config": video_config.__dict__,
+                "audio_config": audio_config.__dict__,
+            },
+        )
+
     def upload(
         self,
         file_path: str = None,
