之前写过一个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()
#上传中显示进度，使用的是paramiko put中的callback选项
def callback(a=10,b=10):
sys.stdout.write('Data Transmission %10f M [%3.2f%%] ') % (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: " % (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