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

Allow positional arguments while rendering expressions

Browse Source
This commit is contained in:
Richard Chien
2019-02-01 19:39:09 +08:00
parent 7860c02973
commit 9190b65b6d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nonebot/helpers.py
View File

@ -63,22 +63,23 @@ async def send(bot: NoneBot, ctx: Context_T,
return None
def render_expression(expr: Expression_T, *,
def render_expression(expr: Expression_T, *args,
escape_args: bool = True, **kwargs) -> str:
"""
Render an expression to message string.
:param expr: expression to render
:param escape_args: should escape arguments or not
:param args: positional arguments used in str.format()
:param kwargs: keyword arguments used in str.format()
:return: the rendered message
"""
if isinstance(expr, Callable):
expr = expr(**kwargs)
expr = expr(*args, **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] = escape(v)
return expr.format(**kwargs)
return expr.format(*args, **kwargs)