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


Posted in Python onMay 25, 2012

局域网内有一百多台电脑,全部都是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()

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

作者:cnblogs ma6174

Python 相关文章推荐
利用soaplib搭建webservice详细步骤和实例代码
Nov 20 Python
python网页请求urllib2模块简单封装代码
Feb 07 Python
Python多线程编程(七):使用Condition实现复杂同步
Apr 05 Python
python图像处理之反色实现方法
May 30 Python
详解Python异常处理中的Finally else的功能
Dec 29 Python
详解10个可以快速用Python进行数据分析的小技巧
Jun 24 Python
python实现集中式的病毒扫描功能详解
Jul 09 Python
Django中自定义查询对象的具体使用
Oct 13 Python
使用python绘制二维图形示例
Nov 22 Python
实现Python与STM32通信方式
Dec 18 Python
python 多线程爬取壁纸网站的示例
Feb 20 Python
Django对接elasticsearch实现全文检索的示例代码
Aug 02 Python
巧用Python装饰器 免去调用父类构造函数的麻烦
May 18 #Python
Python使用Socket(Https)Post登录百度的实现代码
May 18 #Python
写了个监控nginx进程的Python脚本
May 10 #Python
400多行Python代码实现了一个FTP服务器
May 10 #Python
使用PYTHON接收多播数据的代码
Mar 01 #Python
使用PYTHON创建XML文档
Mar 01 #Python
基于python的汉字转GBK码实现代码
Feb 19 #Python
You might like
一个改进的UBB类
2006/10/09 PHP
sphinx增量索引的一个问题
2011/06/14 PHP
php中OR与|| AND与&&的区别总结
2013/10/26 PHP
PHP实现仿Google分页效果的分页函数
2015/07/29 PHP
初识PHP中的Swoole
2016/04/05 PHP
PHP MYSQL简易交互式站点开发
2016/12/27 PHP
php 解决扫描二维码下载跳转问题
2017/01/13 PHP
PHP基于SPL实现的迭代器模式示例
2018/04/22 PHP
php策略模式简单示例分析【区别于工厂模式】
2019/09/25 PHP
jQuery formValidator表单验证插件开源了 含API帮助、源码、示例
2008/08/14 Javascript
JavaScript利用append添加元素报错的解决方法
2014/07/01 Javascript
Javascript实现通过选择周数显示开始日和结束日的实现代码
2016/05/30 Javascript
javascript中对象的定义、使用以及对象和原型链操作小结
2016/12/14 Javascript
详解VueJS 数据驱动和依赖追踪分析
2017/07/26 Javascript
JS实现图片拖拽交换效果
2018/11/30 Javascript
js实现下拉框二级联动
2018/12/04 Javascript
基于element-ui组件手动实现单选和上传功能
2018/12/06 Javascript
javascript获取select值的方法完整实例
2019/06/20 Javascript
Vue基于iview实现登录密码的显示与隐藏功能
2020/03/06 Javascript
VSCode搭建React Native环境
2020/05/07 Javascript
vue+springboot+element+vue-resource实现文件上传教程
2020/10/21 Javascript
在Python的Django框架上部署ORM库的教程
2015/04/20 Python
对numpy中二进制格式的数据存储与读取方法详解
2018/11/01 Python
python绘制无向图度分布曲线示例
2019/11/22 Python
pyhton中__pycache__文件夹的产生与作用详解
2019/11/24 Python
tensorflow使用指定gpu的方法
2020/02/04 Python
Django中FilePathField字段的用法
2020/05/21 Python
Python getsizeof()和getsize()区分详解
2020/11/20 Python
python代码实现猜拳小游戏
2020/11/30 Python
DJI全球:DJI Global
2021/03/15 全球购物
初中生学习的自我评价
2013/11/14 职场文书
幼儿园教育教学反思
2014/01/31 职场文书
大专应届毕业生求职信
2014/07/15 职场文书
高中生综合素质评价范文
2015/08/18 职场文书
2016年十一促销广告语
2016/01/28 职场文书
5分钟教你docker安装启动redis全教程(全新方式)
2021/05/29 Redis