Merge pull request #32 from video-db/ankit/add-public-url

feat: add generate url

Author: ashish-spext

videodb/__about__.py

@@ -1,7 +1,7 @@
 """ About information for videodb sdk"""
 
 
-__version__ = "0.2.9"
+__version__ = "0.2.10"
 __title__ = "videodb"
 __author__ = "videodb"
 __email__ = "[email protected]"

videodb/_constants.py

@@ -70,6 +70,7 @@ class ApiPath:
     storage = "storage"
     download = "download"
     title = "title"
+    generate_url = "generate_url"
 
 
 class Status:

videodb/audio.py

@@ -20,5 +20,12 @@ def __repr__(self) -> str:
             f"length={self.length})"
         )
 
+    def generate_url(self) -> str:
+        url_data = self._connection.post(
+            path=f"{ApiPath.audio}/{self.id}/{ApiPath.generate_url}",
+            params={"collection_id": self.collection_id},
+        )
+        return url_data.get("signed_url", None)
+
     def delete(self) -> None:
         self._connection.delete(f"{ApiPath.audio}/{self.id}")

videodb/client.py

@@ -54,12 +54,15 @@ def get_collections(self) -> List[Collection]:
             for collection in collections_data.get("collections")
         ]
 
-    def create_collection(self, name: str, description: str) -> Collection:
+    def create_collection(
+        self, name: str, description: str, is_public: bool = False
+    ) -> Collection:
         collection_data = self.post(
             path=ApiPath.collection,
             data={
                 "name": name,
                 "description": description,
+                "is_public": is_public,
             },
         )
         self.collection_id = collection_data.get("id", "default")

videodb/image.py

@@ -20,6 +20,13 @@ def __repr__(self) -> str:
             f"url={self.url})"
         )
 
+    def generate_url(self) -> str:
+        url_data = self._connection.post(
+            path=f"{ApiPath.image}/{self.id}/{ApiPath.generate_url}",
+            params={"collection_id": self.collection_id},
+        )
+        return url_data.get("signed_url", None)
+
     def delete(self) -> None:
         self._connection.delete(f"{ApiPath.image}/{self.id}")