mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-12-01 01:25:07 +08:00
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
from functools import update_wrapper
|
|
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
|
|
|
|
|
|
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}')
|