之前写过一个python远程执行命令的脚本，但在一个性能测试中，要将程序批量分发到不同服务器，程序无法使用，再将之前的脚本更新，加入批量上传的功能。之前脚本地址：http://www.cnblogs.com/landhu/p/4961##coding:utf-8
#------------------ #Author:Hu #Created:2016-02-29 #------------------ import paramiko,os,time,sys from optparse import OptionParser #使用optparse做命令行解析 parser = OptionParser(usage="usage:%prog [optinos] dest",version='Hu 3.0_20160229') parser.add_option('-f','--filename',dest='filename',default='ip.txt',help='The Servers info TXT') parser.add_option('-c','--command',dest='cmd',help='You want execute command') parser.add_option('-u','--upload',dest='upload',help='You want to upload file') #parser.add_option('-d','--download',dest='download',help='You want to download file') #parser.add_option('') (options,args)=parser.parse_args()
def callback(a=10,b=10):
sys.stdout.write('Data Transmission %10f M [%3.2f%%]\r') % (a/1024./1024,a*100./int(b)))
sys.stdout.flush()
'''
将功能封装
'''
class bat(object):
def __init__(self,ip,pt,pw,us):
self.ip=ip
self.pt=pt
self.pw=pw
self.us=us
print self.ip
def command(self,comm):
self.comm=comm
try:
ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(self.ip,port=self.pt,username=self.us,password=self.pw)
for COMM in self.comm:
stdin,stdout,stderr = ssh.exec_command(COMM)
print "%s The command %s result is:\n" % (time.strftime("%Y%m%d %H:%M:%S"),COMM)
read = stdout.read()
print read
print '-'*70
ssh.close()
except Exception,e:
print e
print '-'*70
def upload(self,filename):
self.ufilename=filename
try:
#print self.ip,self.pt
client = paramiko.Transport((self.ip, self.pt))
client.connect(username = self.us, password = self.pw)
sftp = paramiko.SFTPClient.from_transport(client)
sftp.put(localpath=self.ufilename,remotepath=self.ufilename,callback=callback)
client.close()
except Exception,e:
print e
#本想写个从服务器批量下载的功能能，但实际上用不上，先不添加，后期再丰富
def download(self,filename):
self.dfilename=filename
try:
client = paramiko.Transport((self.ip, self.pt))
client.connect(username = self.us, password = self.pw)
sftp = paramiko.SFTPClient.from_transport(client)
sftp.get(self.dfilename, self.dfilename)
client.close()
except:
print 'Error'
if __name__ == '__main__':
if os.path.exists(options.filename):
filename=options.filename
else:
print 'Please check %s and ip.txt is exits' % options.server
exit(-1)
print "-"*70
#批量执行命令与上传不能同时存在
if options.cmd and options.upload:
parser.error("options -c and -u are mutually exclusive")
elif options.cmd:
open_ip = open(filename)
command=[]
command=options.cmd.split(',')
for line in open_ip.readlines():
ip,user,passwd,port=line.strip().split()
port=int(port)
dd=bat(ip=ip,pt=port,pw=passwd,us=user)
dd.command(command)
elif options.upload:
fullpwd=os.path.abspath(options.upload)
open_ip = open(filename)
for line in open_ip.readlines():
ip,user,passwd,port=line.strip().split()
port=int(port)
dd=bat(ip=ip,pt=port,pw=passwd,us=user)
print "You want upload file is : %s " % fullpwd
dd.upload(options.upload)
print 'Data Transmission :'
print '-'*70
else:
print 'Arg is error,Use "-h" to help!'
'''
elif options.download:
fullpwd=os.path.abspath(options.download)
open_ip = open(filename)
for line in open_ip.readlines():
ip,user,passwd,port=line.strip().split()
port=int(port)
dd=bat(ip=ip,pt=port,pw=passwd,us=user)
print "You want download file is : %s " % fullpwd
dd.download(options.download)
print 'Data Transmission :'
print '-'*60
'''
默认读取ip.txt中的服务器信息，信息的格式：
IP 用户 密码 端口
192.168.3.200 root rr@#$zlb 36003 192.168.3.250 root rr@#250 22 192.168.14.31 root rr@#31 22 192.168.10.3 root 111 22 192.168.10.3 root 111 22
运行如下：
[root@localhost tools]# python remote.py -h Usage: usage:remote.py [optinos] dest Options: --version show program's version number and exit -h, --help show this help message and exit -f FILENAME, --filename=FILENAME The Servers info TXT -c CMD, --command=CMD You want execute command -u UPLOAD, --upload=UPLOAD You want to upload file
[root@localhost tools]# python remote.py -c pwd ------------------------------------------------------------ 192.168.3.200 20160229 15:07:39 The command pwd result is: /root ---------------------------------------------------------------------- 192.168.3.3 [Errno 111] Connection refused ---------------------------------------------------------------------- 192.168.3.250 20160229 15:07:50 The command pwd result is: /root ---------------------------------------------------------------------- 192.168.14.31 [Errno 111] Connection refused ---------------------------------------------------------------------- 192.168.3.3 [Errno 111] Connection refused ---------------------------------------------------------------------- 192.168.10.3 [Errno 110] Connection timed out ---------------------------------------------------------------------- 192.168.10.3 [Errno 110] Connection timed out ----------------------------------------------------------------------
两个命令不能冲突：
[root@localhost tools]# python remote.py -c pwd -u ip.txt ------------------------------------------------------------ Usage: usage:remote.py [optinos] dest remote.py: error: options -c and -u are mutually exclusive
上传时有一个特别的功能，上传时的进度显示，这里使用的是paramiko put中的callback参数
效果如下：
结果如下：
[root@localhost tools]# python remote.py -u bms.war ------------------------------------------------------------ 192.168.3.200 You want upload file is : /tools/bms.war Data Transmission :77.198704 M [100.00%] ---------------------------------------------------------------------- 192.168.3.3 You want upload file is : /tools/bms.war Unable to connect to 192.168.3.3: [Errno 111] Connection refused Data Transmission : ---------------------------------------------------------------------- 192.168.3.250 You want upload file is : /tools/bms.war Data Transmission :77.198704 M [100.00%] ---------------------------------------------------------------------- 192.168.14.31 You want upload file is : /tools/bms.war Unable to connect to 192.168.14.31: [Errno 111] Connection refused Data Transmission : ---------------------------------------------------------------------- 192.168.3.3 You want upload file is : /tools/bms.war Unable to connect to 192.168.3.3: [Errno 111] Connection refused Data Transmission : ---------------------------------------------------------------------- 192.168.10.3 You want upload file is : /tools/bms.war Unable to connect to 192.168.10.3: [Errno 110] Connection timed out Data Transmission : ---------------------------------------------------------------------- 192.168.10.3 You want upload file is : /tools/bms.war Unable to connect to 192.168.10.3: [Errno 110] Connection timed out Data Transmission : ----------------------------------------------------------------------
参考文档：
paramiko SFTP:http://paramiko-docs.readthedocs.org/en/latest/api/sftp.html