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


Posted in Python onOctober 25, 2016

局域网内有一百多台电脑,全部都是linux操作系统,所有电脑配置相同,系统完全相同(包括用户名和密码),ip地址是自动分配的。现在有个任务是在这些电脑上执行某些命令,者说进行某些操作,比如安装某些软件,拷贝某些文件,批量关机等。如果一台一台得手工去操作,费时又费力,如果要进行多个操作就更麻烦啦。

或许你会想到网络同传,网络同传是什么?就是在一台电脑上把电脑装好,配置好,然后利用某些软件,如“联想网络同传”把系统原样拷贝过去,在装系统时很有用,只要在一台电脑上装好,同传以后所有的电脑都装好操作系统了,很方便。同传要求所有电脑硬件完全相同,在联想的电脑上装的系统传到方正电脑上肯定会出问题的。传系统也是很费时间的,根据硬盘大小,如果30G硬盘,100多台电脑大约要传2个多小时,反正比一台一台地安装快!但是如果系统都传完了,发现忘了装一个软件,或者还需要做些小修改,再同传一次可以,但是太慢,传两次半天时间就没了。这时候我们可以利用ssh去控制每台电脑去执行某些命令。

先让我们回忆一下ssh远程登录的过程:首先执行命令 ssh username@192.168.1.x ,第一次登录的时候系统会提示我们是否要继续连接,我们要输入“yes”,然后等一段时间后系统提示我们输入密码,正确地输入密码之后我们就能登录到远程计算机,然后我们就能执行命令了。我们注意到这里面有两次人机交互,一次是输入‘yes',另一次是输入密码。就是因为有两次交互我们不能简单的用某些命令去完成我们的任务。我们可以考虑把人机交互变成自动交互,python的pexpect模块可以帮我们实现自动交互。下面这段代码是用pexpect实现自动交互登录并执行命令的函数:

#!/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()

上面的程序还是有些技巧的:

  • 利用多线程,同时发出登录请求,同时去连接电脑,这样速度快很多,我试了一下,如果不用多线程,直接一个一个挨着执行的话,大约5~10秒钟才能对一台电脑操作完,具体时间要根据命令的来决定,如果是软件安装或者卸载时间要更长一些。这样下来怎么也要一二十分钟,用多线程后就快多了,所有的命令执行完用了不到2分钟!
  • 最好用root用户登录,因为安装或者卸载软件的时候如果用普通用户又会提示输入密码,这样又多了一次交互,处理起来就比较麻烦!安装软件时apt-get install xxx 最好加上“-y”参数,因为有时安装或删除软件时提示是否继续安装或卸载,这又是一次自动交互!加上那个参数后就没有人机交互了。
  • 循环时循环所有ip,因为计算机的ip是路由器自动分配的,保险起见,最好全部都执行,保证没有遗漏的主机
  • 远端执行命令时如果有交互,可以这样用 stdin.write("Y")来完成交互,“Y”就是输入“Y”。
  • 把所有的命令放到一个列表里面,遍历列表可以依次执行列表里面的命令
  • 为了更好的进行控制,最好在电脑上提前把root用户打开,装好ssh服务器并让其开机自动执行。

希望本文所述对你有所帮助,Python实现ssh批量登录并执行命令内容就给大家介绍到这里了。希望大家继续关注我们的网站!想要学习Python可以继续关注本站。

Python 相关文章推荐
python使用正则表达式提取网页URL的方法
May 26 Python
python MySQLdb使用教程详解
Mar 20 Python
PyQt5每天必学之像素图控件QPixmap
Apr 19 Python
Python查找第n个子串的技巧分享
Jun 27 Python
对Python Class之间函数的调用关系详解
Jan 23 Python
python,Django实现的淘宝客登录功能示例
Jun 12 Python
docker-py 用Python调用Docker接口的方法
Aug 30 Python
python自动化测试无法启动谷歌浏览器问题
Oct 10 Python
浅谈Python3实现两个矩形的交并比(IoU)
Jan 18 Python
Python+PyQt5实现灭霸响指功能
May 25 Python
Python3.9.1中使用match方法详解
Feb 08 Python
python使用pycharm安装pyqt5以及相关配置
Apr 22 Python
详解Python的Lambda函数与排序
Oct 25 #Python
Python脚本实现Web漏洞扫描工具
Oct 25 #Python
python+django快速实现文件上传
Oct 24 #Python
python实现简单爬虫功能的示例
Oct 24 #Python
简单谈谈Python中的反转字符串问题
Oct 24 #Python
Python 内置函数complex详解
Oct 23 #Python
Python检测生僻字的实现方法
Oct 23 #Python
You might like
PHP安装攻略:常见问题解答(三)
2006/10/09 PHP
PHP实现网上点歌(二)
2006/10/09 PHP
PHP接口并发测试的方法(推荐)
2016/12/15 PHP
Yii框架实现记录日志到自定义文件的方法
2017/05/23 PHP
PHP实现限制域名访问的实现代码(本地验证)
2020/09/13 PHP
onkeydown事件解决按回车键直接提交数据的需求
2013/04/11 Javascript
利用JS进行图片的切换即特效展示图片
2013/12/03 Javascript
跟我学习javascript的定时器
2015/11/19 Javascript
javascript特效实现——当前时间和倒计时效果的简单实例
2016/07/20 Javascript
JS中的hasOwnProperty()和isPrototypeOf()属性实例详解
2016/08/11 Javascript
js 数据存储和DOM编程
2017/02/09 Javascript
基于DOM节点删除之empty和remove的区别(详解)
2017/09/11 Javascript
基于对象合并功能的实现示例
2017/10/10 Javascript
js实现鼠标单击Tab表单切换效果
2018/05/16 Javascript
利用Webpack实现小程序多项目管理的方法
2019/02/25 Javascript
vue实现树形结构样式和功能的实例代码
2019/10/15 Javascript
Vue实现手机扫描二维码预览页面效果
2020/05/28 Javascript
JS数组降维的实现Array.prototype.concat.apply([], arr)
2020/04/28 Javascript
JavaScript 如何在浏览器中使用摄像头
2020/12/02 Javascript
python简单读取大文件的方法
2016/07/01 Python
Python tornado队列示例-一个并发web爬虫代码分享
2018/01/09 Python
Python中常见的异常总结
2018/02/20 Python
python实现简易数码时钟
2021/02/19 Python
处理Selenium3+python3定位鼠标悬停才显示的元素
2019/07/31 Python
python读取指定字节长度的文本方法
2019/08/27 Python
使用python将最新的测试报告以附件的形式发到指定邮箱
2019/09/20 Python
使用Bazel编译TensorBoard教程
2020/02/15 Python
使用 prometheus python 库编写自定义指标的方法(完整代码)
2020/06/29 Python
ET Mall东森购物网:东森严选
2017/03/06 全球购物
到底Java是如何传递参数的?是by value或by reference?
2012/07/13 面试题
法律专业个人实习自我鉴定
2013/09/23 职场文书
营销人才自我鉴定范文
2013/12/25 职场文书
担保书格式及范文
2014/04/01 职场文书
重阳节标语大全
2014/10/07 职场文书
JavaWeb Servlet开发注册页面实例
2022/04/11 Java/Android
LeetCode189轮转数组python示例
2022/08/05 Python