handling an edge case when a label already exists in database

e-mission · Jan 9, 2024 · 41fea13 · 41fea13
1 parent 3db594c
commit 41fea13
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/emission/core/wrapper/user.py b/emission/core/wrapper/user.py
@@ -247,11 +247,17 @@ def insertUserCustomLabel(self, inserted_label):
     key = inserted_label['key']
     label = inserted_label['label']
     items = user[key] if key in user else {} 
-    items[label] = {
-      'createdAt': datetime.now(),
-      'frequency': 0,
-      'isActive': True,
-    }
+
+    # if label exists in database, chage it as 'active' label
+    if label in items:
+      items[label]['isActive'] = True
+    else:
+      items[label] = {
+        'createdAt': datetime.now(),
+        'frequency': 0,
+        'isActive': True,
+      }
+
     get_profile_db().update_one({'user_id': self.uuid}, {'$set': {key: items}})
     return self.getUserCustomLabel(key)