diff --git a/src/database/add.py b/src/database/add.py index 6d534b3..54d67d9 100644 --- a/src/database/add.py +++ b/src/database/add.py @@ -13,6 +13,7 @@ def add_account( enc_login: bytes, enc_pass: bytes, ) -> bool: + """Adds account to db. Returns true, if on success""" account = models.Account( user_id=user_id, name=name, salt=salt, enc_login=enc_login, enc_pass=enc_pass ) @@ -27,6 +28,7 @@ def add_account( def add_master_pass(engine: Engine, user_id: int, salt: bytes, passwd: bytes) -> bool: + """Adds master password to db. Returns true, if on success""" master_pass = models.MasterPass(user_id=user_id, salt=salt, passwd=passwd) try: with sqlmodel.Session(engine) as session: diff --git a/src/database/get.py b/src/database/get.py index cb90d1e..5772d71 100644 --- a/src/database/get.py +++ b/src/database/get.py @@ -5,6 +5,7 @@ from . import models def get_master_pass(engine: Engine, user_id: int) -> tuple[bytes, bytes] | None: + """Gets master pass. Returns tuple of salt and password""" statement = sqlmodel.select(models.MasterPass).where( models.MasterPass.user_id == user_id ) @@ -17,6 +18,7 @@ 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) with sqlmodel.Session(engine) as session: result = session.exec(statement) @@ -26,6 +28,7 @@ def get_accounts(engine: Engine, user_id: int) -> list[str]: 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""" statement = sqlmodel.select(models.Account).where( models.Account.user_id == user_id and models.Account.name == name ) diff --git a/src/database/prepare.py b/src/database/prepare.py index 6d8e449..b171aca 100644 --- a/src/database/prepare.py +++ b/src/database/prepare.py @@ -13,4 +13,4 @@ def get_engine(host: str, user: str, passwd: str, db: str) -> Engine: def prepare(engine: Engine) -> None: - sqlmodel.SQLModel.metadata.create_all(engine) + sqlmodel.SQLModel.metadata.create_all(engine, [models.Account, models.MasterPass])