如何从python中另一个脚本中的函数收集返回值？

Question

 fw_test.py

 def _testmethod_():
     x = []
     y = 0
     while y !=5:
        x.append(y)
        y +=1
        return x

 t = _testmethod_()

 main_test.py

 import subprocess
 p = subprocess.call(['python', 'main_test.py'])

由于指导方针，我无法将fw_test.py 导入main_test.py。我希望能够将_testmethod_() 从fw_test.py 返回的值存储在变量inmain_test.py 中。我了解到使用子进程我可以运行fw_test.py, but that is not enough for me.有没有办法解决这个问题？

编辑：fw_test.py 不能导入到main_test.py 的原因是有很多像fw_test.py 这样的测试脚本会根据测试不断变化。 main_test.py 应该是一个通用框架，它具有通过读取fw_test.py（返回值为True/False）的返回值来评估测试通过或失败的基本功能。我认为，如果我在文件顶部导入脚本，它将不再是通用的。我愿意接受其他建议。

为什么投反对票？

Answer 1

您可以使用subprocess.check_output 获取另一个脚本编写的输出

文件“b.py”将其输出写入标准输出

print("hi")
print(1)

文件“a.py”作为子进程执行“b.py”

import subprocess
a = subprocess.check_output(["python", "/tmp/b.py"])
print(a.decode('utf-8'))

注意：check_output返回的值是一个字节数组，需要解码转成字符串

从命令行执行“a.py”

$ python a.py
hi
1

免责声明：这是给定（用于训练）问题的最简单解决方案（不是最好的）。从技术上讲，我将输出打印到标准输出并在父进程中捕获它。在更严重的问题上解决同样的问题应该使用IPC、RPC等机制。

Answer 2

如果不能使用import fw_test语句；你可以use importlib.import_module() instead，从fw_test.testmethod()函数中获取返回值：

import importlib

fw_test = importlib.import_module('fw_test')
value = fw_test.testmethod()

Answer 3

最简单的方法可能是将返回值放到一个文件中，然后让你的 main 去拾取它。

fw_test.py

def _testmethod_():
    x = []
    y = 0
    while y !=5:
       x.append(y)
       y +=1
       return x

t = _testmethod_()

with open("fileToStoreValue.txt", 'w') as f:
    f.write(t)

main_test.py

import subprocess
p = subprocess.call(['python', 'main_test.py'])
t = eval(open("fileToStoreValue.txt", 'r').read().strip())

请注意，我在阅读时包含了strip，以防止您拾取任何尾随的新行。

另外请记住，如果两个 python 文件都在同一个目录中，这将按照所写的方式工作。如果它们的位置不同，则需要在 open 语句中指定 fileToSTtoreValue.txt 的路径。