mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
28 lines
808 B
Python
28 lines
808 B
Python
import random
|
|
from typing import Union, Sequence, Callable
|
|
|
|
from aiocqhttp import message
|
|
|
|
Expression_T = Union[str, Sequence[str], Callable]
|
|
|
|
|
|
def render(expr: Expression_T, *, escape_args=True,
|
|
**kwargs) -> str:
|
|
"""
|
|
Render an expression to message string.
|
|
|
|
:param expr: expression to render
|
|
:param escape_args: should escape arguments or not
|
|
:param kwargs: keyword arguments used in str.format()
|
|
:return: the rendered message
|
|
"""
|
|
if isinstance(expr, Callable):
|
|
expr = expr(**kwargs)
|
|
elif isinstance(expr, Sequence) and not isinstance(expr, str):
|
|
expr = random.choice(expr)
|
|
if escape_args:
|
|
for k, v in kwargs.items():
|
|
if isinstance(v, str):
|
|
kwargs[k] = message.escape(v)
|
|
return expr.format(**kwargs)
|