mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
114 lines
3.1 KiB
Python
114 lines
3.1 KiB
Python
from typing import List, Optional, Callable, Union
|
|
|
|
from aiocqhttp import Event as CQEvent
|
|
from aiocqhttp.bus import EventBus
|
|
|
|
from . import NoneBot
|
|
from .exceptions import CQHttpError
|
|
from .log import logger
|
|
from .session import BaseSession
|
|
|
|
_bus = EventBus()
|
|
|
|
class EventHandler:
|
|
__slots__ = ('events', 'func')
|
|
|
|
def __init__(self, events: List[str], func: Callable):
|
|
self.events = events
|
|
self.func = func
|
|
|
|
|
|
def _make_event_deco(post_type: str) -> Callable:
|
|
def deco_deco(arg: Optional[Union[str, Callable]] = None,
|
|
*events: str) -> Callable:
|
|
def deco(func: Callable) -> EventHandler:
|
|
if isinstance(arg, str):
|
|
events_tmp = list(map(lambda x: f"{post_type}.{x}", [arg] + list(events)))
|
|
for e in events_tmp:
|
|
_bus.subscribe(e, func)
|
|
return EventHandler(events_tmp, func)
|
|
else:
|
|
_bus.subscribe(post_type, func)
|
|
return EventHandler([post_type], func)
|
|
|
|
if isinstance(arg, Callable):
|
|
return deco(arg) # type: ignore
|
|
return deco
|
|
|
|
return deco_deco
|
|
|
|
|
|
on_notice = _make_event_deco('notice')
|
|
on_request = _make_event_deco('request')
|
|
|
|
|
|
class NoticeSession(BaseSession):
|
|
__slots__ = ()
|
|
|
|
def __init__(self, bot: NoneBot, event: CQEvent):
|
|
super().__init__(bot, event)
|
|
|
|
|
|
class RequestSession(BaseSession):
|
|
__slots__ = ()
|
|
|
|
def __init__(self, bot: NoneBot, event: CQEvent):
|
|
super().__init__(bot, event)
|
|
|
|
async def approve(self, remark: str = '') -> None:
|
|
"""
|
|
Approve the request.
|
|
|
|
:param remark: remark of friend (only works in friend request)
|
|
"""
|
|
try:
|
|
await self.bot.call_action(
|
|
action='.handle_quick_operation_async',
|
|
self_id=self.event.self_id,
|
|
context=self.event,
|
|
operation={'approve': True, 'remark': remark}
|
|
)
|
|
except CQHttpError:
|
|
pass
|
|
|
|
async def reject(self, reason: str = '') -> None:
|
|
"""
|
|
Reject the request.
|
|
|
|
:param reason: reason to reject (only works in group request)
|
|
"""
|
|
try:
|
|
await self.bot.call_action(
|
|
action='.handle_quick_operation_async',
|
|
self_id=self.event.self_id,
|
|
context=self.event,
|
|
operation={'approve': False, 'reason': reason}
|
|
)
|
|
except CQHttpError:
|
|
pass
|
|
|
|
|
|
async def handle_notice_or_request(bot: NoneBot, event: CQEvent) -> None:
|
|
if event.type == 'notice':
|
|
_log_notice(event)
|
|
session = NoticeSession(bot, event)
|
|
else: # must be 'request'
|
|
_log_request(event)
|
|
session = RequestSession(bot, event)
|
|
|
|
ev_name = event.name
|
|
logger.debug(f'Emitting event: {ev_name}')
|
|
try:
|
|
await _bus.emit(ev_name, session)
|
|
except Exception as e:
|
|
logger.error(f'An exception occurred while handling event {ev_name}:')
|
|
logger.exception(e)
|
|
|
|
|
|
def _log_notice(event: CQEvent) -> None:
|
|
logger.info(f'Notice: {event}')
|
|
|
|
|
|
def _log_request(event: CQEvent) -> None:
|
|
logger.info(f'Request: {event}')
|