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

Question

问题可以与使用python子进程模块作为命令行模拟器有关

我写了一些名为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期望所有环境变量都有一个类似map的数据结构。我的应用程序需要导出和设置几个变量。用于设置和导出的脚本是通过运行say '/bin/appload myapp'生成的(假设appload在路径上始终可用)。我目前的工作当我调用p. communication时，我执行以下操作

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、进程替换等等)。另一方面，如果您确定/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，该shell对appload myapp环境进行eval，然后在它下面执行另一个shell，这样第一个shell就不会出现在ps清单中，并且给create_shell_process的命令拥有整个shell(尽管这并不重要)。您可以做很多事情，这取决于您对如何调用shell的关注，它在ps清单中的外观，如果appload myapp输出在eval时产生错误，您是否希望命令仍然运行，等等。但对于一般的解决方案，我认为你的方案是非常好的。

2. 我没有看到任何真正的问题与实现，除了化妆的东西或次要的东西，可能只来自复制和粘贴代码:create_shell_process不使用它的cwd参数，和on_success和on_failure参数看起来像他们是可选的，但默认值会破坏的东西(你不能调用None)。