feat: add timeout exception

ankit-v2-3 · ankit-v2-3 · commit 577c7788c025 · 2024-06-13T16:31:00.000+05:30
diff --git a/videodb/_utils/_http_client.py b/videodb/_utils/_http_client.py
@@ -19,6 +19,7 @@
 from videodb.exceptions import (
     AuthenticationError,
     InvalidRequestError,
+    RequestTimeoutError,
 )
 
 logger = logging.getLogger(__name__)
@@ -109,8 +110,8 @@ def _handle_request_error(self, e: requests.exceptions.RequestException) -> None
             ) from None
 
         elif isinstance(e, requests.exceptions.Timeout):
-            raise InvalidRequestError(
-                "Invalid request: Request timed out", e.response
+            raise RequestTimeoutError(
+                "Timeout error: Request timed out", e.response
             ) from None
 
         elif isinstance(e, requests.exceptions.ConnectionError):
diff --git a/videodb/exceptions.py b/videodb/exceptions.py
@@ -37,6 +37,16 @@ def __init__(self, message, response=None):
         self.response = response
 
 
+class RequestTimeoutError(VideodbError):
+    """
+    Raised when a request times out.
+    """
+
+    def __init__(self, message, response=None):
+        super(RequestTimeoutError, self).__init__(message)
+        self.response = response
+
+
 class SearchError(VideodbError):
     """
     Raised when a search is invalid.
diff --git a/videodb/video.py b/videodb/video.py
@@ -275,17 +275,7 @@ def create_scene_index(
                 "callback_url": callback_url,
             },
         )
-        return [
-            Scene(
-                scene.get("scene_id"),
-                self.id,
-                scene.get("start"),
-                scene.get("end"),
-                [],
-                scene.get("description"),
-            )
-            for scene in scenes_data.get("scene_index_records", [])
-        ]
+        return scenes_data.get("scene_index_records", [])
 
     def get_scene_indexes(self) -> List:
         index_data = self._connection.get(
