mirror of
https://github.com/Genshin-bots/gsuid_core.git
synced 2025-05-12 06:55:49 +08:00
✨ 新增select_user_data_by_user_id
系列方法
This commit is contained in:
parent
a68df12faa
commit
cad7c6f113
@ -207,6 +207,15 @@ 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(
|
||||||
|
self, user_id: str
|
||||||
|
) -> Optional[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
|
||||||
|
|
||||||
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:
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
@ -456,10 +465,18 @@ class SQLA:
|
|||||||
data = await self.select_user_data(uid)
|
data = await self.select_user_data(uid)
|
||||||
return data.cookie if data else None
|
return data.cookie if data else None
|
||||||
|
|
||||||
|
async def get_user_cookie_by_user_id(self, user_id: str) -> Optional[str]:
|
||||||
|
data = await self.select_user_data_by_user_id(user_id)
|
||||||
|
return data.cookie if data else None
|
||||||
|
|
||||||
async def cookie_validate(self, uid: str) -> bool:
|
async def cookie_validate(self, uid: str) -> bool:
|
||||||
data = await self.select_user_data(uid)
|
data = await self.select_user_data(uid)
|
||||||
return True if data and data.status is None else False
|
return True if data and data.status is None else False
|
||||||
|
|
||||||
|
async def get_user_stoken_by_user_id(self, user_id: str) -> Optional[str]:
|
||||||
|
data = await self.select_user_data_by_user_id(user_id)
|
||||||
|
return data.stoken if data and data.stoken else None
|
||||||
|
|
||||||
async def get_user_stoken(self, uid: str) -> Optional[str]:
|
async def get_user_stoken(self, uid: str) -> Optional[str]:
|
||||||
data = await self.select_user_data(uid)
|
data = await self.select_user_data(uid)
|
||||||
return data.stoken if data and data.stoken else None
|
return data.stoken if data and data.stoken else None
|
||||||
|
Loading…
x
Reference in New Issue
Block a user