subprocess - 使用 subprocess.check_output Python 的环境变量

Question

我正在尝试使用 Python 在我的服务器上进行一些基本的模块设置。这有点困难，因为我无法访问互联网。

这是我的代码

import sys
import os
from subprocess import CalledProcessError, STDOUT, check_output

def run_in_path(command, dir_path, env_var=''):
    env_var = os.environ["PATH"] = os.environ["PATH"] + env_var
    print(env_var)
    try:
        p = check_output(command, cwd=dir_path, stderr=STDOUT)
    except CalledProcessError as e:
        sys.stderr.write(e.output.decode("utf-8"))
        sys.stderr.flush()
        return e.returncode
    else:
        return 0

def main():
    requests_install = run_in_path('python setup.py build', 
                'D:\installed_software\python modules\kennethreitz-requests-e95e173')
    SQL_install = run_in_path('python setup.py install', # install SQL module pypyodbc
                   'D:\installed_software\python modules\pypyodbc-1.3.3\pypyodbc-1.3.3')
    setup_tools = run_in_path('python setup.py install', # install setup tools
                   'D:\installed_software\python modules\setuptools-17.1.1')
    psycopg2 = run_in_path('easy_install psycopg2-2.6.1.win-amd64-py3.3-pg9.4.4-release', # install setup tools
                   'D:\installed_software\python modules', ';C:\srv_apps\Python33\Scripts\easy_install.exe')
    print('setup complete')

if __name__ == "__main__":
    sys.exit(main())

现在当我开始尝试使用简易安装时它变得棘手了。我的 subprocess.check_output 调用似乎没有使用我的环境变量

File "C:\srv_apps\Python33\lib\subprocess.py", line 1110, in _execute_child
    raise WindowsError(*e.args)
FileNotFoundError: [WinError 2] The system cannot find the file specified

我不想升级到默认安装了 easy install 的 3.4，因为 3.4 不支持我的其他模块。我的主要挑战是 subprocess.check_call 方法不将环境变量作为输入，并且我对尝试使用 Popen() 持谨慎态度，因为我过去从未真正让它成功运行过。任何帮助将不胜感激。

Answer 1

PATH 应该包含目录，例如，r'C:\Python33\Scripts'，而不是像这样的文件:r' C:\Python33\Scripts\easy_install.exe'

不要为任意命令硬编码 utf-8，您可以使用 universal_newlines 参数启用文本模式(未测试):

#!/usr/bin/env python3
import locale
import sys
from subprocess import CalledProcessError, STDOUT, check_output

def run(command, *, cwd=None, env=None):
    try:
        ignored = check_output(command, cwd=cwd, env=env, 
                               stderr=STDOUT,
                               universal_newlines=True)
    except CalledProcessError as e:
        sys.stderr.write(e.output)
        sys.stderr.flush()
        return e.returncode
    else:
        return 0

例子:

import os

path_var = os.pathsep.join(os.environ.get('PATH', os.defpath), some_dir)
env = dict(os.environ, PATH=path_var)
run("some_command", cwd=some_path, env=env)
run("another_command", cwd=another_path, env=env)