diff options
| author | 0w0 <[email protected]> | 2022-12-29 16:38:27 +0800 |
|---|---|---|
| committer | GitHub <[email protected]> | 2022-12-29 16:38:27 +0800 |
| commit | c0e5f0398bd4c3c55378cff303f31ff968822346 (patch) | |
| tree | d90bc810150d859c76e64f6e71d8524478014f31 /ATRI/plugins/code_runner | |
| parent | 3d857da2ae1984c6abf286b55888d6d3e4c5d823 (diff) | |
| parent | bb13b38f862d859007d1d37941f5b5c9d5f38cbc (diff) | |
| download | ATRI-c0e5f0398bd4c3c55378cff303f31ff968822346.tar.gz ATRI-c0e5f0398bd4c3c55378cff303f31ff968822346.tar.bz2 ATRI-c0e5f0398bd4c3c55378cff303f31ff968822346.zip | |
Merge branch 'main' into main
Diffstat (limited to 'ATRI/plugins/code_runner')
| -rw-r--r-- | ATRI/plugins/code_runner/__init__.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ATRI/plugins/code_runner/__init__.py b/ATRI/plugins/code_runner/__init__.py index 26fcce5..95d3b0c 100644 --- a/ATRI/plugins/code_runner/__init__.py +++ b/ATRI/plugins/code_runner/__init__.py @@ -6,6 +6,7 @@ from nonebot.adapters.onebot.v11 import MessageEvent, Message, MessageSegment, u from nonebot.adapters.onebot.v11.helpers import Cooldown from ATRI.service import Service +from ATRI.utils import MessageChecker from .data_source import CodeRunner @@ -26,7 +27,7 @@ async def _code_runner(matcher: Matcher, args: Message = CommandArg()): matcher.set_arg("opt", args) else: content = "请键入 /code.help 以获取帮助~！" - await code_runner.finish(Message(content)) + await code_runner.finish(content) @code_runner.got("opt", prompt="需要运行的语言及代码？\n获取帮助：/code.help") @@ -38,6 +39,9 @@ async def _(event: MessageEvent, opt: str = ArgPlainText("opt")): await code_runner.finish(CodeRunner().help()) content = MessageSegment.at(user_id) + str(await CodeRunner().runner(unescape(opt))) + is_save = MessageChecker(str(content)).check_cq_code + if not is_save: + await code_runner.finish("有潜在的风险, 不予发送") await code_runner.finish(Message(content)) |