Fix handling on negative numbers in ast.literal_eval().

This commit is contained in:
Raymond Hettinger 2010-10-08 00:47:45 +00:00
parent 4f3abb0f09
commit bc95973b51
3 changed files with 20 additions and 8 deletions

View file

@ -65,19 +65,25 @@ def literal_eval(node_or_string):
elif isinstance(node, Name):
if node.id in _safe_names:
return _safe_names[node.id]
elif isinstance(node, UnaryOp) and \
isinstance(node.op, (UAdd, USub)) and \
isinstance(node.operand, (Num, UnaryOp, BinOp)):
operand = _convert(node.operand)
if isinstance(node.op, UAdd):
return + operand
else:
return - operand
elif isinstance(node, BinOp) and \
isinstance(node.op, (Add, Sub)) and \
isinstance(node.right, Num) and \
isinstance(node.right.n, complex) and \
isinstance(node.left, Num) and \
isinstance(node.left.n, (int, float)):
left = node.left.n
right = node.right.n
isinstance(node.right, (Num, UnaryOp, BinOp)) and \
isinstance(node.left, (Num, UnaryOp, BinOp)):
left = _convert(node.left)
right = _convert(node.right)
if isinstance(node.op, Add):
return left + right
else:
return left - right
raise ValueError('malformed string')
raise ValueError('malformed node or string: ' + repr(node))
return _convert(node_or_string)