fix: media_id TypeError

ankit-v2-3 · ankit-v2-3 · commit aae1b06fd3e4 · 2024-02-19T12:18:50.000+05:30
diff --git a/videodb/client.py b/videodb/client.py
@@ -57,9 +57,9 @@ def upload(
             description,
             callback_url,
         )
-        if upload_data.get("id").startswith("m-"):
+        if upload_data.get("id", "").startswith("m-"):
             return Video(self, **upload_data) if upload_data else None
-        elif upload_data.get("id").startswith("a-"):
+        elif upload_data.get("id", "").startswith("a-"):
             return Audio(self, **upload_data) if upload_data else None
-        elif upload_data.get("id").startswith("i-"):
+        elif upload_data.get("id", "").startswith("i-"):
             return Image(self, **upload_data) if upload_data else None
diff --git a/videodb/collection.py b/videodb/collection.py
@@ -101,9 +101,9 @@ def upload(
             description,
             callback_url,
         )
-        if upload_data.get("id").startswith("m-"):
+        if upload_data.get("id", "").startswith("m-"):
             return Video(self._connection, **upload_data) if upload_data else None
-        elif upload_data.get("id").startswith("a-"):
+        elif upload_data.get("id", "").startswith("a-"):
             return Audio(self._connection, **upload_data) if upload_data else None
-        elif upload_data.get("id").startswith("i-"):
+        elif upload_data.get("id", "").startswith("i-"):
             return Image(self._connection, **upload_data) if upload_data else None
