session-close-fix

resolvers/stat.py

@@ -52,13 +52,13 @@ def get_topic_shouts_stat(topic_id: int):
             )
         )
     )
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0
 
 
 def get_topic_authors_stat(topic_id: int):
-    # authors
-    q = (
+    count_query = (
         select(func.count(distinct(ShoutAuthor.author)))
         .select_from(join(ShoutTopic, Shout, ShoutTopic.shout == Shout.id))
         .join(ShoutAuthor, ShoutAuthor.shout == Shout.id)
@@ -70,7 +70,10 @@ def get_topic_authors_stat(topic_id: int):
             )
         )
     )
-    result = local_session().execute(q).first()
+
+    # Выполняем запрос и получаем результат
+    with local_session() as session:
+        result = session.execute(count_query).first()
     return result[0] if result else 0
 
 
@@ -79,7 +82,8 @@ def get_topic_followers_stat(topic_id: int):
     q = select(func.count(distinct(aliased_followers.follower))).filter(
         aliased_followers.topic == topic_id
     )
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0
 
 
@@ -106,8 +110,8 @@ def get_topic_comments_stat(topic_id: int):
         ShoutTopic.topic == topic_id
     )
     q = q.outerjoin(sub_comments, ShoutTopic.shout == sub_comments.c.shout_id)
-
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0
 
 
@@ -141,7 +145,8 @@ def get_author_authors_stat(author_id: int):
             aliased_authors.author != author_id,
         )
     )
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0
 
 
@@ -150,7 +155,8 @@ def get_author_followers_stat(author_id: int):
     q = select(func.count(distinct(aliased_followers.follower))).filter(
         aliased_followers.author == author_id
     )
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0
 
 
@@ -172,8 +178,8 @@ def get_author_comments_stat(author_id: int):
         .subquery()
     )
     q = select(sub_comments.c.comments_count).filter(sub_comments.c.id == author_id)
-
-    result = local_session().execute(q).first()
+    with local_session() as session:
+        result = session.execute(q).first()
     return result[0] if result else 0