small alert fix

cef_3M/sql.py

@@ -23,9 +23,3 @@ ergoEngine = create_engine(
 )
 
 SessionMaker = sessionmaker(autocommit=False, autoflush=False, bind=engine, )
-
-
-def ergoQueryFetchOne(q: str, **kwargs):
-    with ergoEngine.connect() as connection:
-        connection.execute(text("use ergo"))
-        return connection.execute(text(q), kwargs).fetchone()

cef_3M/util.py

@@ -12,10 +12,12 @@ from minio import Minio
 from redis.asyncio import Redis
 from pywebpush import WebPusher
 
-from .sql import SessionMaker, AlertEndpoints, ergoQueryFetchOne
+from .sql import SessionMaker, AlertEndpoints
 
 from fastapi import UploadFile
 
+from .sql_generated import History
+
 safeName = re.compile(r"[^\w\d\.-]")
 
 
@@ -53,11 +55,12 @@ class ErgoClient:
                     "p256dh": target.p256dh
                 }
             })
-            messageQuery = ergoQueryFetchOne("SELECT `data` FROM `history` WHERE `msgid` = :id", id=int(msgid)).data
+            messageQuery: History = session.query(History).filter(History.msgid == int(msgid)).first()
-            message = json.loads(messageQuery.decode("utf8"))
+
+            message = json.loads(messageQuery.data.decode("utf8"))
             encoded = json.dumps({
                 "channel": channel,
-                "from": message["AccountName"],
+                "from": message["Account"],
                 "content": message["Message"]["Message"]
             }).encode("utf8")
 
@@ -79,8 +82,9 @@ class ErgoClient:
                 message = await pubsub.get_message(ignore_subscribe_messages=True, timeout=None)
                 try:
                     await self.handleUserMessage(message)
-                except Exception as e:
+                except:
-                    print("Error in handling user message", e)
+                    print("Error in handling user message")
+                    traceback.print_exc()