@@ -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