linux下远程服务器批量执行命令及SFTP上传文件 -- python实现

之前写过一个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%%]\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
 :: The command pwd result is:

/root

----------------------------------------------------------------------
192.168.3.3
[Errno ] Connection refused
----------------------------------------------------------------------
192.168.3.250
 :: The command pwd result is:

/root

----------------------------------------------------------------------
192.168.14.31
[Errno ] Connection refused
----------------------------------------------------------------------
192.168.3.3
[Errno ] Connection refused
----------------------------------------------------------------------
192.168.10.3
[Errno ] Connection timed out
----------------------------------------------------------------------
192.168.10.3
[Errno ] 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 ] 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 ] 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 ] 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 ] 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 ] Connection timed out
Data Transmission :
----------------------------------------------------------------------

参考文档：

paramiko SFTP:http://paramiko-docs.readthedocs.org/en/latest/api/sftp.html