From b924d30616d78eaa1da5d5cf0940a3b8eeee667a Mon Sep 17 00:00:00 2001 From: Patrick_Pluto Date: Fri, 20 Sep 2024 15:26:18 +0200 Subject: [PATCH] More Database Functions --- py/api.py | 4 +++- py/db.py | 16 +++++++++++----- 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/py/api.py b/py/api.py index aa226eb..7b09e1d 100644 --- a/py/api.py +++ b/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) diff --git a/py/db.py b/py/db.py index 37c9ca0..ef82111 100644 --- a/py/db.py +++ b/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 -- 2.39.5