如何使用python脚本在sudo之后执行一组命令

Question

我正在尝试使用 python 自动化部署过程。在部署中，我首先执行“dzdo su - sysid”，然后执行部署过程。但我无法在 python 中处理这部分。我在shell中做了类似的事情，我使用了以下代码，

/bin/bash

psh su - sysid  << EOF

. /users/home/sysid/.bashrc

./deployment.sh

EOF

这很好地处理了 deployment.sh 的执行。它执行 sudo，然后使用 sysid id 执行脚本。 我正在尝试使用 python 做类似的事情，但我无法在 python 中找到

我正在使用 subprocess.Popen 执行 dzdo 部分，它执行 dzdo，但是当我尝试执行下一个命令时，例如说“ls -l”，然后这个命令将不会使用 sysid 执行，相反，我必须退出 sysid 会话，一旦我退出，它将在我的主目录中执行“ls -l”，这是没有的利用。有人可以帮我解决这个问题吗？

还有一件事，在这种情况下，我不会调用任何 deployment.sh，而是会调用诸如 cp、rm、mkdir 等命令。

Answer 1

您的 shell 脚本示例中 << EOF 和 EOF 之间的文本将写入 psh 进程的标准输入。因此，您必须重定向 Popen 实例的标准输入并将数据直接写入实例的 stdin 文件或使用 communicate() 方法：

#!/usr/bin/env python
# coding: utf8
from __future__ import absolute_import, division, print_function
from subprocess import Popen, PIPE

SHELL_COMMANDS = r'''\
. /users/home/sysid/.bashrc
./deployment.sh
'''


def main():
    process = Popen(['psh', 'su', '-', 'sysid'], stdin=PIPE)
    process.communicate(SHELL_COMMANDS)


if __name__ == '__main__':
    main()

如果您需要进程的标准输出和/或错误的输出，那么您也需要通过管道传输这些内容并使用 communicate() 调用的返回值。