nonebot2/nonebot/plugins/single_session.py

32 lines
834 B
Python
Raw Normal View History

from typing import Dict
2021-01-25 18:15:25 +08:00
from nonebot.adapters import Event
2021-09-27 00:19:30 +08:00
from nonebot.message import (IgnoredException, run_preprocessor,
run_postprocessor)
2021-01-25 18:15:25 +08:00
_running_matcher: Dict[str, int] = {}
2021-01-25 18:15:25 +08:00
@run_preprocessor
async def preprocess(event: Event):
2021-02-24 22:21:47 +08:00
try:
session_id = event.get_session_id()
2021-02-24 23:56:52 +08:00
except Exception:
2021-02-24 22:21:47 +08:00
return
current_event_id = id(event)
event_id = _running_matcher.get(session_id, None)
if event_id and event_id != current_event_id:
2021-03-11 10:55:18 +08:00
raise IgnoredException("Another matcher running")
2021-02-24 22:21:47 +08:00
_running_matcher[session_id] = current_event_id
2021-01-25 18:15:25 +08:00
@run_postprocessor
async def postprocess(event: Event):
2021-02-24 22:21:47 +08:00
try:
session_id = event.get_session_id()
2021-02-24 23:56:52 +08:00
except Exception:
2021-02-24 22:21:47 +08:00
return
if session_id in _running_matcher:
del _running_matcher[session_id]