Added exception handling

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(
+    try:
-        "INSERT INTO master_pass (user_id, salt, passwd) VALUES (?, ?, ?)",
+        cursor.execute(
-        [id, hashed_passwd, salt],
+            "INSERT INTO master_pass (user_id, salt, passwd) VALUES (?, ?, ?)",
-    )
+            [id, hashed_passwd, salt],
-    cursor.close()
+        )
+        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(
+    try:
-        "INSERT INTO accounts (user_id, acc_name, salt, enc_login, enc_pass) VALUES (?, ?, ?, ?, ?, ?)",
+        cursor.execute(
-        [id, acc_name, salt, enc_login, enc_passwd],
+            "INSERT INTO accounts (user_id, acc_name, salt, enc_login, enc_pass) VALUES (?, ?, ?, ?, ?, ?)",
-    )
+            [id, acc_name, salt, enc_login, enc_passwd],
-    cursor.close()
+        )
+        cursor.close()
+    except Exception:
+        traceback.print_exc()
+        return False
+    else:
+        return True