python实现ssh批量登录并执行命令,pythonssh执行命令,#!/usr/bin/e


#!/usr/bin/env python# -*- coding: utf-8 -*-import pexpectdef ssh_cmd(ip, passwd, cmd):    ret = -1    ssh = pexpect.spawn('ssh root@%s "%s"' % (ip, cmd))    try:        i = ssh.expect(['password:', 'continue connecting (yes/no)?'], timeout=5)        if i == 0 :            ssh.sendline(passwd)        elif i == 1:            ssh.sendline('yes\n')            ssh.expect('password: ')            ssh.sendline(passwd)        ssh.sendline(cmd)        r = ssh.read()        print r        ret = 0    except pexpect.EOF:        print "EOF"        ssh.close()        ret = -1    except pexpect.TIMEOUT:        print "TIMEOUT"        ssh.close()        ret = -2    return ret
                                      利用pexpect模块我们可以做很多事情,由于他提供了自动交互功能,因此我们可以实现ftp,telnet,ssh,scp等的自动登录,还是比较实用的。根据上面的代码相信读者已经知道怎么实现了(python就是那么简单!)。  用上面的代码去完成任务还是比较费时间的,因为程序要等待自动交互出现,另外ubuntu用ssh连接就是比较慢,要进行一系列的验证,这样才体现出ssh的安全。我们要提高效率,在最短的时间内完成。后来我发现了python里面的paramiko模块,用这个实现ssh登录更加简单。看下面的代码:
#-*- coding: utf-8 -*-#!/usr/bin/python import paramikoimport threadingdef ssh2(ip,username,passwd,cmd):    try:        ssh = paramiko.SSHClient()        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())        ssh.connect(ip,22,username,passwd,timeout=5)        for m in cmd:            stdin, stdout, stderr = ssh.exec_command(m)#           stdin.write("Y")   #简单交互,输入 ‘Y’             out = stdout.readlines()            #屏幕输出            for o in out:                print o,        print '%s\tOK\n'%(ip)        ssh.close()    except :        print '%s\tError\n'%(ip)if __name__=='__main__':    cmd = ['cal','echo hello!']#你要执行的命令列表    username = ""  #用户名    passwd = ""    #密码    threads = []   #多线程    print "Begin......"    for i in range(1,254):        ip = '192.168.1.'+str(i)        a=threading.Thread(target=ssh2,args=(ip,username,passwd,cmd))        a.start()
                                上面的程序还是有些技巧的:

1.利用多线程,同时发出登录请求,同时去连接电脑,这样速度快很多,我试了一下,如果不用多线程,直接一个一个挨着执行的话,大约5~10秒钟才能对一台电脑操作完,具体时间要根据命令的来决定,如果是软件安装或者卸载时间要更长一些。这样下来怎么也要一二十分钟,用多线程后就快多了,所有的命令执行完用了不到2分钟!

2.最好用root用户登录,因为安装或者卸载软件的时候如果用普通用户又会提示输入密码,这样又多了一次交互,处理起来就比较麻烦!安装软件时apt-get install; xxx; 最好加上“-y”参数,因为有时安装或删除软件时提示是否继续安装或卸载,这又是一次自动交互!加上那个参数后就没有人机交互了。

循环时循环所有ip,因为计算机的ip是路由器自动分配的,保险起见,最好全部都执行,保证没有遗漏的主机

4.远端执行命令时如果有交互,可以这样用 stdin.write;("Y")来完成交互,“Y”就是输入“Y”。

5.把所有的命令放到一个列表里面,遍历列表可以依次执行列表里面的命令

6.为了更好的进行控制,最好在电脑上提前把root用户打开,装好ssh服务器并让其开机自动执行。

评论关闭