【python】用python脚本Paramiko实现远程执行命令、下载、推送/上传文件功能

Paramiko：

　　paramiko模块，基于SSH用于连接远程服务器并执行相关操作。

　　SSHClient：

　　　　用于连接远程服务器并执行基本命令

　　SFTPClient：

　　　　用于连接远程服务器并执行上传下载

以下该脚本可以实现：上传文件、下载文件、执行命令

 #!/usr/bin/env python
 #定义函数
 import paramiko
 import os
 import datetime
 import sys
 import re
 #定义基础命令格式
 def perform_command(hostname, port, username, password, comm):
     try:
         t=paramiko.Transport((hostname,port))
         t.connect(username=username,password=password)
         try:
             print "\033[33mIP:\033[0m", hostname
             print "\033[33mPerform command:\033[0m", comm
             print '\033[32m###########################################################\033[0m'
             s=paramiko.SSHClient()
             s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
             s.connect(hostname = hostname, port = port, username = username, password = password)
             stdin,stdout,stderr=s.exec_command(comm)
             print stdout.read()
             print '\033[32m###########################################################\033[0m'
             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
             s.close()
         except Exception:
             print "\033[31mERROR - Perform error\033[0m"
     except Exception:
         print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname

 #定义发送文件格式
 def send_file(hostname,port,username,password,local_dir,client_dir,file):
     try:
         t=paramiko.Transport((hostname,port))
         t.connect(username=username,password=password)
         try:
             print "\033[33mIP:\033[0m", hostname
             print "\033[33mSend file:\033[0m", file
             print "\033[33mClient_dir:\033[0m", client_dir
             sftp=paramiko.SFTPClient.from_transport(t)
             sftp.put(os.path.join(local_dir,file),os.path.join(client_dir,file))
             t.close()
             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
         except Exception:
             print "\033[31mERROR - Perform error\033[0m"
     except Exception:
             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
 #定义下载文件格式
 def get_file(hostname,port,username,password,local_dir,client_dir,file):
     try:
         t=paramiko.Transport((hostname,port))
         t.connect(username=username,password=password)
         try:
             print "\033[33mIP:\033[0m", hostname
             print "\033[33mGet file:\033[0m", file
             print "\033[33mClient_dir:\033[0m", client_dir
             sftp=paramiko.SFTPClient.from_transport(t)
             sftp.get(os.path.join(client_dir,file),os.path.join(local_dir,file))
             t.close()
             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
         except Exception:
             print "\033[31mERROR - Perform error\033[0m"
     except Exception:
             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
 #定义帮助文件
 def help():
     print """
 \033[33mHelp:

   Welcome to use  python script, this script can help you run commands, send files and get files.

 Use this script, you can running for one host or a host group, and you know get file can't for a host group.

 Give you examples:

 If you want perform command:
     Usage: (for one host) ./paramiko-upload.py -H [hostip] -c [command]
     Example: ./paramiko-upload.py -H 10.100.139.245 -c 'ls -al'

     Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -c [command]
     Example: ./paramiko-upload.py -F 'ip.txt' -c 'ls -al'

 If you want send file:

     Usage: (for one host) ./paramiko-upload.py -H [hostip] -s [local_dir] [client_dir] [file]
     Example: ./paramiko-upload.py -H 10.100.139.245 -s '/opt/Felix' '/tmp' ip.txt

     Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -s [local_dir] [client_dir] [file]
     Example: ./paramiko-upload.py -F ip.txt -s '/opt/Felix' '/tmp' ip.txt

 If you want get file:
     Usage: (for one host) ./paramiko-upload.py -H [hostip] -g [local_dir] [client_dir] [file]
     Example: ./paramiko-upload.py -H 10.100.139.245 -g '/opt/Felix' '/tmp' ip.txt

 ip.txt file content like this:
     1.1.1.1
     2.2.2.2\033[0m
     """
     sys.exit()

 #定义命令详情

 if __name__ == '__main__':
 #定义账号、端口、密码等信息
     username='root'
     port=2222
     password='abcdefg'

     if len(sys.argv) > 2:
         if sys.argv[1] == '-H':
             if len(sys.argv) == 5 and sys.argv[3] == '-c':
                 hostname=sys.argv[2]
                 comm = sys.argv[4]
                 perform_command(hostname, port, username, password, comm)
             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
                 hostname=sys.argv[2]
                 local_dir=sys.argv[4]
                 client_dir=sys.argv[5]
                 file=sys.argv[6]
                 send_file(hostname,port,username,password,local_dir,client_dir,file)
             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
                 hostname=sys.argv[2]
     username='root'
     port=2222
     password='abcdefg'

     if len(sys.argv) > 2:
         if sys.argv[1] == '-H':
             if len(sys.argv) == 5 and sys.argv[3] == '-c':
                 hostname=sys.argv[2]
                 comm = sys.argv[4]
                 perform_command(hostname, port, username, password, comm)
             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
                 hostname=sys.argv[2]
                 local_dir=sys.argv[4]
                 client_dir=sys.argv[5]
                 file=sys.argv[6]
                 send_file(hostname,port,username,password,local_dir,client_dir,file)
             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
                 hostname=sys.argv[2]
                 local_dir=sys.argv[4]
                 client_dir=sys.argv[5]
                 file=sys.argv[6]
                 get_file(hostname,port,username,password,local_dir,client_dir,file)
             else:
                 help()
         elif sys.argv[1] == '-F':
             ip_file=sys.argv[2]
             file_stat=os.path.exists(ip_file)
             if file_stat:
                 f=open(ip_file)
                 ips=f.readlines()
                 f.close()
                 for ip in ips:
                     ip = ip.strip('\n')
                     if re.match(r'^#', ip):
                         continue
                     if len(sys.argv) == 5 and sys.argv[3] == '-c':
                         hostname = ip
                         comm = sys.argv[4]
                         perform_command(hostname, port, username, password, comm)
                     elif len(sys.argv) == 7 and sys.argv[3] == '-s':
                         hostname=ip
                         local_dir=sys.argv[4]
                         client_dir=sys.argv[5]
                         file=sys.argv[6]
                         send_file(hostname,port,username,password,local_dir,client_dir,file)
                     else:
                         help()
             else:
                 print "\033[31mERROR - Not found ip_file!\033[0m"
                 help()
         else:
             help()
     else:
         help()