分类: Python/Ruby
2009-08-01 16:23:01
今天想做一个自动登录服务器执行一些命令,然后上传一些东西进行部署的东西。看到有一个叫telnetlib的库可以使用,于是做了一个简单的小程序来登录到服务器执行一些命令。代码如下:
import telnetlib
host['ip']='127.0.0.1'
host['user']='test'
host['password']='test'
host['commands']=['cd lib', 'ls']
def do(host):
tn = telnetlib.Telnet(host['ip'])
tn.set_debuglevel(2)tn.read_until("login: ")
tn.write(host['user'] + "\n")
tn.read_until("Password: ")
tn.write(host['password'] + "\n")
for command in host['commands']:
tn.write(command+'\n')
tn.write("exit\n")
print tn.read_all()
print 'Finish!'
上述程序只是一个示例,并不能真正运行。不过感到使用 的模块真是太方便了。
但 也遇到一些问题,主要就是如何判断程序执行的结果。telnetlib提供好多方法可以得到数据,象read_until()是当结果中存在想要的信息时 返回,read_some()是只要有结果就返回,read_very_lazy()是返回缓冲区中的数据。经过看代码,telnetlib采用缓冲的处 理方式,因此数据并不是一下子就返回的,而是先放在了缓冲区中。许多的读取处理都是围绕着这个缓冲区来的。而缓冲区的信息何时到达就不说不清楚了,也许很 快,也许很慢,也许分别到达,也许一下子就收到了。因此,对于数据不一定到齐的这种情况,就采用了象read_until()来判断缓冲区中的数据是否有 想要的内容,如果没有就等待,除非到达了超时时间。再有可以使用expect方法,与read_until差不多,但是它可以支持正则表达式,功能要强大 得多。
在网上还有一个叫 它好象功能强大,它支持多种协议,它的主页就建议使用 Pexpect 来下载它的文档。不过因为网络不太好就没有下来,不知道支不支持 Telnet 协议。
上 面的程序再配以 ftplib 的功能就可以做一个远程更新的维护程序。只不过如何判断后台返回的信息,我没有找到好的方法,是使用read_lazy()还是read_eager() 呢?还是什么别的方法。最主要的问题是,返回的数据不一定完整。好象没有提供一个read_line()的方法。