From c8410a877c57aeaf1bdea39380ab3269574d8311 Mon Sep 17 00:00:00 2001
From: StNicolay <103897650+StNicolay@users.noreply.github.com>
Date: Thu, 29 Sep 2022 15:58:56 +0300
Subject: [PATCH] Created get.py

---
 src/database/get.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 src/database/get.py

diff --git a/src/database/get.py b/src/database/get.py
new file mode 100644
index 0000000..a16fd66
--- /dev/null
+++ b/src/database/get.py
@@ -0,0 +1,31 @@
+import mariadb
+
+
+def get_master_pass(con: mariadb.Connection, id: int) -> tuple[bytes, bytes]:
+    """Returns tuple of salt and hashed master password"""
+    cursor = con.cursor()
+    cursor.execute("SELECT salt, passwd FROM master_pass IF user_id = ?", [id])
+    result = cursor.fetchone()
+    cursor.close()
+    return result
+
+
+def get_accounts(con: mariadb.Connection, id: int) -> list[str]:
+    """Returns list of user accounts"""
+    cursor = con.cursor()
+    cursor.execute("SELECT acc_name FROM accounts IF user_id = ?", [id])
+    return [i[0] for i in cursor.fetchall()]
+
+
+def get_account_info(
+    id: int, name: str, con: mariadb.Connection
+) -> tuple[bytes, bytes, bytes]:
+    """Returns tuple of salt, login and password"""
+    cursor = con.cursor()
+    cursor.execute(
+        """SELECT salt, enc_login, enc_pass IF user_id = ? AND acc_name = ?""",
+        [id, name],
+    )
+    result = cursor.fetchone()
+    cursor.close()
+    return result