ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-10-31 15:06:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix single_session bug

Browse Source
This commit is contained in:
AkiraXie
2021-02-24 22:21:47 +08:00
parent 869df38238
commit 6f2b9a1d13
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
nonebot/plugins/single_session.py
View File

@@ -10,17 +10,23 @@ _running_matcher: Dict[str, int] = {}
@run_preprocessor
async def _(matcher: Matcher, bot: Bot, event: Event, state: T_State):
session_id = event.get_session_id()
event_id = id(event)
if _running_matcher.get(session_id, None) != event_id:
try:
session_id = event.get_session_id()
except ValueError:
return
current_event_id = id(event)
event_id = _running_matcher.get(session_id, None)
if event_id and event_id != current_event_id:
raise IgnoredException("Annother matcher running")
_running_matcher[session_id] = event_id
_running_matcher[session_id] = current_event_id
@run_postprocessor
async def _(matcher: Matcher, exception: Optional[Exception], bot: Bot, event: Event, state: T_State):
session_id = event.get_session_id()
try:
session_id = event.get_session_id()
except ValueError:
return
if session_id in _running_matcher:
del _running_matcher[session_id]