Pexpect 是一个 Python 库,用于自动化与终端交互的任务。它提供了一种简单的方式来编写脚本,以便与终端程序进行交互。以下是 Pexpect 的一些常用参数和方法的更详细说明:
常用参数:
spawn():
这是启动一个新的命令行程序的方法。你可以传递一个命令行程序的路径作为参数。例如,ftp = pexpect.spawn(‘/usr/bin/ftp’) 将启动 FTP 客户端。
expect():
这是等待命令行程序输出特定字符串或正则表达式的方法。你可以传递一个字符串或正则表达式作为参数,然后 Pexpect 将等待直到程序输出这个字符串或正则表达式。例如,ftp.expect('ftp> ') 将等待直到 FTP 客户端提示你输入命令。
sendline():
这是向命令行程序发送一行命令的方法。你可以传递一个字符串作为参数,然后 Pexpect 将把这个字符串发送给程序。例如,ftp.sendline(‘ls’) 将发送 ‘ls’ 命令到 FTP 客户端,然后等待程序输出结果。
close():
这是关闭与命令行程序的连接的方法。当你完成与程序的交互后,你应该调用这个方法来释放资源。例如,ftp.close() 将关闭与 FTP 客户端的连接。
interact():
这是启动交互模式的方法。在交互模式下,你可以手动控制与命令行程序的交互。例如,你可以使用键盘输入命令,然后使用 sendline() 方法将它们发送到程序。
Pexpect可以通过以下方式进行安装:
#下载tcl源码包:
wget http://prdownloads.sourceforge.net/tcl/tcl8.4.20-src.tar.gz。
#解压并编译安装tcl:
tar zxvf tcl8.4.20-src.tar.gz,cd tcl8.4.20/unix/,./configure --prefix=/usr/tcl --enable-shared,make,make install。
#下载expect源码包:wget http://sourceforge.net/projects/expect/files/Expect/5.45/expect5.45.tar.gz。
#解压并编译安装expect:
tar zxvf expect5.45.tar.gz,
cd expect5.45,
./configure --prefix=/usr/expect --with-tcl=/usr/tcl/lib --with-tclinclude=../tcl8.4.20/generic/,
make,
make install
#安装完成之后做一个软连接:
ln -s /usr/tcl/bin/expect /usr/expect/bin/expect
#加入环境变量并测试:
echo 'export PATH=$PATH:/usr/expect/bin' >> /etc/profile
示例:
#!/usr/bin/env python
import pexpect
devices = {'iosv-1': {'prompt': 'lax-edg-r1#', 'ip': '192.168.2.51'},
'iosv-2': {'prompt': 'lax-edg-r2#', 'ip': '192.168.2.52'}}
username = 'cisco'
password = 'cisco'
for device in devices.keys():
device_prompt = devices[device]['prompt']
child = pexpect.spawn('telnet ' + devices[device]['ip'])
child.expect('Username:')
child.sendline(username)
child.expect('Password:')
child.sendline(password)
child.expect(device_prompt)
child.sendline('show version | i V')
child.expect(device_prompt)
print(child.before)
child.sendline('exit')
ssh:
from pexpect import pxssh
>>> child = pxssh.pxssh()
>>> child.login('192.168.2.51', 'cisco', 'cisco', auto_prompt_reset=False)
True
>>> child.sendline('show version | i V')
19
>>> child.expect('lax-edg-r1#')
0
>>> child.before
b'show version | i V\r\nCisco IOS Software, IOSv Software (VIOSADVENTERPRISEK9-M), Version 15.8(3)M2, RELEASE SOFTWARE (fc2)\r\nProcessor
board ID 98U40DKV403INHIULHYHB\r\n'
>>> child.logout()
ftp:
import pexpect
# 启动 FTP 子程序
ftp = pexpect.spawn('ftp')
# 期望看到 'ftp>' 提示符
ftp.expect('ftp> ')
# 发送用户名和密码(假设为 'user' 和 'pass')
ftp.sendline('user')
ftp.sendline('pass')
# 期望登录成功
ftp.expect('230 Login successful')
# 发送命令 'dir' 查看目录列表
ftp.sendline('dir')
ftp.expect('Directory listing')
# 关闭 FTP 子程序
ftp.sendline('bye')
log:
import pexpect
import time
# 创建日志文件并设置 Pexpect 记录日志
log_file = open('ftp_log.txt', 'w')
ftp = pexpect.spawn('ftp', logfile=log_file)
time.sleep(2) # 让子程序运行一段时间以便记录日志
log_file.close() # 关闭日志文件
telnet:
import pexpect
# 启动 Telnet 子程序
telnet = pexpect.spawn('telnet', ['hostname', 'port'])
# 期望看到 Telnet 提示符
telnet.expect('login: ')
# 发送用户名和密码(假设为 'user' 和 'pass')
telnet.sendline('user')
telnet.expect('Password:')
telnet.sendline('pass')
# 期望登录成功并看到提示符(例如 '#' 或 '$')
telnet.expect('#')
# 使用 interact 方法进行交互(注意:interact 方法会阻塞程序执行直到子程序关闭)
telnet.interact()