mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-12-16 13:40:53 +08:00
63 lines
2.8 KiB
Python
63 lines
2.8 KiB
Python
|
import re
|
|||
|
|
import aiohttp
|
|||
|
|
from ._language_dict import search_dict, suffix_list
|
|||
|
|
|
|||
|
|
GLOT_TOKEN = "2b42fd59-563f-4855-870d-f564c4e407b5"
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def run(command: str) -> str:
|
|||
|
|
# "py -i 50,100,3 import random" -> {'language_type': 'py', 'stdin': '50,100,3', 'code': 'import random'}
|
|||
|
|
match_obj = re.match(r"^(?P<language_type>[^ \n]+) ?(-i *(?P<stdin>[^ \n]+))?[\n ]+(?P<code>[\w\W]+)", command)
|
|||
|
|
code_type = match_obj.group("language_type").strip()
|
|||
|
|
if code_type not in search_dict: # 如果发送的语言类型不在查询字典里面
|
|||
|
|
return f"输入有误, www找不到{code_type}, 请检查您的输入"
|
|||
|
|
code = match_obj.group("code").strip() # 获取code的字段
|
|||
|
|
if len(code) == 0: # 如果code为空
|
|||
|
|
return "运行代码不能为空~"
|
|||
|
|
stdin = match_obj.group("stdin") # 获取stdin的输入
|
|||
|
|
type_name = search_dict[code_type] # 从查询字典中拿到运行代码的完整语言名
|
|||
|
|
|
|||
|
|
# 请求部分
|
|||
|
|
header = {
|
|||
|
|
"Authorization": f"Token {GLOT_TOKEN}",
|
|||
|
|
"Content-type": "application/json"
|
|||
|
|
}
|
|||
|
|
data = {
|
|||
|
|
"files": [
|
|||
|
|
{
|
|||
|
|
'name': f'main.{suffix_list[type_name]}', # 固定文件名的话用户在处理java类文件的时候类名可能会不方便, 需要改进
|
|||
|
|
'content': code
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
"stdin": stdin if stdin is not None else "",
|
|||
|
|
"command": ""
|
|||
|
|
}
|
|||
|
|
try:
|
|||
|
|
async with aiohttp.TCPConnector(ssl=False) as connector:
|
|||
|
|
async with aiohttp.request("POST", headers=header,
|
|||
|
|
json=data,
|
|||
|
|
url=f"https://glot.io/api/run/{type_name}/latest",
|
|||
|
|
connector=connector
|
|||
|
|
) as resp:
|
|||
|
|
rep_json = await resp.json() # 获取响应
|
|||
|
|
if 'message' in rep_json: # 如果message在里面的话基本上都是请求失败了, 运行时间过长或者api出错
|
|||
|
|
return rep_json['message']
|
|||
|
|
|
|||
|
|
result = "\n"
|
|||
|
|
stdout = rep_json['stdout'].rstrip()
|
|||
|
|
error = rep_json["error"].rstrip()
|
|||
|
|
stderr = rep_json["stderr"].rstrip()
|
|||
|
|
if stdout:
|
|||
|
|
result += f"{stdout}\n"
|
|||
|
|
if stderr:
|
|||
|
|
result += f"{stderr}\n"
|
|||
|
|
if error:
|
|||
|
|
result += f"{error}\n"
|
|||
|
|
if len(result) > 500 or result.count("\n") > 15:
|
|||
|
|
return f"输出结果过长，仅显示前500：\n{result[0:500]}"
|
|||
|
|
return result
|
|||
|
|
except Exception as e: # 网络请求失败时走该分支
|
|||
|
|
from . import sv
|
|||
|
|
sv.logger.error(f"{e} occurred when request l: {code_type} c:{code}")
|
|||
|
|
return f"请求失败...请联系维护者"
|