Skip to content

Commit 8559737

Browse files
committed
Fix parsers
1 parent 7c20d30 commit 8559737

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

superannotate/mixp/utils/parsers.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1041,7 +1041,7 @@ def upload_preannotations_from_folder_to_project(*args, **kwargs):
10411041

10421042

10431043
def upload_images_from_folder_to_project(*args, **kwargs):
1044-
project = kwargs.get("source_project", None)
1044+
project = kwargs.get("project", None)
10451045
if not project:
10461046
project = args[0]
10471047
folder_path = kwargs.get("folder_path", None)
@@ -1056,24 +1056,22 @@ def upload_images_from_folder_to_project(*args, **kwargs):
10561056
else:
10571057
recursive_subfolders = False
10581058

1059-
from ...common import DEFAULT_IMAGE_EXTENSIONS
1060-
extensions = DEFAULT_IMAGE_EXTENSIONS
10611059
extensions = kwargs.get("extensions", None)
10621060
if not extensions:
10631061
extensions = args[2:3]
10641062
if extensions:
10651063
extensions = extensions[0]
10661064
else:
1065+
from ...common import DEFAULT_IMAGE_EXTENSIONS
10671066
extensions = DEFAULT_IMAGE_EXTENSIONS
10681067

1069-
from ...common import DEFAULT_FILE_EXCLUDE_PATTERNS
1070-
exclude_file_patterns = DEFAULT_FILE_EXCLUDE_PATTERNS
10711068
exclude_file_patterns = kwargs.get("exclude_file_patterns", None)
10721069
if not exclude_file_patterns:
10731070
exclude_file_patterns = args[5:6]
10741071
if exclude_file_patterns:
10751072
exclude_file_patterns = exclude_file_patterns[0]
10761073
else:
1074+
from ...common import DEFAULT_FILE_EXCLUDE_PATTERNS
10771075
exclude_file_patterns = DEFAULT_FILE_EXCLUDE_PATTERNS
10781076

10791077
from pathlib import Path
@@ -1644,14 +1642,17 @@ def delete_images(*args, **kwargs):
16441642
project = kwargs.get("project", None)
16451643
if not project:
16461644
project = args[0]
1647-
image_names = kwargs.get("image_names", None)
1645+
image_names = kwargs.get("image_names", False)
16481646
if not image_names:
16491647
image_names = args[1]
1648+
if image_names == None:
1649+
from superannotate.db.images import search_images as sa_search_images
1650+
image_names = sa_search_images(project)
16501651
return {
16511652
"event_name": "delete_images",
16521653
"properties":
16531654
{
16541655
"project_name": get_project_name(project),
16551656
"Image Count": len(image_names),
16561657
}
1657-
}
1658+
}

0 commit comments

Comments
 (0)