我有一个脚本a.py
,在执行它时会向用户询问某些查询并以json格式对输出进行帧化。使用python子进程,我可以从名为b.py
的另一个脚本调用此脚本。一切都按预期工作,除了我无法获得变量的输出?我在Python 3中这样做。
答案 0 :(得分:4)
使用subprocess
模块从另一个调用Python脚本并传递一些输入并获取其输出:
#!/usr/bin/env python3
import os
import sys
from subprocess import check_output
script_path = os.path.join(get_script_dir(), 'a.py')
output = check_output([sys.executable, script_path],
input='\n'.join(['query 1', 'query 2']),
universal_newlines=True)
其中get_script_dir()
function is defined here。
更灵活的替代方法是导入模块a
并调用函数以获取结果(确保a.py
使用if __name__=="__main__"
保护,以避免在导入时运行不需要的代码) :
#!/usr/bin/env python
import a # the dir with a.py should be in sys.path
result = [a.search(query) for query in ['query 1', 'query 2']]
您可以使用mutliprocessing
在单独的进程中运行每个查询(如果执行查询是CPU密集型的,那么它可能会提高时间性能):
#!/usr/bin/env python
from multiprocessing import freeze_support, Pool
import a
if __name__ == "__main__":
freeze_support()
pool = Pool() # use all available CPUs
result = pool.map(a.search, ['query 1', 'query 2'])
答案 1 :(得分:1)
除了提到的另一种方法是使用内置函数exec
这个函数获取一串python代码并执行它
要在脚本文件中使用它,您可以简单地read
将其作为文本文件,如下所示:
#dir is the directory of a.py
#a.py, for example, contains the variable 'x=1'
exec(open(dir+'\\a.py').read())
print(x) #outputs 1