Skip to content

Commit 9b393bb

Browse files
committed
Added categorizatioon check.
1 parent e80ffdf commit 9b393bb

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

src/superannotate/lib/core/usecases/annotations.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1949,7 +1949,9 @@ def get_or_create_folder(self, folder_name: str) -> FolderEntity:
19491949
if not folder_name:
19501950
return self._root_folder
19511951

1952-
response = self._service_provider.folders.get_by_name(self._project, folder_name)
1952+
response = self._service_provider.folders.get_by_name(
1953+
self._project, folder_name
1954+
)
19531955
if response.ok:
19541956
return response.data
19551957

@@ -1969,7 +1971,6 @@ def get_or_create_folder(self, folder_name: str) -> FolderEntity:
19691971

19701972
return response.data
19711973

1972-
19731974
def attach_items(
19741975
self, folder: FolderEntity, item_names: List[str]
19751976
) -> List[BaseItemEntity]:
@@ -2010,10 +2011,18 @@ def serialize_folder_name(val):
20102011
val = val.strip().translate(translation_table)
20112012
return val.lower()
20122013

2014+
@property
2015+
def categorization_enabled(self):
2016+
settings = self._service_provider.projects.list_settings(self._project).data
2017+
for setting in settings:
2018+
if setting.attribute == "CategorizeItems":
2019+
return bool(setting.value)
2020+
return False
2021+
20132022
def execute(self):
20142023
if self.is_valid():
20152024
# TODO check categories status in the project
2016-
skip_categorization = False
2025+
categorization_enabled = self.categorization_enabled
20172026
serialized_original_folder_map = {}
20182027
failed, skipped, uploaded = [], [], []
20192028
# folder_id -> item_name -> annotation
@@ -2087,7 +2096,7 @@ def execute(self):
20872096
{i.item.name for i in items_to_upload}
20882097
- set(failed_annotations).union(skipped)
20892098
)
2090-
if not skip_categorization:
2099+
if categorization_enabled:
20912100
item_id_category_map = {}
20922101
for item_name in uploaded_annotations:
20932102
category = name_annotation_map[item_name]["metadata"].get(

0 commit comments

Comments
 (0)