feat: add get, create collection

ankit-v2-3 · ankit-v2-3 · commit 320b1a483567 · 2024-03-10T16:55:20.000+05:30
diff --git a/videodb/client.py b/videodb/client.py
@@ -3,6 +3,7 @@
 from typing import (
     Optional,
     Union,
+    List,
 )
 
 from videodb._constants import (
@@ -39,6 +40,33 @@ def get_collection(self, collection_id: Optional[str] = "default") -> Collection
             collection_data.get("description"),
         )
 
+    def get_collections(self) -> List[Collection]:
+        collections_data = self.get(path=ApiPath.collection)
+        return [
+            Collection(
+                self,
+                collection.get("id"),
+                collection.get("name"),
+                collection.get("description"),
+            )
+            for collection in collections_data.get("collections")
+        ]
+
+    def create_collection(self, name: str, description: str) -> Collection:
+        collection_data = self.post(
+            path=ApiPath.collection,
+            data={
+                "name": name,
+                "description": description,
+            },
+        )
+        return Collection(
+            self,
+            collection_data.get("id"),
+            collection_data.get("name"),
+            collection_data.get("description"),
+        )
+
     def upload(
         self,
         file_path: str = None,
diff --git a/videodb/collection.py b/videodb/collection.py
@@ -27,6 +27,14 @@ def __init__(self, _connection, id: str, name: str = None, description: str = No
         self.name = name
         self.description = description
 
+    def __repr__(self) -> str:
+        return (
+            f"Collection("
+            f"id={self.id}, "
+            f"name={self.name}, "
+            f"description={self.description})"
+        )
+
     def get_videos(self) -> List[Video]:
         videos_data = self._connection.get(path=f"{ApiPath.video}")
         return [Video(self._connection, **video) for video in videos_data.get("videos")]
