From 91518559b06671853e428d383623c58ed1d17bea Mon Sep 17 00:00:00 2001
From: StNicolay <103897650+StNicolay@users.noreply.github.com>
Date: Wed, 5 Oct 2022 14:52:23 +0300
Subject: [PATCH] Added exception handling

---
 src/database/add.py | 33 +++++++++++++++++++++++----------
 1 file changed, 23 insertions(+), 10 deletions(-)

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