feat: add thumbnail timestamp

ankit-v2-3 · ankit-v2-3 · commit 2bcd7e38cdcc · 2024-03-26T14:06:16.000+05:30
diff --git a/videodb/_constants.py b/videodb/_constants.py
@@ -39,6 +39,7 @@ class ApiPath:
     image = "image"
     stream = "stream"
     thumbnail = "thumbnail"
+    thumbnails = "thumbnails"
     upload_url = "upload_url"
     transcription = "transcription"
     index = "index"
diff --git a/videodb/image.py b/videodb/image.py
@@ -9,13 +9,15 @@ def __init__(self, _connection, id: str, collection_id: str, **kwargs) -> None:
         self.id = id
         self.collection_id = collection_id
         self.name = kwargs.get("name", None)
+        self.url = kwargs.get("url", None)
 
     def __repr__(self) -> str:
         return (
             f"Image("
             f"id={self.id}, "
             f"collection_id={self.collection_id}, "
-            f"name={self.name})"
+            f"name={self.name}), "
+            f"url={self.url}"
         )
 
     def delete(self) -> None:
diff --git a/videodb/video.py b/videodb/video.py
@@ -7,6 +7,7 @@
     SubtitleStyle,
     Workflows,
 )
+from videodb.image import Image
 from videodb.search import SearchFactory, SearchResult
 from videodb.shot import Shot
 
@@ -88,15 +89,31 @@ def generate_stream(self, timeline: Optional[List[Tuple[int, int]]] = None) -> s
         )
         return stream_data.get("stream_url", None)
 
-    def generate_thumbnail(self):
-        if self.thumbnail_url:
+    def generate_thumbnail(self, time: Optional[float] = None) -> Union[str, Image]:
+        if self.thumbnail_url and not time:
             return self.thumbnail_url
+
+        if time:
+            thumbnail_data = self._connection.post(
+                path=f"{ApiPath.video}/{self.id}/{ApiPath.thumbnail}",
+                data={
+                    "time": time,
+                },
+            )
+            return Image(self._connection, **thumbnail_data)
+
         thumbnail_data = self._connection.get(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.thumbnail}"
         )
         self.thumbnail_url = thumbnail_data.get("thumbnail_url")
         return self.thumbnail_url
 
+    def get_thumbnails(self) -> List[Image]:
+        thumbnails_data = self._connection.get(
+            path=f"{ApiPath.video}/{self.id}/{ApiPath.thumbnails}"
+        )
+        return [Image(self._connection, **thumbnail) for thumbnail in thumbnails_data]
+
     def _fetch_transcript(self, force: bool = False) -> None:
         if self.transcript and not force:
             return
