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()