StNicolay/PassManager
Archived
1
1
Fork 0
Code Pull Requests Releases 8 Activity

Changed database.get file. Added fetchall method calls to close session sooner

Browse Source
This commit is contained in:
StNicolay 2022-12-25 17:59:26 +03:00
parent f299173e56
commit bbc9650357
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/database/get.py
View File

@ -35,8 +35,8 @@ def get_accounts(
if to_sort: if to_sort:
statement = statement.order_by(models.Account.name) statement = statement.order_by(models.Account.name)
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
result = list(session.exec(statement)) result = session.exec(statement).fetchall()
return result return result
def get_all_accounts( def get_all_accounts(
@ -52,16 +52,16 @@ def get_all_accounts(
.order_by(models.Account.name) .order_by(models.Account.name)
) )
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
result = session.exec(statement) result = session.exec(statement).fetchall()
yield from ( yield from (
( (
account.name, account.name,
account.salt, account.salt,
account.enc_login, account.enc_login,
account.enc_password, account.enc_password,
)
for account in result
) )
for account in result
)
def get_account_info( def get_account_info(