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 相关文章推荐
python 解析html之BeautifulSoup
Jul 07 Python
零基础写python爬虫之爬虫的定义及URL构成
Nov 04 Python
分享Python开发中要注意的十个小贴士
Aug 30 Python
如何在python中使用selenium的示例
Dec 26 Python
Python实现网站表单提交和模板
Jan 15 Python
python3.4爬虫demo
Jan 22 Python
Python、 Pycharm、Django安装详细教程(图文)
Apr 12 Python
Python opencv实现人眼/人脸识别以及实时打码处理
Apr 29 Python
numpy.transpose()实现数组的转置例子
Dec 02 Python
Python 内存管理机制全面分析
Jan 16 Python
python Protobuf定义消息类型知识点讲解
Mar 02 Python
Python 的演示平台支持 WSGI 接口的应用
Apr 20 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
收藏的一个php小偷的核心程序
2007/04/09 PHP
解析PHP中常见的mongodb查询操作
2013/06/20 PHP
PHP按行读取、处理较大CSV文件的代码实例
2014/04/09 PHP
PHP小教程之实现双向链表
2014/06/12 PHP
PHP+FFMPEG实现将视频自动转码成H264标准Mp4文件
2014/09/24 PHP
php将文本文件转换csv输出的方法
2014/12/31 PHP
万能的php分页类
2017/07/06 PHP
浅析PHP类的反射来实现依赖注入过程
2018/02/06 PHP
jQuery 判断页面元素是否存在的代码
2009/08/14 Javascript
Javascript load Page,load css,load js实现代码
2010/03/31 Javascript
读jQuery之十二 删除事件核心方法
2011/07/31 Javascript
Wireshark基本介绍和学习TCP三次握手
2016/08/15 Javascript
详解Nodejs的timers模块
2016/12/22 NodeJs
Vue结合Video.js播放m3u8视频流的方法示例
2018/05/04 Javascript
JS实现图片拖拽交换效果
2018/11/30 Javascript
javascript开发实现贪吃蛇游戏
2020/07/31 Javascript
vue 解决data中定义图片相对路径页面不显示的问题
2020/08/13 Javascript
[02:43]DOTA2英雄基础教程 圣堂刺客
2013/12/09 DOTA
python使用socket进行简单网络连接的方法
2015/04/29 Python
零基础小白多久能学会python
2020/06/22 Python
Python操作Word批量生成合同的实现示例
2020/08/28 Python
如何用tempfile库创建python进程中的临时文件
2021/01/28 Python
HTML5 Canvas实现玫瑰曲线和心形图案的代码实例
2014/04/10 HTML / CSS
html5实现的便签特效(实战分享)
2013/11/29 HTML / CSS
汽车运用工程专业毕业生推荐信
2013/12/25 职场文书
我的动漫时代的创业计划书范文
2014/01/27 职场文书
大学生水果店创业计划书
2014/01/28 职场文书
服务员岗位责任制
2014/02/11 职场文书
赔偿协议书范本
2014/04/15 职场文书
社区党的群众路线教育实践活动领导班子对照检查材料
2014/09/25 职场文书
2014年最新个人对照检查材料范文
2014/09/25 职场文书
网络营销计划书
2015/01/17 职场文书
出国留学单位推荐信
2015/03/26 职场文书
2015年园林绿化工作总结
2015/05/23 职场文书
Python批量解压&压缩文件夹的示例代码
2022/04/04 Python
Web应用开发TypeScript使用详解
2022/05/25 Javascript