pexpect,一个python自动化利器!

🚀 开场白：聊聊自动化交互的那点事儿

工作中难免遇到需要和命令行程序打交道的场景，比如自动登录服务器、批量处理设备配置。如果每次都要手动输入账号密码，效率太低且不现实。pexpect 正是解决这类问题的利器，它能在Python中轻松实现与子程序的自动化交互，堪称运维和测试人员的“省心助手”。

📘 案例一：模拟SSH登录的简单示例

我们先看一个最基础的远程登录例子。这段代码会启动一个SSH进程，然后等待终端出现“password”提示，接着自动填充密码。

import pexpect
child = pexpect.spawn('ssh user@192.168.1.1')
child.expect('password:')
child.sendline('mypassword')
child.expect('$')
print("登录成功，当前提示符为:", child.after)

运行之后，程序会自行完成密码输入，屏幕上展示“登录成功，当前提示符为: $”。
这个过程里，expect 像一位哨兵，牢牢盯着屏幕输出，一旦匹配到期待的关键字便触发响应。

🛠️ 案例二：交互过程中的异常捕获

网络环境不稳定，登录时可能遇到“Connection refused”或超时。我们需要把异常处理加进来，让程序更加健壮。

import pexpect, sys
child = pexpect.spawn('ssh user@192.168.1.1')
try:
    i = child.expect(['password:', 'Connection refused', pexpect.TIMEOUT], timeout=5)
if i == 0: child.sendline('mypassword')
elif i == 1: print("连接被拒，请检查IP")
elif i == 2: print("连接超时")
except Exception as e:
print("发生未知错误:", str(e))

若连接被拒，屏幕上会直接打印出错误描述，而非让代码崩溃。
指定 timeout=5 避免了永久卡死，这在批量操作上百台设备时至关重要。

📁 案例三：记录交互日志方便排查

很多时候我们需要留存操作记录，pexpect 支持将屏幕输入输出实时写入日志文件，便于事后复盘。

import pexpect
child = pexpect.spawn('telnet 192.168.1.2')
child.logfile = open('telnet.log', 'wb')
child.expect('login:')
child.sendline('admin')
child.expect('Password:')
child.sendline('admin123')
child.expect('#')
child.sendline('show version')

上述代码会把整个Telnet会话原样记录到 telnet.log 中。
打开该日志，会发现 “login:”、“admin” 以及后续的命令和回显全部清晰可见，如同观看一场回放电影。

⚡ 案例四：多种匹配模式的灵活切换

某些场景下，程序返回的提示符可能因环境不同而变化，我们可以使用正则或列表组合来适配多种可能性。

import pexpect
child = pexpect.spawn('ftp 192.168.1.3')
child.expect('Name .*:')
child.sendline('ftpuser')
child.expect('Password:')
child.sendline('ftppass')
child.expect(['ftp>', 'Permission denied'])
child.sendline('dir')
print("当前目录列表已发送")

如果密码错误，expect 会捕捉到 “Permission denied” 并停止后续操作。
这种列表匹配方式让代码能够同时处理“成功”与“失败”两条分支，逻辑非常直观。

🆚 优势对比分析

相较于 subprocess 模块，pexpect 专为交互式程序而生，能优雅处理 ssh、telnet 等需要等待特定输出的场景。subprocess 更适合一次性执行并获取结果，面对动态密码提示则显得力不从心。不过 pexpect 依赖 pty（伪终端），在Windows上的支持不如Linux完美。建议开发调试优先选Linux环境，若需跨平台可考虑 pexpect 的Windows变种 winpexpect。

✍️ 总结与互动

今天我们从登录、异常处理、日志记录到多模式匹配，逐步拆解了 pexpect 的核心用法。它虽非万能，但在自动化运维和测试领域绝对是性价比极高的工具。你在项目里是否遇到过棘手的交互难题？欢迎在评论区分享你的故事，顺手点个 在看 支持一下，让更多朋友告别手工输入的烦恼！😊