具有预填充环境的 Python 子进程模块

Question

问题可以与Use python subprocess module like a command line simulator相关

我编写了一些名为 my_shell 的基础结构代码，您可以向其中传递我的应用程序的 shell 命令，如下所示

class ApplicationTestShell(object):
    def __init__(self):
        '''
        Constructor
        '''
        self.play_ground_dir = "/var/tmp/MyAppDir" 
        ensure_dir_exists_and_empty(self.play_ground_dir)

    def execute_command(self, command, on_success = None, on_failure = None):
        p = create_shell_process(self, self.play_ground_dir)
        sout, serr = p.communicate(input = command)
        if p.returncode == 0:
            on_success(sout)
        else:
            on_failure(serr)

    def create_shell_process(self, cwd):
        return Popen("/bin/bash", env= {WHAT DO I DO HERE?},cwd = test_dir, stdout=PIPE, stderr=PIPE, stdin=PIPE)

这里对我来说有趣的是 env 参数。 Python 期望所有环境变量都有一个“映射”数据结构。我的应用程序需要导出和设置多个变量。用于设置和导出的脚本是通过运行“/bin/appload myapp”生成的(假设 appload 在路径上始终可用)。我现在做什么当我调用 p.communicate 时，我执行以下操作

p.communicate(input = "eval `/bin/appload myapp`;" + command)

所以基本上在运行命令之前我调用基础设施设置。

1. 有没有办法在 Python 中以更好的方式做到这一点。我想以某种方式将 eval/bin/appload 部分推送到 Popen 类上的 env 参数，或者作为 shell 创建过程的一部分。
2. 我当前的实现存在哪些问题？ (我觉得这很hacky，但我可能是错的)

Answer 1

1. 这取决于 /bin/appload myapp 的工作方式。如果它只保证输出 bash 语法，然后在 Python 中解析该输出以构造环境对象，那么几乎肯定会比它值得的麻烦更多(您可能需要支持参数和变量扩展、子 shell、进程替换等) ， ETC)。另一方面，如果您确定 /bin/appload myapp 只会输出“VARIABLENAME=someword”形式的行，那么解析起来就非常简单了Python，如果您愿意，您可以将其移至您的 Python 代码中。

   满足这些要求有很多不同的方向；您可以将 appload myapp 的输出捕获到临时文件中，并将子进程的 $BASH_ENV 设置为该文件名；这将导致 shell 在运行命令之前以某些人可能认为更干净的方式获取环境设置。您可以将命令(带有 eval-ing 前缀)作为 Popen 的第一个参数并传递 shell=True，然后让 Popen 自行执行 bash 调用(如有必要，将 $SHELL 显式设置为 bash)。您可以使用 bash 的 -c 选项来指定在命令行上运行的代码，而不是通过 stdin 运行。您可以通过从 Python 调用 shell 来采用多层方法，其中 eval 是 appload myapp 环境，然后 exec 在其下面的另一个 shell，这样第一个 shell 就不会出现在 ps 列表和命令中给 create_shell_process 的 shell 完全属于它自己(尽管这并不重要)。您可以做很多事情，具体取决于您对 shell 的调用方式、它在 ps 列表中的外观的关注点，以及您是否希望在 appload myapp 时仍然运行命令。输出在评估等时会产生错误。但对于通用解决方案，我认为您所拥有的完全没问题。

2. 除了装饰性的事情或可能仅来自复制和粘贴代码的小事情之外，我没有看到任何真正的实现问题:create_shell_process 不使用其 cwd 参数，以及 on_success 和 on_failure 参数看起来像是可选的，但默认值会破坏一些东西(你不能调用 None )。