#!/bin/python
#coding=utf-8
#python-version=2.75           #使用python2
from ftplib import FTP    #引用ftplib庫中的FTP功能模塊,進行ftp下載使用
import time           #引用time模塊
import os            #引用os模塊
"""
使用字典，定義交換機主機，一個字典包含多個鍵 ，一個鍵使用一個列表，包含多個主機地址按照實際情況定義
"""
dic = {
  'tongjiju':['XXXX.XXXX.XXXX.XXXX',
     'XXXX.XXXX.XXXX.XXXX',
     'XXXX.XXXX.XXXX.XXXX'],
    'dishui':['XXXX.XXXX.XXXX.XXXX','XXXX.XXXX.XXXX.XXXX']
}

"""
定義函數save，hosts
"""
def save(hosts, ftp):             #函數名save，兩個參數，hosts和ftp
  url = '/root/backup_new/{}'.format(hosts)      #定義下載文件保存路徑和目錄url ，位置為/root/backup_new/,按照hosts內容格式化名稱
  Today = time.strftime("%Y-%m-%d", time.localtime())      #定義時間，按照本地時間年月日定義  
  if not os.path.exists(url):                  #判斷url文件是否存在，不存在
    os.mkdir(url)                      #則新建目錄url
  for host in dic[hosts]:                    #循環，逐一循環字典中鍵和鍵值
    os.chdir(url)                      #切換目錄到url（/root/backup_new/）
    if not os.path.exists(host):              #判斷如果沒有定義和創建host(按照列表中的主機地址）的文件目錄
      os.mkdir(host)                  #創建host目錄
      os.chdir(host)                  #切換到host目錄
    else:                          #否則（當host目錄存在時）
      os.chdir(host)                  #切換到host目錄
    try:                           #錯誤處理機制try，當發生錯誤時，按照指定方式進行，因為有時主機不可達，ftp連接不上會報錯
      ftp.connect(host, '21')             #ftp.connect方法，主機為host（列表中地址）,端口21
      ftp.login('admin', 'xxxx')           #ftp.login方法，賬戶名為admin,密碼為xxxx
  print "login " + host                    #打印 login和host,例如 login 172.31.216.2
  print ftp.getwelcome()                     #ftp.getwelcome方法，打印ftp登錄成功信息
      bufsize = 1024                 #緩存區大小1024
      filename = "{}.zip".format(Today)       #ftp下載的文件按照今天(Today前面定義過）時間命名，例如2018=07-01.zip
      file_handle = open(filename, "wb").write   #以ftp讀寫的打開文件filename
      ftp.retrbinary("RETR vrpcfg.zip", file_handle, bufsize)  #ftp.retrbinary方法，以二進制的方式下載vrpcfg.zip的文件（華為交換機的配置文件），然后以讀寫的方式保存
  print host + " ftp down ok"                #打印host和"ftp down ok"，例如172.31.216.2 ftp down ok
    except Exception as e:                #如果以上try語句部分發生錯誤，ftp不可達到，則跳轉到該處，執行，整個過程不中斷
      print '{} is loss , msg:-{}'.format(host, e)  #打印錯誤信息，例如172.31.216.2 is loss,msg:-error
 """
 定義主函數
 """
def main():                               #函數名
  ftp = FTP()                           #定義ftp=FTP()空函數
  # ftp.set_debuglevel(2)                     #打開ftp調試開關，有0-3不同級別，打開后顯示ftp詳細信息，可以選擇使用
  # 0主動模式 1 #被動模式
  ftp.set_pasv(0)                          #這里使用ftp.set_pasv方法，確定使用主動模式，一般ftp采用被動模式，有很大機率登錄異常
  for hosts in dic.keys():                    #循環dic字典的鍵值
    save(hosts, ftp)                     #調用save函數
  # 關閉調試模式
  # ftp.set_debuglevel(0)                     #關閉調試開關，前面如果打開了開關，則這里需要關閉
  ftp.quit()                            #ftp退出，必須使用
main()