chenxu233/nonebot_plugin_dialectlist
1
0
Fork 0
mirror of https://github.com/ChenXu233/nonebot_plugin_dialectlist.git synced 2025-02-07 19:36:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 修复排名错位的问题

Browse Source
This commit is contained in:
Chen_Xu233 2024-07-28 09:00:37 +08:00
parent 44fd75e570
commit 065619f0a8
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
nonebot_plugin_dialectlist/utils.py
View File

@ -46,15 +46,12 @@ async def ensure_group(matcher: Matcher, session: Session = Depends(extract_sess
async def persist_id2user_id(ids: List) -> List[str]:
whereclause: List[ColumnElement[bool]] = []
whereclause.append(or_(*[SessionModel.id == id for id in ids]))
statement = (
select(SessionModel).where(*whereclause)
# .join(SessionModel, SessionModel.id == MessageRecord.session_persist_id)
)
user_ids = []
async with get_session() as db_session:
records = (await db_session.scalars(statement)).all()
return [i.id1 for i in records]
for i in ids:
user_id = (await db_session.scalar(select(SessionModel).where(or_(*[SessionModel.id == i])))).id1 # type: ignore
user_ids.append(user_id)
return user_ids
async def user_id2persist_id(ids: List[str]) -> List[int]: