Genshin-bots/gsuid_core
1
0
Fork 0
mirror of https://github.com/Genshin-bots/gsuid_core.git synced 2025-05-12 06:55:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

🎨 适配多条绑定记录

Browse Source
This commit is contained in:
‘KimigaiiWuyi’ 2023-05-12 10:59:16 +08:00
parent cad7c6f113
commit dd3d370df6
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
gsuid_core/utils/database/dal.py
View File

@ -207,14 +207,21 @@ class SQLA:
result = await session.execute(sql) result = await session.execute(sql)
return data[0] if (data := result.scalars().all()) else None return data[0] if (data := result.scalars().all()) else None
async def select_user_data_by_user_id( async def select_user_all_data_by_user_id(
self, user_id: str self, user_id: str
) -> Optional[GsUser]: ) -> Optional[List[GsUser]]:
async with self.async_session() as session: async with self.async_session() as session:
async with session.begin(): async with session.begin():
sql = select(GsUser).where(GsUser.user_id == user_id) sql = select(GsUser).where(GsUser.user_id == user_id)
result = await session.execute(sql) result = await session.execute(sql)
return data[0] if (data := result.scalars().all()) else None data = result.scalars().all()
return data if data else None
async def select_user_data_by_user_id(
self, user_id: str
) -> Optional[GsUser]:
data = await self.select_user_all_data_by_user_id(user_id)
return data[0] if data else None
async def select_cache_cookie(self, uid: str) -> Optional[str]: async def select_cache_cookie(self, uid: str) -> Optional[str]:
async with self.async_session() as session: async with self.async_session() as session: