From cdb9d31fa42a7823a08e114eab7a06a9616876f3 Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Wed, 29 Nov 2023 13:50:20 +0300
Subject: [PATCH] query-fix

---
 resolvers/reaction.py | 42 ++++++++++++++++++++++--------------------
 1 file changed, 22 insertions(+), 20 deletions(-)

diff --git a/resolvers/reaction.py b/resolvers/reaction.py
index d94d42e5..4343aed6 100644
--- a/resolvers/reaction.py
+++ b/resolvers/reaction.py
@@ -380,27 +380,29 @@ async def load_reactions_by(_, info, by, limit=50, offset=0):
     q = q.where(Reaction.deleted_at.is_(None))
     q = q.limit(limit).offset(offset)
     reactions = []
-    for [
-        reaction,
-        author,
-        shout,
-        reacted_stat,
-        commented_stat,
-        rating_stat,
-    ] in local_session().execute(q):
-        reaction.created_by = author
-        reaction.shout = shout
-        reaction.stat = {
-            "rating": rating_stat,
-            "commented": commented_stat,
-            "reacted": reacted_stat,
-        }
-        reaction.kind = reaction.kind.name
-        reactions.append(reaction)
+    with local_session() as session:
+        result_rows = session.execute(q)
+        for [
+            reaction,
+            author,
+            shout,
+            reacted_stat,
+            commented_stat,
+            rating_stat,
+        ] in result_rows:
+            reaction.created_by = author
+            reaction.shout = shout
+            reaction.stat = {
+                "rating": rating_stat,
+                "commented": commented_stat,
+                "reacted": reacted_stat,
+            }
+            reaction.kind = reaction.kind.name
+            reactions.append(reaction)
 
-    # sort if by stat is present
-    if by.get("stat"):
-        reactions = sorted(reactions, key=lambda r: r.stat.get(by["stat"]) or r.created_at)
+        # sort if by stat is present
+        if by.get("stat"):
+            reactions = sorted(reactions, key=lambda r: r.stat.get(by["stat"]) or r.created_at)
 
     return reactions