如何使用 Python pexpect 执行操作系统命令？

Posted 2023-02-23

【中文标题】如何使用 Python pexpect 执行操作系统命令？

【英文标题】：How can I execute OS Commands with Python pexpect?

【发布时间】：2021-05-02 11:51:57

【问题描述】：

我想执行一个python脚本，通过自动写入用户密码切换到另一个用户。两个用户都没有 root 权限。登录后我想执行操作系统命令“whoami”来检查登录是否成功。代码如下：

child = pexpect.spawn('su - otheruser)
child.expect_exact('Password:')
child.sendline('password')
print("logged in...")
child.expect('')
child.sendline('whoami')
print(child.before)

我想将命令的输出打印到控制台（仅用于调试），但输出类似于“b272”（随机字母的组合）而不是实际的 whoami 用户。我该如何解决这个问题？

稍后我想从切换的用户创建一些文件等等。所以基本上，我想在其他用户登录的 python 脚本中执行操作系统命令。

【参考方案1】：

Pexpect 搜索并不贪心，所以它会在第一次匹配时停止。当我用before、match.groups()、after 和buffer 测试您的代码时，我没有得到EOF 或TIMEOUT，所以它必须在读取的开头匹配并且什么也没返回（我很惊讶你得到任何结果）。

我建议始终在 expect 后面跟着 sendline，提示符 (]$) 的结尾是一件好事，而不是空字符串。

这是我对您的代码的看法，包括创建文件：

注意 - 在 Centos 7.9 上测试，使用 Python 2.7。

import pexpect

child = pexpect.spawn("su - orcam")
child.expect_exact("Password:")
child.sendline("**********")
child.expect_exact("]$")
print("Logged in...\n")
child.sendline("whoami")
child.expect_exact("]$")
print(child.before + "\n")
child.sendline("echo -e 'Hello, world.' >> hello.txt")
child.expect_exact("]$")
child.sendline("cat hello.txt")
child.expect_exact("]$")
print(child.before + "\n")
child.sendline("exit")
index = child.expect_exact(["logout", pexpect.EOF, ])
print("Logged out: 0".format(index))

输出：

Logged in...

 whoami
orcam
[orcam@localhost ~

 cat hello.txt
Hello, world.
[orcam@localhost ~

Logged out: 0