From dd3d370df6e66e2aadad4ca53a5e59e558cdbeeb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E2=80=98KimigaiiWuyi=E2=80=99?= <444835641@qq.com> Date: Fri, 12 May 2023 10:59:16 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20=E9=80=82=E9=85=8D=E5=A4=9A?= =?UTF-8?q?=E6=9D=A1=E7=BB=91=E5=AE=9A=E8=AE=B0=E5=BD=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gsuid_core/utils/database/dal.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/gsuid_core/utils/database/dal.py b/gsuid_core/utils/database/dal.py index bf9b4a9..99ca830 100644 --- a/gsuid_core/utils/database/dal.py +++ b/gsuid_core/utils/database/dal.py @@ -207,14 +207,21 @@ class SQLA: result = await session.execute(sql) 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 - ) -> Optional[GsUser]: + ) -> Optional[List[GsUser]]: async with self.async_session() as session: async with session.begin(): sql = select(GsUser).where(GsUser.user_id == user_id) 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 with self.async_session() as session: