From bbc96503571fb7b01d3744fff20f8ebba31f2659 Mon Sep 17 00:00:00 2001
From: StNicolay <stnicolay2007@mail.ru>
Date: Sun, 25 Dec 2022 17:59:26 +0300
Subject: [PATCH] Changed database.get file. Added fetchall method calls to
 close session sooner

---
 src/database/get.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/src/database/get.py b/src/database/get.py
index 99242ce..0e1c714 100644
--- a/src/database/get.py
+++ b/src/database/get.py
@@ -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(