From 789d50c406560068d4677e82d9c017bceb7183ae Mon Sep 17 00:00:00 2001 From: beerpsi Date: Thu, 14 Nov 2024 13:10:14 +0700 Subject: [PATCH] use AsyncSession directly see the warnings in https://docs.sqlalchemy.org/en/14/orm/extensions/asyncio.html#using-asyncio-scoped-session --- core/data/database.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/core/data/database.py b/core/data/database.py index 16bd67b..0095a20 100644 --- a/core/data/database.py +++ b/core/data/database.py @@ -1,4 +1,3 @@ -import asyncio import logging import os import secrets @@ -15,10 +14,8 @@ import pymysql.err from sqlalchemy.ext.asyncio import ( AsyncEngine, AsyncSession, - async_scoped_session, create_async_engine, ) -from sqlalchemy.orm import sessionmaker from core.config import CoreConfig from core.data.schema import ArcadeData, BaseData, CardData, UserData, metadata @@ -47,8 +44,7 @@ class Data: self.__engine = Data.engine if Data.session is MISSING: - s = sessionmaker(Data.engine, expire_on_commit=False, class_=AsyncSession) - Data.session = async_scoped_session(s, asyncio.current_task) + Data.session = AsyncSession(Data.engine, expire_on_commit=False) if Data.user is MISSING: Data.user = UserData(self.config, self.session)