Tendokyu/artemis
1
0
Fork 0
mirror of synced 2025-02-12 16:34:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

use AsyncSession directly

Browse Source 
see the warnings in https://docs.sqlalchemy.org/en/14/orm/extensions/asyncio.html#using-asyncio-scoped-session
This commit is contained in:
beerpsi 2024-11-14 13:10:14 +07:00
parent 4c33f4282a
commit 789d50c406
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/data/database.py
View File

@ -1,4 +1,3 @@
import asyncio
import logging import logging
import os import os
import secrets import secrets
@ -15,10 +14,8 @@ import pymysql.err
from sqlalchemy.ext.asyncio import ( from sqlalchemy.ext.asyncio import (
AsyncEngine, AsyncEngine,
AsyncSession, AsyncSession,
async_scoped_session,
create_async_engine, create_async_engine,
) )
from sqlalchemy.orm import sessionmaker
from core.config import CoreConfig from core.config import CoreConfig
from core.data.schema import ArcadeData, BaseData, CardData, UserData, metadata from core.data.schema import ArcadeData, BaseData, CardData, UserData, metadata
@ -47,8 +44,7 @@ class Data:
self.__engine = Data.engine self.__engine = Data.engine
if Data.session is MISSING: if Data.session is MISSING:
s = sessionmaker(Data.engine, expire_on_commit=False, class_=AsyncSession) Data.session = AsyncSession(Data.engine, expire_on_commit=False)
Data.session = async_scoped_session(s, asyncio.current_task)
if Data.user is MISSING: if Data.user is MISSING:
Data.user = UserData(self.config, self.session) Data.user = UserData(self.config, self.session)