fix: scene and frame class

Author: ankit-v2-3

videodb/image.py

@@ -32,14 +32,12 @@ def __init__(
         video_id: str,
         scene_id: str,
         url: str,
-        frame_no: int,
         frame_time: float,
         description: str,
     ):
         super().__init__(_connection=_connection, id=id, collection_id=None, url=url)
         self.scene_id = scene_id
         self.video_id = video_id
-        self.frame_no = frame_no
         self.frame_time = frame_time
         self.description = description
 
@@ -50,7 +48,6 @@ def __repr__(self) -> str:
             f"video_id={self.video_id}, "
             f"scene_id={self.scene_id}, "
             f"url={self.url}, "
-            f"frame_no={self.frame_no}, "
             f"frame_time={self.frame_time}, "
             f"description={self.description})"
         )
@@ -61,7 +58,6 @@ def to_json(self):
             "video_id": self.video_id,
             "scene_id": self.scene_id,
             "url": self.url,
-            "frame_no": self.frame_no,
             "frame_time": self.frame_time,
             "description": self.description,
         }

videodb/scene.py

@@ -8,12 +8,12 @@
 class Scene:
     def __init__(
         self,
-        id: str,
         video_id: str,
         start: float,
         end: float,
-        frames: List[Frame],
         description: str,
+        id: str = None,
+        frames: List[Frame] = [],
     ):
         self.id = id
         self.video_id = video_id

videodb/video.py

@@ -200,18 +200,17 @@ def _format_scene_collection(self, collection_data: dict) -> SceneCollection:
                     self.id,
                     scene.get("scene_id"),
                     frame.get("url"),
-                    frame.get("frame_no"),
                     frame.get("frame_time"),
                     frame.get("description"),
                 )
                 frames.append(frame)
             scene = Scene(
-                scene.get("scene_id"),
-                self.id,
-                scene.get("start"),
-                scene.get("end"),
-                frames,
-                scene.get("description"),
+                video_id=self.id,
+                start=scene.get("start"),
+                end=scene.get("end"),
+                description=scene.get("description"),
+                id=scene.get("scene_id"),
+                frames=frames,
             )
             scenes.append(scene)