Question

我正在尝试制作一个简单的Python脚本，该脚本在使用'su'命令（或任何其他需要管理员特权或仅需要密码才能执行的命令）之后在命令行中输入给定密码。

我尝试使用Subprocess模块以及Pynput，但是无法弄清楚。

import subprocess
import os

# os.system('su') # Tried using this too

process = subprocess.Popen('su', stdin=subprocess.PIPE, stdout=subprocess.PIPE)
process.stdin.write(b"password_to_enter")
print(process.communicate()[0])
process.stdin.close()

我希望在键入“ su”命令后在给定的密码提示下输入“ password_to_enter”，但是没有。我也尝试提供正确的密码，但是仍然无法正常工作。

我在做什么错了？

PS：我在Mac上

Answer 1

su命令期望从终端读取。在我的Linux机器上运行上面的示例将返回以下错误：

su: must be run from a terminal

这是因为su试图确保它正在从终端运行。您可以通过分配pty并自行管理输入和输出来绕过此操作，但是正确进行此操作可能会非常棘手，因为您必须在之后 su提示输入密码才能输入密码。例如：

import subprocess
import os
import pty
import time

# Allocate the pty to talk to su with.
master, slave = pty.openpty()

# Open the process, pass in the slave pty as stdin.
process = subprocess.Popen('su', stdin=slave, stdout=subprocess.PIPE, shell=True)

# Make sure we wait for the "Password:" prompt.
# The correct way to do this is to read from stdout and wait until the message is printed.
time.sleep(2)

# Open a write handle to the master end of the pty to write to.
pin = os.fdopen(master, "w")
pin.write("password_to_enter\n")
pin.flush()

# Clean up
print(process.communicate()[0])
pin.close()
os.close(slave)

有一个名为pexpect的库，它使与交互式应用程序的交互变得非常简单：

import pexpect
import sys

child = pexpect.spawn("su")
child.logfile_read = sys.stdout
child.expect("Password:")
child.sendline("your-password-here")
child.expect("#")
child.sendline("whoami")
child.expect("#")

如何在Python的终端密码提示中输入密码

1 个答案: