返回顶部

收藏

python实现ssh批量登录并执行命令

更多
#!/usr/bin/env python

# -*- coding: utf-8 -*-

import pexpect

def 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 paramiko

import threading

def 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”参数,因为有时安装或删除软件时提示是否继续安装或卸载,这又是一次自动交互!加上那个参数后就没有人机交互了。

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

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

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

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

标签:ssh,登录,python

收藏

0人收藏

支持

0

反对

0

相关聚客文章
  1. 彭剑锋 发表 2015-12-06 19:19:00 ubuntu server用户必须本地登录后才能远程ssh登陆
  2. LI Daobing 发表 2009-05-20 13:15:00 ssh 登录过慢的问题
  3. LI Daobing 发表 2009-05-20 13:15:00 ssh 登录过慢的问题
  4. 博主 发表 2014-03-17 16:00:00 Python模拟新浪微博登录
  5. 沈 冠雄 发表 2013-10-08 02:57:16 Linux配置ssh证书无密码登录
  6. 博主 发表 2013-03-19 04:15:00 Python模拟登录新浪微薄(使用RSA加密方式和Cookies文件)
  7. 博主 发表 2014-09-03 00:00:00 ssh 无密码登录
  8. hackhp 发表 2015-12-12 14:08:55 Linux开启SSH登录email通知
  9. Huqiu Liao / Srain 发表 2014-07-01 16:00:00 ssh + expect 自动登录服务器
  10. admin 发表 2015-11-01 14:49:44 MAC OSX终端登录ssh保持激活状态防止超时
  11. 博主 发表 2017-09-01 13:03:56 记一次诡异的 ssh 互信免密码登录失败
  12. yakiang 发表 2013-09-11 04:00:00 Notes of Summer

发表评论