У меня есть код из этого вопроса: Почему eval() Python отклоняет эту многострочную строку и как я могу это исправить?
def multiline_eval(expr, context = {}):
"Evaluate several lines of input, returning the result of the last line"
tree = ast.parse(expr)
eval_exprs = []
exec_exprs = []
for module in tree.body:
if isinstance(module, ast.Expr):
eval_exprs.append(module.value)
else:
exec_exprs.append(module)
exec_expr = ast.Module(exec_exprs, type_ignores=[])
exec(compile(exec_expr, 'file', 'exec'), context)
results = []
for eval_expr in eval_exprs:
results.append(eval(compile(ast.Expression((eval_expr)), 'file', 'eval'), context))
return '\n'.join([str(r) for r in results])
Когда я использую этот код:
multiline_eval('''
print("Hello World1")
for a in range(5):
print(a)
print("Hello World2")
''')
Результат:
0
1
2
3
4
Hello World1
Hello World2
Я ожидаю этого:
Hello World1
0
1
2
3
4
Hello World2
Как я могу изменить код? Я пытался изменить код, но мне это не удалось.
Код, который вы опубликовали, сортирует тело дерева ast по 2 ведрам, ast.Expr
s и остальным, а затем обрабатывает их по ведрам. Итак, вам нужно удалить «ковш». Вы можете попробовать что-то вроде следующего:
import ast
def multiline_eval(expr, ctx = {}):
results = []
for node in ast.parse(expr).body:
if isinstance(node, ast.Expr):
result = eval(compile(ast.Expression(node.value), '<string>', 'eval'), ctx)
results.append(result)
else:
module = ast.Module([node], type_ignores=[])
results.append(exec(compile(module, '<string>', 'exec'), ctx))
return '\n'.join(map(str, results))