Video file limit

Author: hovnatan

superannotate/common.py

@@ -244,6 +244,7 @@ def dump_output(output_dir, platform, classes, files_dict):
         save_desktop_format(output_dir, classes, files_dict)
 
 
+MAX_VIDEO_SIZE = 5 * 1024 * 1024 * 1024  # 5 GB limit
 MAX_IMAGE_SIZE = 100 * 1024 * 1024  # 100 MB limit
 MAX_IMAGE_RESOLUTION = {
     "Vector": 100_000_000,

superannotate/db/projects.py

@@ -233,6 +233,9 @@ def upload_video_to_project(
     :rtype: list of strs
     """
     logger.info("Uploading from video %s.", str(video_path))
+    file_size = Path(video_path).stat().st_size
+    if file_size > common.MAX_VIDEO_SIZE:
+        raise SAVideoSizeTooLarge(file_size)
     rotate_code = None
     try:
         meta_dict = ffmpeg.probe(str(video_path))
@@ -413,15 +416,23 @@ def upload_videos_from_folder_to_project(
 
     filenames = []
     for path in filtered_paths:
-        filenames += upload_video_to_project(
-            project,
-            path,
-            target_fps=target_fps,
-            start_time=start_time,
-            end_time=end_time,
-            annotation_status=annotation_status,
-            image_quality_in_editor=image_quality_in_editor
-        )
+        try:
+            filename = upload_video_to_project(
+                project,
+                path,
+                target_fps=target_fps,
+                start_time=start_time,
+                end_time=end_time,
+                annotation_status=annotation_status,
+                image_quality_in_editor=image_quality_in_editor
+            )
+        except Exception as e:
+            logger.warning(
+                "Couldn't upload video %s. Exception: ", path, str(e)
+            )
+            pass
+        else:
+            filenames += filename
 
     return filenames
 

superannotate/exceptions.py

@@ -25,6 +25,15 @@ def __init__(self, file_size, file_name=""):
         )
 
 
+class SAVideoSizeTooLarge(SABaseException):
+    def __init__(self, file_size, file_name=""):
+        super().__init__(
+            0, "Video " + file_name + " size " + str(file_size // 1024**3) +
+            " GB is larger than " + str(common.MAX_VIDEO_SIZE // 1024**3) +
+            " GB limit."
+        )
+
+
 class SAExistingProjectNameException(SABaseException):
     pass
 

superannotate/version.py

@@ -1 +1 @@
-__version__ = "2.5.4"
+__version__ = "2.5.5b1"