python终端模拟工具:pexpect
jopen
10年前
Pexpect是一个纯Python模块。 可以用来和ssh、ftp、passwd、telnet 等命令行程序进行自动交互。实现与命令行交互的自动化.
它提供了两个主要接口 run()函数和spawn类.
run()函数
run功能相对简单,只能实现简单交互
run(command,timeout=-1,withexitstatus=False,events=None,extra_args=None, logfile=None, cwd=None, env=None)
run运行命令,然后返回结果,与os.system类似.
示例
pexpect.run('ls -la') # 返回值(输出,退出状态) (command_output, exitstatus) = pexpect.run('ls -l /bin', withexitstatus=1)
spawn类
spawn功能比run强大,可以实现更复杂交互
class spawn __init__(self, command, args=[], timeout=30, maxread=2000, searchwindowsize=None, logfile=None, cwd=None, env=None)
timeout指定交互是等待的超时值
maxread设置read buffer大小. 每次pexpect尝试从TTY(Teletype终端)从读取的最大字节数.
searchwindowsize 指定了从输入缓冲区中进行模式匹配的位置,默认从开始匹配.
logfile 参数指定日志的记录位置
工作过程
# 第一步与终端建立连接 child = pexpect.spawn('scp foo user@example.com:.') # 第二步等待终端返回特定内容 child.expect('Password:') # 第三步根据返回内容发送命令进行交互 child.sendline(mypassword)
pxssh类
pxssh是pexpect的派生类,用于建立ssh连接,比pexpect好用.
login() 建立到目标机器的ssh连接
logout() 释放该连接
prompt() 等待提示符,通常用于等待命令执行结束
工作过程与pexpect相同
# 初始化pxssh示例 s = pxssh.pxssh() hostname = raw_input('hostname: ') username = raw_input('username: ') password = getpass.getpass('password: ') # 建立ssh连接 s.login (hostname, username, password) # 发送命令到远程终端 s.sendline ('uptime') # run a command # 等待远程终端返回 s.prompt() # match the prompt # 打印匹配到的内容 print s.before # print everything before the propt. # 发送命令到远程终端 s.sendline ('ls -l') # 等待远程终端返回 s.prompt() # 打印匹配到的内容 print s.before s.sendline ('df') s.prompt() print s.before # 释放ssh连接 s.logout()