Add insert user custom mode function

e-mission · Dec 29, 2023 · 5c8f20c · 5c8f20c
1 parent 29a4222
commit 5c8f20c
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/emission/core/wrapper/user.py b/emission/core/wrapper/user.py
@@ -238,6 +238,18 @@ def getUserCustomModes(self):
       return list(sortedModes)  
     else:
       return []
+
+  def insertUserCustomMode(self, new_mode):
+    from datetime import datetime
+    user = get_profile_db().find_one({'user_id': self.uuid})
+    modes = user['modes'] if 'modes' in user else {} 
+    modes[new_mode] = {
+      'createdAt': datetime.now(),
+      'frequency': 0,
+      'isActive': True,
+    }
+    get_profile_db().update_one({'user_id': self.uuid}, {'$set': {'modes': modes}})
+    return self.getUserCustomModes()
 
   def updateUserCustomMode(self, updated_mode):
     from datetime import datetime

diff --git a/emission/net/api/cfc_webapp.py b/emission/net/api/cfc_webapp.py
@@ -313,6 +313,16 @@ def getUserCustomModes():
   user = User.fromUUID(user_uuid)
   return { 'modes' : user.getUserCustomModes() }
 
+@post('/mode/insert')
+def insertUserCustomModes():
+  logging.debug("Called insertUserMode")
+  new_mode = request.json['new_mode']
+  user_uuid = getUUID(request)
+  user = User.fromUUID(user_uuid)
+  to_return = user.insertUserCustomMode(new_mode)
+  logging.debug("Successfully inserted mode for user %s" % user_uuid)
+  return { 'modes' : to_return }
+
 @post('/mode/update')
 def updateUserCustomMode():
   logging.debug("Called updateUserMode")