PythonREPL是python提供的 “Read–Eval–Print Loop”(读取-求值-打印-循环)的缩写,它是一种简单的、交互式的编程环境,我们之前在用AI生成代码后执行就使用这个方法,例如
def py_repl_tool(self, realcode: str):
py_repl = PythonREPL()
...
res = py_repl.run(realcode)
之前使用的挺好,但最近,生成带函数的代码时,就出现了执行报错:
Python REPL can execute arbitrary code. Use with caution.
NameError(“name ‘getReportDataLocal’ is not defined”)
需要执行的代码如下:
from akshare import index_stock_cons
from pythonRepository import getReportDataLocal, CalAnnualRatio
def genHS300EPS(report_date):
# 获取指定日期的业绩报表
df_report = getReportDataLocal(report_date)
# 计算前一年的报表日期
......
# 打印并返回处理后的stock_list
print(stock_list)
return stock_list
# 示例调用
print(genHS300EPS("20231231"))
代码本身是完整且正确的
但执行结果报错
查了一下报错信息,发现这种报错是,没有引入import库和方法,从而提示X方法无定义。 我看一下PythonREPL()方法,发现实际执行的是exec(command, globals, locals)
我搜索后发现,猜想可能是全局变量和局部变量作用域的相关问题,于是尝试自己写一个exec(command,globals)函数
def py_repl_tool_1(self, realcode: str):
# py_repl = PythonREPL()
...
# res = py_repl.run(realcode)
try:
result[i] = exec(realcode, globals())
print('执行成功')
except Exception as e:
print("执行失败: %s, %s" % e % result[i])
# print(traceback.format_exception(e))
traceback.print_exc()
成功了