Common image extensions

Author: hovnatan

superannotate/__main__.py

@@ -2,11 +2,12 @@
 import json
 import logging
 import sys
-from pathlib import Path
 import tempfile
+from pathlib import Path
 
 import superannotate as sa
 
+from . import common
 from .exceptions import SABaseException
 
 logger = logging.getLogger("superannotate-python-sdk")
@@ -198,7 +199,7 @@ def video_upload(command_name, args):
     parser.add_argument(
         '--extensions',
         required=False,
-        default=None,
+        default=common.DEFAULT_VIDEO_EXTENSIONS,
         type=_list_str,
         help=
         'List of video extensions to include. Default is mp4,avi,mov,webm,flv,mpg,ogg'
@@ -264,7 +265,7 @@ def image_upload(command_name, args):
     )
     parser.add_argument(
         '--extensions',
-        default=None,
+        default=common.DEFAULT_IMAGE_EXTENSIONS,
         type=_list_str,
         help=
         'List of image extensions to include. Default is jpg,jpeg,png,tif,tiff,webp,bmp'

superannotate/common.py

@@ -12,6 +12,11 @@
 
 logger = logging.getLogger("superannotate-python-sdk")
 
+DEFAULT_IMAGE_EXTENSIONS = ("jpg", "jpeg", "png", "tif", "tiff", "webp", "bmp")
+DEFAULT_FILE_EXCLUDE_PATTERNS = ("___save.png", "___fuse.png")
+
+DEFAULT_VIDEO_EXTENSIONS = ("mp4", "avi", "mov", "webm", "flv", "mpg", "ogg")
+
 _PROJECT_TYPES = {"Vector": 1, "Pixel": 2}
 
 _ANNOTATION_STATUSES = {

superannotate/db/projects.py

@@ -35,7 +35,9 @@
     search_annotation_classes
 )
 from .images import get_image_metadata, search_images
-from .project_api import get_project_metadata_bare, get_project_and_folder_metadata
+from .project_api import (
+    get_project_and_folder_metadata, get_project_metadata_bare
+)
 from .users import get_team_contributor_metadata
 
 logger = logging.getLogger("superannotate-python-sdk")
@@ -335,7 +337,7 @@ def upload_video_to_project(
 def upload_videos_from_folder_to_project(
     project,
     folder_path,
-    extensions=("mp4", "avi", "mov", "webm", "flv", "mpg", "ogg"),
+    extensions=common.DEFAULT_VIDEO_EXTENSIONS,
     exclude_file_patterns=(),
     recursive_subfolders=False,
     target_fps=None,
@@ -424,10 +426,10 @@ def upload_videos_from_folder_to_project(
 def upload_images_from_folder_to_project(
     project,
     folder_path,
-    extensions=("jpg", "jpeg", "png", "tif", "tiff", "webp", "bmp"),
+    extensions=common.DEFAULT_IMAGE_EXTENSIONS,
     annotation_status="NotStarted",
     from_s3_bucket=None,
-    exclude_file_patterns=("___save.png", "___fuse.png"),
+    exclude_file_patterns=common.DEFAULT_FILE_EXCLUDE_PATTERNS,
     recursive_subfolders=False,
     image_quality_in_editor=None
 ):