@@ -76,13 +76,25 @@ def saveRegionInDB(self, database, idColumn, uid, data, status): # TODO if regio
76
76
if index is not None : # set -> diff -> list
77
77
# Ensure 'selected-classes' is a string before splitting
78
78
print (f"Updating existing data in database: { data } " )
79
- selected_classes_str = data ['selected-classes' ]
79
+ selected_classes_str = data .get ('selected-classes' , data ['class' ])
80
+
81
+ print (f"selected_classes_str: { selected_classes_str } " )
80
82
if isinstance (selected_classes_str , list ) and len (selected_classes_str ) == 1 and isinstance (selected_classes_str [0 ], str ):
81
83
selected_classes_str = selected_classes_str [0 ]
82
84
83
- # Split the strings to create sets
84
- new_cat_set = set (selected_classes_str .split (';' ))
85
- old_cat_set = set (database .loc [index , 'selected-classes' ].split (';' ))
85
+ if 'selected-classes' in data :
86
+ if ';' in selected_classes_str :
87
+ old_cat_set = set (database .loc [index , 'selected-classes' ].split (';' ))
88
+ # Split the strings to create sets
89
+ new_cat_set = set (selected_classes_str .split (';' ))
90
+ else :
91
+ old_cat_set = set (database .loc [index , 'selected-classes' ])
92
+ # Split the strings to create sets
93
+ new_cat_set = set (selected_classes_str )
94
+ else :
95
+ old_cat_set = set (database .loc [index , 'class' ])
96
+ # Split the strings to create sets
97
+ new_cat_set = set (selected_classes_str )
86
98
87
99
for key , value in data .items ():
88
100
_value = value [0 ] if status == 0 else value
0 commit comments