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

This commit is contained in:
StNicolay 2022-12-25 17:59:26 +03:00
parent f299173e56
commit bbc9650357

View File

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