Python子过程 /环境变量脚本

Question

我有一个编译的fortran程序（在〜/codedir中）链接到Intel MKL，我想在某个目录中使用Python脚本运行（假设calculationDir> ）。 我可以导航到calculationDir并执行〜/codedir/code，其中代码是可执行文件的名称。一切都起作用。但是，如果我尝试从下面的Python脚本启动代码：

import subprocess

command = '~/codedir/code'

with open('job.out', 'a') as f:
    process = subprocess.Popen(command.split(), stdout=f, shell=True)
    output, error = process.communicate()

我会收到以下错误：

/home/codedir/code: error while loading shared libraries: libmkl_intel_lp64.so.1: cannot open shared object file: No such file or directory

我怀疑这可能与环境变量有关。我配置了我的外壳，每次启动shell时，都执行了一个Intel One API的脚本setVars.shsevars.sh，该API设置了很多东西。如果我使用Python的子过程，这些变量是否可以设置？我是否必须以某种方式告诉子过程才能执行setVars-script？

Answer 1

您似乎正在寻找

import subprocess
import shlex

with open('job.out', 'a') as f:
    subprocess.run(
        shlex.split(command),
        cwd=calculationdir,
        stdout=f, check=True)

There is no need to specify shell=True ，事实上，当您自己拆分命令时，在类 Unix 平台上这样做是错误的。 （Windows 上的语义略有不同，因此虽然它仍然是一个错误，但症状可以忽略不计。）另请参阅 subprocess 中 shell=True 的实际含义

split 是错误的，因为它不知道如何实现应付反斜杠转义和引用；如果您的命令当前不包含这些构造中的任何一个，您可以使用基本字符串 split;但标准库提供了一个无论如何都能正常工作的函数，我认为没有理由不在这里使用 shlex.split 。

正如文档所解释的，如果可以的话，您应该更喜欢 subprocess.run 而不是普通的 Popen，因为后者需要您复制/粘贴几行样板代码，即使这样除非您确切地知道自己在做什么，否则不太健壮，可能会以添加更多用于错误处理等的样板代码为代价。

最后，真正的牛肉：关键字参数cwd允许您设置工作子进程的目录。