最近要使用python做一个在web上管理交换机的程序,需要远程登录,就查了点资料,由于还没有搞到交换机,就先用自己的机器测试一下。
首先python的标准库中包含telnet,用起来也很方便,查看一下文档写了个小程序:
- #!/usr/bin/env python
- #coding=utf-8
- import telnetlib
- host = "127.0.0.1"
- userName = 'root'
- password = '123456'
- enter = '\n'
- t = telnetlib.Telnet(host)
- t.read_until("login: ",1)
- t.write(userName + enter)
- t.read_until("Password: ",1)
- t.write(password + enter)
- t.write("ls"+enter)
- t.write("exit"+enter)
- print t.read_all()
输出结果:
- Last login: Wed Nov 2 14:51:36 on console
- shi-kefumatoiMac:~ root# .CFUserTextEncoding .subversion Library
- .forward .viminfo nat.sh
- .sh_history .vimrc noc
- shi-kefumatoiMac:~ root# logout
程序很简单,登录自己的机器,执行ls命令,然后输出结果。
telnet很好用,但是总是有人喜欢更强大更好用的程序,于是就有了pexpect,pexpect 是 Don Libes 的 Expect 语言的一个 Python 实现,是一个用来启动子程序,并使用正则表达式对程序输出做出特定响应,以此实现与其自动交互的 Python 模块。 Pexpect 的使用范围很广,可以用来实现与 ssh、ftp 、telnet 等程序的自动交互;可以用来自动复制软件安装包并在不同机器自动安装;还可以用来实现软件测试中与命令行交互的自动化。
看了写资料,也用pexpect写了一个小程序,实现刚才同样的功能:
- #!/usr/bin/env python
- #coding=utf-8
- import pexpect
- address = '127.0.0.1'
- userName = 'root'
- password = '123456'
- cmd = 'telnet ' + address
- prompt = '[$#>]'
- child = pexpect.spawn(cmd)
- index = child.expect(['login',pexpect.EOF,pexpect.TIMEOUT],timeout=1)
- if index == 0:
- child.sendline(userName)
- index = child.expect('Password',timeout=1)
- child.sendline(password)
- child.expect(prompt,timeout=1)
- child.sendline('ls')
- child.expect('ls',timeout=1)
- child.expect(prompt,timeout=1)
- print child.before
- else:
- print 'expect "login",but get EOF or TIMEOUT'
- child.close()
输出结果:
- .CFUserTextEncoding .subversion Library
- .forward .viminfo nat.sh
- .sh_history .vimrc noc
- shi-kefumatoiMac:~ root
当然,这个程序只是pexpect的一个实例程序,简单介绍的它的简单用法,还不足以说明它t的强大。
参考资料:
python文档:http://docs.python.org/library/telnetlib.html
IBM developerWorks: http://www.ibm.com/developerworks/cn/linux/l-cn-pexpect1/
http://www.ibm.com/developerworks/cn/linux/l-cn-pexpect2/