ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-31 18:19:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve message sending APIs

Browse Source
This commit is contained in:
Richard Chien
2018-12-23 20:20:40 +08:00
parent 7cb634c77a
commit 7dff9f7fd5
4 changed files with 35 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
none/helpers.py
View File

@ -44,22 +44,17 @@ def context_id(ctx: Context_T, *,
return ctx_id
async def send(bot: NoneBot, ctx: Context_T, message: Message_T, *,
ignore_failure: bool = True) -> None:
async def send(bot: NoneBot, ctx: Context_T,
message: Message_T, *,
ensure_private: bool = False,
ignore_failure: bool = True,
**kwargs) -> None:
"""Send a message ignoring failure by default."""
try:
if ctx.get('post_type') == 'message':
await bot.send(ctx, message)
else:
if ensure_private:
ctx = ctx.copy()
if 'message' in ctx:
del ctx['message']
if 'group_id' in ctx:
await bot.send_group_msg(**ctx, message=message)
elif 'discuss_id' in ctx:
await bot.send_discuss_msg(**ctx, message=message)
elif 'user_id' in ctx:
await bot.send_private_msg(**ctx, message=message)
ctx['message_type'] = 'private'
await bot.send(ctx, message, **kwargs)
except CQHttpError:
if not ignore_failure:
raise