nonebot2/none/helpers.py
2018-07-01 20:01:05 +08:00

53 lines
1.6 KiB
Python

from typing import Dict, Any, Union, List, Sequence, Callable
from aiocqhttp import CQHttp, Error as CQHttpError
from . import expression
def context_id(ctx: Dict[str, Any]) -> str:
"""
Calculate a unique id representing the current user.
"""
src = ''
if ctx.get('group_id'):
src += f'/group/{ctx["group_id"]}'
elif ctx.get('discuss_id'):
src += f'/discuss/{ctx["discuss_id"]}'
if ctx.get('user_id'):
src += f'/user/{ctx["user_id"]}'
return src
async def send(bot: CQHttp, ctx: Dict[str, Any],
message: Union[str, Dict[str, Any], List[Dict[str, Any]]],
*, ignore_failure: bool = True) -> None:
"""
Send a message ignoring failure by default.
"""
try:
if ctx.get('post_type') == 'message':
await bot.send(ctx, message)
else:
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)
except CQHttpError:
if not ignore_failure:
raise
async def send_expr(bot: CQHttp, ctx: Dict[str, Any],
expr: Union[str, Sequence[str], Callable],
**kwargs):
"""
Sending a expression message ignoring failure by default.
"""
return await send(bot, ctx, expression.render(expr, **kwargs))