100 lines
4.3 KiB
Python
100 lines
4.3 KiB
Python
from .constant import consts
|
||
from core.exceptions import NoReportException
|
||
from core.builtins.message import MessageSession
|
||
from core.component import on_command
|
||
from simpleeval import InvalidExpression, SimpleEval, DEFAULT_FUNCTIONS, DEFAULT_NAMES, DEFAULT_OPERATORS
|
||
import ast
|
||
import operator as op
|
||
import asyncio
|
||
import math
|
||
import statistics
|
||
from typing import Callable, Any
|
||
|
||
|
||
def func_wrapper(func: Callable[..., Any], *args, **kwargs):
|
||
for arg in args:
|
||
if isinstance(arg, (int, float)) and arg > 100000:
|
||
raise NoReportException('参数过大，无法计算。')
|
||
return func(*args)
|
||
|
||
|
||
funcs = {}
|
||
|
||
|
||
def add_func(module):
|
||
for name in dir(module):
|
||
item = getattr(module, name)
|
||
if not name.startswith('_') and callable(item):
|
||
funcs[name] = lambda *args, item = item, **kwargs: func_wrapper(
|
||
item, *args, **kwargs)
|
||
|
||
|
||
add_func(math)
|
||
add_func(statistics)
|
||
|
||
s_eval = SimpleEval(
|
||
operators={
|
||
**DEFAULT_OPERATORS,
|
||
ast.Pow: lambda *args, item=op, **kwargs: func_wrapper(
|
||
item, *args, **kwargs),
|
||
ast.BitOr: op.or_,
|
||
ast.BitAnd: op.and_,
|
||
ast.BitXor: op.xor,
|
||
ast.Invert: op.invert,
|
||
},
|
||
functions={**funcs, **DEFAULT_FUNCTIONS},
|
||
names={
|
||
**DEFAULT_NAMES, **consts,
|
||
'pi': math.pi,
|
||
'e': math.e,
|
||
'tau': math.tau,
|
||
'inf': math.inf, 'nan': math.nan,
|
||
},)
|
||
|
||
c = on_command('calc', developers=[
|
||
'Dianliang233'], desc='安全地计算 Python ast 表达式。', required_superuser=True)
|
||
|
||
|
||
@c.handle('<math_expression>', options_desc={'+': '和/正数：1 + 2 -> 3',
|
||
'-': '差/负数：3 - 1 -> 2',
|
||
'/': '商：6 / 3 -> 2',
|
||
'//': '整除：7 // 4 -> 1',
|
||
'*': '积：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',
|
||
'not': '非：not True -> False',
|
||
'is': 'x 与 y 是同一个对象：1 is 1 -> True',
|
||
'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"',
|
||
'更多数学函数': 'https://docs.python.org/zh-cn/3/library/math.html',
|
||
'更多统计函数': 'https://docs.python.org/zh-cn/3/library/statistics.html',
|
||
})
|
||
async def _(msg: MessageSession):
|
||
try:
|
||
res = await asyncio.wait_for(async_eval(msg.parsed_msg["<math_expression>"]), 15)
|
||
await msg.finish(f'{(msg.parsed_msg["<math_expression>"])} = {str(res)}')
|
||
except InvalidExpression as e:
|
||
await msg.finish(f"表达式无效：{e}")
|
||
except asyncio.TimeoutError:
|
||
raise TimeoutException()
|
||
except Exception as e:
|
||
raise NoReportException(e)
|
||
|
||
|
||
async def async_eval(expr: str):
|
||
return s_eval.eval(expr)
|
||
|
||
|
||
class TimeoutException(NoReportException):
|
||
'''计算超时，最大计算时间为 15 秒。'''
|
||
pass
|