Massive code cleanup

src/database/add.py

@@ -11,11 +11,15 @@ def add_account(
     name: str,
     salt: bytes,
     enc_login: bytes,
-    enc_pass: bytes,
+    enc_password: bytes,
 ) -> bool:
     """Adds account to the database. Returns true on success, false otherwise"""
     account = models.Account(
-        user_id=user_id, name=name, salt=salt, enc_login=enc_login, enc_pass=enc_pass
+        user_id=user_id,
+        name=name,
+        salt=salt,
+        enc_login=enc_login,
+        enc_pass=enc_password,
     )
     try:
         with sqlmodel.Session(engine) as session:
@@ -27,9 +31,18 @@ def add_account(
         return True
 
 
-def add_master_pass(engine: Engine, user_id: int, salt: bytes, passwd: bytes) -> bool:
+def add_master_pass(
+    engine: Engine,
+    user_id: int,
+    salt: bytes,
+    password_hash: bytes,
+) -> bool:
     """Adds master password the database. Returns true on success, false otherwise"""
-    master_pass = models.MasterPass(user_id=user_id, salt=salt, passwd=passwd)
+    master_pass = models.MasterPass(
+        user_id=user_id,
+        salt=salt,
+        passwd=password_hash,
+    )
     try:
         with sqlmodel.Session(engine) as session:
             session.add(master_pass)

src/database/change.py

@@ -5,13 +5,13 @@ from . import models
 
 
 def change_master_pass(
-    engine: Engine, user_id: int, salt: bytes, passwd: bytes
+    engine: Engine, user_id: int, salt: bytes, password: bytes
 ) -> None:
     """Changes master password and salt in the database"""
     statement = (
         sqlmodel.update(models.MasterPass)
         .where(models.MasterPass.user_id == user_id)
-        .values(salt=salt, passwd=passwd)
+        .values(salt=salt, passwd=password)
     )
     with sqlmodel.Session(engine) as session:
         session.exec(statement)

src/database/delete.py

@@ -6,7 +6,9 @@ from . import models
 
 def purge_accounts(engine: Engine, user_id: int) -> None:
     """Deletes all user's accounts"""
-    statement = sqlmodel.delete(models.Account).where(models.Account.user_id == user_id)
+    statement = sqlmodel.delete(models.Account).where(
+        models.Account.user_id == user_id,
+    )
     with sqlmodel.Session(engine) as session:
         session.exec(statement)
         session.commit()
@@ -25,7 +27,8 @@ def delete_master_pass(engine: Engine, user_id: int) -> None:
 def delete_account(engine: Engine, user_id: int, name: str) -> None:
     """Deletes specific user account"""
     statement = sqlmodel.delete(models.Account).where(
-        models.Account.user_id == user_id, models.Account.name == name
+        models.Account.user_id == user_id,
+        models.Account.name == name,
     )
     with sqlmodel.Session(engine) as session:
         session.exec(statement)

src/database/get.py

@@ -1,14 +1,19 @@
+from typing import Iterator
+
 import sqlmodel
 from sqlalchemy.future import Engine
 
 from . import models
 
 
-def get_master_pass(engine: Engine, user_id: int) -> tuple[bytes, bytes] | None:
+def get_master_pass(
+    engine: Engine,
+    user_id: int,
+) -> tuple[bytes, bytes] | None:
     """Gets master pass. Returns tuple of salt and password
     or None if it wasn't found"""
     statement = sqlmodel.select(models.MasterPass).where(
-        models.MasterPass.user_id == user_id
+        models.MasterPass.user_id == user_id,
     )
     with sqlmodel.Session(engine) as session:
         result = session.exec(statement).first()
@@ -19,19 +24,38 @@ def get_master_pass(engine: Engine, user_id: int) -> tuple[bytes, bytes] | None:
 
 def get_accounts(engine: Engine, user_id: int) -> list[str]:
     """Gets list of account names"""
-    statement = sqlmodel.select(models.Account).where(models.Account.user_id == user_id)
+    statement = sqlmodel.select(models.Account).where(
+        models.Account.user_id == user_id,
+    )
     with sqlmodel.Session(engine) as session:
         result = session.exec(statement)
         return [account.name for account in result]
 
 
+def get_all_accounts(
+    engine: Engine, user_id: int
+) -> Iterator[tuple[str, bytes, bytes, bytes]]:
+    """Returns an iterator of tuples, where values represent account's name, salt,
+    encrypted login and encrypted password"""
+    statement = sqlmodel.select(models.Account).where(
+        models.Account.user_id == user_id,
+    )
+    with sqlmodel.Session(engine) as session:
+        result = session.exec(statement)
+        yield from (
+            (account.name, account.salt, account.enc_login, account.enc_pass)
+            for account in result
+        )
+
+
 def get_account_info(
     engine: Engine, user_id: int, name: str
 ) -> tuple[bytes, bytes, bytes]:
     """Gets account info. Returns tuple of salt, login and password
     or None if it wasn't found"""
     statement = sqlmodel.select(models.Account).where(
-        models.Account.user_id == user_id, models.Account.name == name
+        models.Account.user_id == user_id,
+        models.Account.name == name,
     )
     with sqlmodel.Session(engine) as session:
         result = session.exec(statement).first()

src/database/prepare.py

@@ -6,7 +6,8 @@ from . import models
 
 def get_engine(host: str, user: str, passwd: str, db: str) -> Engine:
     """Creates an engine for mariadb with pymysql as connector"""
-    engine = sqlmodel.create_engine(f"mariadb+pymysql://{user}:{passwd}@{host}/{db}")
+    uri = f"mariadb+pymysql://{user}:{passwd}@{host}/{db}"
+    engine = sqlmodel.create_engine(uri)
     return engine