from typing import Dict, Any, Optional, Callable, Union from aiocqhttp import CQHttp, Error as CQHttpError from aiocqhttp.bus import EventBus from .session import BaseSession from .log import logger _bus = EventBus() def _make_event_deco(post_type: str) -> Callable: def deco_deco(arg: Optional[Union[str, Callable]] = None, *events: str) -> Callable: def deco(func: Callable) -> Callable: if isinstance(arg, str): for e in [arg] + list(events): _bus.subscribe(f'{post_type}.{e}', func) else: _bus.subscribe(post_type, func) return func if isinstance(arg, Callable): return deco(arg) 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: CQHttp, ctx: Dict[str, Any]): super().__init__(bot, ctx) class RequestSession(BaseSession): __slots__ = () def __init__(self, bot: CQHttp, ctx: Dict[str, Any]): super().__init__(bot, ctx) async def approve(self, remark: str = ''): # TODO: should use ".handle_quick_operation" action in the future try: if self.ctx['request_type'] == 'friend': await self.bot.set_friend_add_request(**self.ctx, approve=True, remark=remark) elif self.ctx['request_type'] == 'group': await self.bot.set_group_add_request(**self.ctx, type=self.ctx['sub_type'], approve=True) except CQHttpError: pass async def reject(self, reason: str = ''): # TODO: should use ".handle_quick_operation" action in the future try: if self.ctx['request_type'] == 'friend': await self.bot.set_friend_add_request(**self.ctx, approve=False) elif self.ctx['request_type'] == 'group': await self.bot.set_group_add_request(**self.ctx, type=self.ctx['sub_type'], approve=False, reason=reason) except CQHttpError: pass async def handle_notice_or_request(bot: CQHttp, ctx: Dict[str, Any]) -> None: post_type = ctx['post_type'] # "notice" or "request" detail_type = ctx[f'{post_type}_type'] event = f'{post_type}.{detail_type}' if ctx.get('sub_type'): event += f'.{ctx["sub_type"]}' if post_type == 'notice': session = NoticeSession(bot, ctx) else: session = RequestSession(bot, ctx) logger.debug(f'Emitting event: {event}') await _bus.emit(event, session)