ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-27 16:21:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add approve and reject method to RequestSession

Browse Source
This commit is contained in:
Richard Chien
2018-06-30 18:26:48 +08:00
parent 0c6468fa70
commit 4c07626124
2 changed files with 35 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
none/notice_request.py
View File

@ -1,6 +1,6 @@
from typing import Dict, Any, Optional, Callable, Union
from aiocqhttp import CQHttp
from aiocqhttp import CQHttp, Error as CQHttpError
from aiocqhttp.bus import EventBus
from .session import BaseSession
@ -44,7 +44,33 @@ class RequestSession(BaseSession):
def __init__(self, bot: CQHttp, ctx: Dict[str, Any]):
super().__init__(bot, ctx)
# TODO: 添加 approve、deny 等方法
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: