diff --git a/src/database/add.py b/src/database/add.py index 8efeef3..121e413 100644 --- a/src/database/add.py +++ b/src/database/add.py @@ -1,15 +1,22 @@ import mariadb +import traceback def add_master_pass( id: int, hashed_passwd: bytes, salt: bytes, con: mariadb.Connection ) -> bool: cursor = con.cursor() - cursor.execute( - "INSERT INTO master_pass (user_id, salt, passwd) VALUES (?, ?, ?)", - [id, hashed_passwd, salt], - ) - cursor.close() + try: + cursor.execute( + "INSERT INTO master_pass (user_id, salt, passwd) VALUES (?, ?, ?)", + [id, hashed_passwd, salt], + ) + cursor.close() + except Exception: + traceback.print_exc() + return False + else: + return True def add_account( @@ -21,8 +28,14 @@ def add_account( con: mariadb.Connection, ) -> bool: cursor = con.cursor() - cursor.execute( - "INSERT INTO accounts (user_id, acc_name, salt, enc_login, enc_pass) VALUES (?, ?, ?, ?, ?, ?)", - [id, acc_name, salt, enc_login, enc_passwd], - ) - cursor.close() + try: + cursor.execute( + "INSERT INTO accounts (user_id, acc_name, salt, enc_login, enc_pass) VALUES (?, ?, ?, ?, ?, ?)", + [id, acc_name, salt, enc_login, enc_passwd], + ) + cursor.close() + except Exception: + traceback.print_exc() + return False + else: + return True