2023-01-19 15:10:08 +00:00
|
|
|
|
import os
|
|
|
|
|
|
2023-01-19 11:27:31 +00:00
|
|
|
|
from core.exceptions import NoReportException
|
2023-01-17 02:59:22 +00:00
|
|
|
|
from core.builtins.message import MessageSession
|
|
|
|
|
from core.component import on_command
|
2023-01-19 11:52:05 +00:00
|
|
|
|
|
2023-01-19 17:13:11 +00:00
|
|
|
|
import asyncio
|
2023-01-19 07:28:01 +00:00
|
|
|
|
|
2023-01-17 02:59:22 +00:00
|
|
|
|
c = on_command('calc', developers=[
|
2023-01-19 15:10:08 +00:00
|
|
|
|
'Dianliang233'], desc='安全地计算 Python ast 表达式。')
|
2023-01-17 02:59:22 +00:00
|
|
|
|
|
|
|
|
|
|
2023-01-19 08:55:12 +00:00
|
|
|
|
@c.handle('<math_expression>', options_desc={'+': '和/正数:1 + 2 -> 3',
|
|
|
|
|
'-': '差/负数:3 - 1 -> 2',
|
2023-01-17 02:59:22 +00:00
|
|
|
|
'/': '商:6 / 3 -> 2',
|
2023-01-19 08:55:12 +00:00
|
|
|
|
'//': '整除:7 // 4 -> 1',
|
2023-01-17 02:59:22 +00:00
|
|
|
|
'*': '积:2 * 3 -> 6',
|
|
|
|
|
'**': 'x 的 y 次幂(由于性能问题,结果不得超过 4e+6):2 ** 3 -> 8',
|
|
|
|
|
'%': '取模:5 % 2 -> 1',
|
|
|
|
|
'==': '等于:1 == 1 -> True',
|
|
|
|
|
'<': '小于:1 < 2 -> True',
|
|
|
|
|
'>': '大于:2 > 1 -> True',
|
|
|
|
|
'<=': '小于等于:1 <= 2 -> True',
|
|
|
|
|
'>>': 'x 右移 y 位(相当于 x / (2 ** y),y < 10000):32 >> 5 -> 1',
|
|
|
|
|
'<<': 'x 左移 y 位(相当于 x * (2 ** y),y < 10000):1 << 5 -> 32',
|
|
|
|
|
'in': 'x 在 y 中:"hat" in "what" -> True',
|
2023-01-19 08:55:12 +00:00
|
|
|
|
'not': '非:not True -> False',
|
|
|
|
|
'is': 'x 与 y 是同一个对象:1 is 1 -> True',
|
2023-01-17 02:59:22 +00:00
|
|
|
|
'randint(x)': '小于 x 的随机整数:randint(6) -> 5',
|
|
|
|
|
'rand()': '0 与 1 之间的随机浮点数:rand() -> 0.5789015836448923',
|
|
|
|
|
'int()': '转换为整数:int(1.5) -> 1',
|
|
|
|
|
'float()': '转换为浮点数:float(1) -> 1.0',
|
|
|
|
|
'str()': '转换为字符串:str(1) -> "1"',
|
2023-01-19 11:52:05 +00:00
|
|
|
|
'更多数学函数': 'https://docs.python.org/zh-cn/3/library/math.html',
|
|
|
|
|
'更多统计函数': 'https://docs.python.org/zh-cn/3/library/statistics.html',
|
2023-01-17 02:59:22 +00:00
|
|
|
|
})
|
|
|
|
|
async def _(msg: MessageSession):
|
|
|
|
|
try:
|
2023-01-19 17:13:11 +00:00
|
|
|
|
p = await asyncio.create_subprocess_shell(" ".join(
|
|
|
|
|
['python', os.path.abspath("./modules/calc/calc.py"), msg.parsed_msg["<math_expression>"]]),
|
|
|
|
|
stdout=asyncio.subprocess.PIPE,
|
|
|
|
|
stderr=asyncio.subprocess.PIPE,
|
|
|
|
|
)
|
|
|
|
|
try:
|
|
|
|
|
await asyncio.wait_for(p.wait(), timeout=10)
|
|
|
|
|
except asyncio.TimeoutError:
|
|
|
|
|
p.kill()
|
|
|
|
|
raise NoReportException('计算超时。')
|
|
|
|
|
stdout_data, stderr_data = await p.communicate()
|
|
|
|
|
if p.returncode == 0:
|
|
|
|
|
res = stdout_data.decode('utf-8')
|
|
|
|
|
|
|
|
|
|
if res[0:6] == 'Result':
|
|
|
|
|
await msg.finish(f'{(msg.parsed_msg["<math_expression>"])} = {res[7:]}')
|
|
|
|
|
else:
|
|
|
|
|
await msg.finish(f'表达式无效:{res[7:]}')
|
2023-01-19 07:28:01 +00:00
|
|
|
|
except Exception as e:
|
|
|
|
|
raise NoReportException(e)
|