py/api.py

@@ -49,7 +49,9 @@ class API:
             if action == "change_password":
                 self.db.update_password(data)
             if action == "get_data":
-                self.db.get_additional_info(data)
+                self.db.get_data(data)
+            if action == "change_data":
+                self.db.change_data(data)
             if action == "check_credentials":
                 self.db.check_credentials(data)
 

py/db.py

@@ -19,9 +19,17 @@ class DB:
         user_data = {"hashed_password": hashed_password}
         self.database[username] = user_data
 
+    def change_data(self, data):
+        username = data.get['username']
+        data = data.get['data']
+        if not self.check_credentials(data):
+            return False
+
+        self.database[username]['data'] = data
+        return True
+
     def update_password(self, data):
         username = data.get['username']
-        old_password = data.get['old_password']
         new_password = data.get['new_password']
         if not self.check_credentials(data):
             return False
@@ -40,12 +48,10 @@ class DB:
         entered_hashed_password = self.hash_password(password)
         return stored_hashed_password == entered_hashed_password
 
-    def get_additional_info(self, data):
+    def get_data(self, data):
         username = data.get['username']
-        password = data.get['password']
         if not self.check_credentials(data):
             return None
 
-        send_back = self.database[username]
-        del send_back['hashed_password']
+        send_back = self.database[username].get['data']
         return send_back