nonebot2/none/helpers.py

from typing import Dict, Any, Union, List, Sequence, Callable

from aiocqhttp import Error as CQHttpError

from . import NoneBot, 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: NoneBot, 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: NoneBot, 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))