計算字符串中的數(shù)學表達式stringExp = "2^4"intVal = int(stringExp) # Expected value: 16這將返回以下錯誤：Traceback (most recent call last): File "<stdin>", line 1, in <module>ValueError: invalid literal for int()with base 10: '2^4'我知道eval是否有一種更好、更安全的方法來計算存儲在字符串中的數(shù)學表達式？
3 回答
長風秋雁
TA貢獻1757條經驗 獲得超7個贊
eval是邪惡的
eval("__import__('os').remove('important file')") # arbitrary commands
eval("9**9**9**9**9**9**9**9", {'__builtins__': None}) # CPU, memory
注意：即使您使用SET__builtins__到None仍然有可能通過內省來爆發(fā)：
eval('(1).__class__.__bases__[0].__subclasses__()', {'__builtins__': None})
使用ast
import ast
import operator as op
# supported operators
operators = {ast.Add: op.add, ast.Sub: op.sub, ast.Mult: op.mul,
ast.Div: op.truediv, ast.Pow: op.pow, ast.BitXor: op.xor,
ast.USub: op.neg}
def eval_expr(expr):
"""
>>> eval_expr('2^6')
4
>>> eval_expr('2**6')
64
>>> eval_expr('1 + 2*3**(4^5) / (6 + -7)')
-5.0
"""
return eval_(ast.parse(expr, mode='eval').body)
def eval_(node):
if isinstance(node, ast.Num): # <number>
return node.n
elif isinstance(node, ast.BinOp): # <left> <operator> <right>
return operators[type(node.op)](eval_(node.left), eval_(node.right))
elif isinstance(node, ast.UnaryOp): # <operator> <operand> e.g., -1
return operators[type(node.op)](eval_(node.operand))
else:
raise TypeError(node)
您可以很容易地限制每個操作或任何中間結果的允許范圍，例如限制輸入參數(shù)a**b:
def power(a, b):
if any(abs(n) > 100 for n in [a, b]):
raise ValueError((a,b))
return op.pow(a, b)
operators[ast.Pow] = power
或者限制中間結果的大?。?/p>
import functools
def limit(max_=None):
"""Return decorator that limits allowed returned values."""
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
ret = func(*args, **kwargs)
try:
mag = abs(ret)
except TypeError:
pass # not applicable
else:
if mag > max_:
raise ValueError(ret)
return ret
return wrapper
return decorator
eval_ = limit(max_=10**100)(eval_)
例
>>> evil = "__import__('os').remove('important file')"
>>> eval_expr(evil) #doctest:+IGNORE_EXCEPTION_DETAIL
Traceback (most recent call last):
...
TypeError:
>>> eval_expr("9**9")
387420489
>>> eval_expr("9**9**9**9**9**9**9**9") #doctest:+IGNORE_EXCEPTION_DETAIL
Traceback (most recent call last):
...
ValueError:
慕無忌1623718
TA貢獻1744條經驗 獲得超4個贊
eval()sympy.sympify().evalf()*:
*sympify
警告：
請注意，此函數(shù)使用 eval，因此不應該用于未經消毒的輸入。
添加回答
舉報
0/150
提交
取消