🚑 Fix db_session and CQ escape

2022-09-29 14:33:48 +08:00 · 2022-09-29 14:33:48 +08:00 · b650acb674
commit b650acb674
parent 3d21d83bf0
1 changed files with 7 additions and 6 deletions
--- a/qqbot/bot.py
+++ b/qqbot/bot.py
@ -25,12 +25,13 @@ def recv_msg():
        and data["group_id"] == 346058845
    ):
        name = data["raw_message"]
-        query = select(m for m in Melee if m.name == name)
-        if query.count() > 0:
-            return {
-                "reply": f"[CQ]:image,file=http://localhost:7000/pics/{name}.png",
-                "at_sender": False,
-            }
+        with db_session:
+            query = select(m for m in Melee if m.name == name)
+            if query.count() > 0:
+                return {
+                    "reply": f"[CQ:image,file=http://localhost:7000/pics/{name}.png]",
+                    "at_sender": False,
+                }
    return "OK"