【python】用python指令碼實現ansible的推送、下載、上傳檔案功能

ringoo_ming發表於2017-12-14
Python指令碼

該指令碼可以實現：上傳檔案、下載檔案、執行命令


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()


指令碼如下：

  1. #!/usr/bin/env python
  2. #定義函式
  3. import paramiko
  4. import os
  5. import datetime
  6. import sys
  7. import re
  8. #定義基礎命令格式
  9. def perform_command(hostname, port, username, password, comm):
  10.     try:
  11.         t=paramiko.Transport((hostname,port))
  12.         t.connect(username=username,password=password)
  13.         try:
  14.             print "\033[33mIP:\033[0m", hostname
  15.             print "\033[33mPerform command:\033[0m", comm
  16.             print '\033[32m###########################################################\033[0m'
  17.             s=paramiko.SSHClient()
  18.             s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  19.             s.connect(hostname = hostname, port = port, username = username, password = password)
  20.             stdin,stdout,stderr=s.exec_command(comm)
  21.             print stdout.read()
  22.             print '\033[32m###########################################################\033[0m'
  23.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
  24.             s.close()
  25.         except Exception:
  26.             print "\033[31mERROR - Perform error\033[0m"
  27.     except Exception:
  28.         print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname

  30. #定義傳送檔案格式
  31. def send_file(hostname,port,username,password,local_dir,client_dir,file):
  32.     try:
  33.         t=paramiko.Transport((hostname,port))
  34.         t.connect(username=username,password=password)
  35.         try:
  36.             print "\033[33mIP:\033[0m", hostname
  37.             print "\033[33mSend file:\033[0m", file
  38.             print "\033[33mClient_dir:\033[0m", client_dir
  39.             sftp=paramiko.SFTPClient.from_transport(t)
  40.             sftp.put(os.path.join(local_dir,file),os.path.join(client_dir,file))
  41.             t.close()
  42.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
  43.         except Exception:
  44.             print "\033[31mERROR - Perform error\033[0m"
  45.     except Exception:
  46.             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
  47. #定義下載檔案格式
  48. def get_file(hostname,port,username,password,local_dir,client_dir,file):
  49.     try:
  50.         t=paramiko.Transport((hostname,port))
  51.         t.connect(username=username,password=password)
  52.         try:
  53.             print "\033[33mIP:\033[0m", hostname
  54.             print "\033[33mGet file:\033[0m", file
  55.             print "\033[33mClient_dir:\033[0m", client_dir
  56.             sftp=paramiko.SFTPClient.from_transport(t)
  57.             sftp.get(os.path.join(client_dir,file),os.path.join(local_dir,file))
  58.             t.close()
  59.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
  60.         except Exception:
  61.             print "\033[31mERROR - Perform error\033[0m"
  62.     except Exception:
  63.             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
  64. #定義幫助檔案
  65. def help():
  66.     print """
  67. \033[33mHelp:

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

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

  73. Give you examples:

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

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

  82. If you want send file:

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

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

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

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

  100. #定義命令詳情

  102. if __name__ == '__main__':
  103. #定義賬號、埠、密碼等資訊
  104.     username='root'
  105.     port=2222
  106.     password='abcdefg'


  109.     if len(sys.argv) > 2:
  110.         if sys.argv[1] == '-H':
  111.             if len(sys.argv) == 5 and sys.argv[3] == '-c':
  112.                 hostname=sys.argv[2]
  113.                 comm = sys.argv[4]
  114.                 perform_command(hostname, port, username, password, comm)
  115.             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
  116.                 hostname=sys.argv[2]
  117.                 local_dir=sys.argv[4]
  118.                 client_dir=sys.argv[5]
  119.                 file=sys.argv[6]
  120.                 send_file(hostname,port,username,password,local_dir,client_dir,file)
  121.             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
  122.                 hostname=sys.argv[2]
  123.     username='root'
  124.     port=2222
  125.     password='abcdefg'


  128.     if len(sys.argv) > 2:
  129.         if sys.argv[1] == '-H':
  130.             if len(sys.argv) == 5 and sys.argv[3] == '-c':
  131.                 hostname=sys.argv[2]
  132.                 comm = sys.argv[4]
  133.                 perform_command(hostname, port, username, password, comm)
  134.             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
  135.                 hostname=sys.argv[2]
  136.                 local_dir=sys.argv[4]
  137.                 client_dir=sys.argv[5]
  138.                 file=sys.argv[6]
  139.                 send_file(hostname,port,username,password,local_dir,client_dir,file)
  140.             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
  141.                 hostname=sys.argv[2]
  142.                 local_dir=sys.argv[4]
  143.                 client_dir=sys.argv[5]
  144.                 file=sys.argv[6]
  145.                 get_file(hostname,port,username,password,local_dir,client_dir,file)
  146.             else:
  147.                 help()
  148.         elif sys.argv[1] == '-F':
  149.             ip_file=sys.argv[2]
  150.             file_stat=os.path.exists(ip_file)
  151.             if file_stat:
  152.                 f=open(ip_file)
  153.                 ips=f.readlines()
  154.                 f.close()
  155.                 for ip in ips:
  156.                     ip = ip.strip('\n')
  157.                     if re.match(r'^#', ip):
  158.                         continue
  159.                     if len(sys.argv) == 5 and sys.argv[3] == '-c':
  160.                         hostname = ip
  161.                         comm = sys.argv[4]
  162.                         perform_command(hostname, port, username, password, comm)
  163.                     elif len(sys.argv) == 7 and sys.argv[3] == '-s':
  164.                         hostname=ip
  165.                         local_dir=sys.argv[4]
  166.                         client_dir=sys.argv[5]
  167.                         file=sys.argv[6]
  168.                         send_file(hostname,port,username,password,local_dir,client_dir,file)
  169.                     else:
  170.                         help()
  171.             else:
  172.                 print "\033[31mERROR - Not found ip_file!\033[0m"
  173.                 help()
  174.         else:
  175.             help()
  176.     else:
  177.         help()

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/31442725/viewspace-2148711/，如需轉載，請註明出處，否則將追究法律責任。

相關文章