# -*- coding: UTF8 -*-
# 2022-3-8
# 作者：小藍棗
# python連接ftp服務(wù)器
from ftplib import FTP

def conn_ftp():
    '''
     作用：連接ftp服務(wù)器
     參數(shù)：無
     返回：ftp服務(wù)器連接的對象
    '''
    
    # FTP連接信息
    ftp_ip = "xx.xx.xx.xx"
    # 默認(rèn)端口21
    ftp_port = 21
    # 如果未指定，使用默認(rèn)用戶名為Anonymous，密碼為空
    ftp_user = "Anonymous"
    ftp_password = ""

    ftp = FTP()
    # 連接ftp
    ftp.connect(ftp_ip, ftp_port)
    # ftp登錄
    ftp.login(ftp_user, ftp_password)
    # 查看歡迎信息
    print(ftp.getwelcome())
    
    return ftp
    
ftp = conn_ftp()

def display_dir(ftp, path):
    '''
     作用：進入并展示指定的目錄內(nèi)容
     參數(shù)1：ftp連接對象
     參數(shù)2：要展示的目錄
     返回：無
    '''
    
    # 進入指定目錄
    ftp.cwd(path)
    # 顯示當(dāng)前所在位置
    print("當(dāng)前所在位置為：")
    print(ftp.pwd())
    # 展示目錄內(nèi)容
    print("\n顯示目錄內(nèi)容：")
    print(ftp.dir())
    # 展示目錄下的文件名，*文件夾和文件都會顯示
    print("\n文件和文件夾名為：")
    for i in ftp.nlst():
        print(i)

path = "/CaseData/nc.vo.sdp.testcase.testcase.TestcaseHVO/"
display_dir(ftp, path)

def diff_dir(ftp, path):
    '''
     作用：區(qū)分文件和文件夾
     參數(shù)1：ftp連接對象
     參數(shù)2：要展示的目錄
     返回：無
    '''
    
    # 進入指定目錄
    ftp.cwd(path)
    # 顯示當(dāng)前所在位置
    print("當(dāng)前所在位置為：")
    print(ftp.pwd())
    # 展示目錄內(nèi)容
    print("\n顯示目錄內(nèi)容：")
    dirs = []
    ftp.dir(".", dirs.append)
    for i in dirs:
        # 區(qū)分文件和文件夾
        if("<DIR>" in i):
            print("目錄為：" + i.split(" ")[-1])
        else:
            print("文件為：" + i.split(" ")[-1])

path = "/CaseData/nc.vo.sdp.testcase.testcase.TestcaseHVO/"
diff_dir(ftp, path)

def get_dir_name(s):
    '''
     作用：需要文件或文件夾名
     參數(shù)1：需要截取的字符串
     返回：文件或文件夾名
    '''
    dir_name = ""
    k = 0
    record = ""
    for i in s:
        if(record == " " and i != " "):
            k = k + 1;
        if(k >= 3):
            dir_name = dir_name + i;
        record = i
        
    print(dir_name)
    return dir_name

# 測試兩條數(shù)據(jù)
get_dir_name("03-08-22  09:52AM       <DIR>          C2021.11_ZDHJC_004a")
get_dir_name("03-08-25  10:32AM                89098 hello .exe")

import pathlib

def download_dir(ftp, path, local_path):
    '''
     作用: 下載目錄
     參數(shù)1：ftp連接對象
     參數(shù)2：要展示的目錄
     參數(shù)3：本地存放路徑
     返回：無
    '''
    
    # 進入指定目錄
    ftp.cwd(path)
    # 區(qū)分文件和文件夾
    dirs = []
    ftp.dir(".", dirs.append)
    for i in dirs:
        try:
            # 識別為目錄進行遞歸
            if("<DIR>" in i):
                dir_name = get_dir_name(i)
                local_path_new = local_path + "/" + dir_name
                # 本地創(chuàng)建文件夾
                pathlib.Path(local_path_new).mkdir(parents=True, exist_ok=True)
                # 調(diào)用下載目錄方法
                download_dir(ftp, dir_name, local_path_new)
            # 識別為文件進行下載
            else:
                file_name = get_dir_name(i)
                local_filename = local_path + "/" + file_name
                f = open(local_filename, "wb")
                # 下載ftp文件
                ftp.retrbinary('RETR ' + file_name, f.write)
                f.close()
        except Exception as e:
            print(e)
    
    # 退出當(dāng)前目錄
    ftp.cwd("..")

def download_file(ftp, key, path, local_path):
    '''
     作用: 根據(jù)關(guān)鍵詞下載文件
     參數(shù)1：ftp連接對象
     參數(shù)2：下載的關(guān)鍵詞
     參數(shù)3：要展示的目錄
     參數(shù)4：本地存放路徑
     返回：無
    '''
    
    # 進入指定目錄
    ftp.cwd(path)
    # 區(qū)分文件和文件夾
    dirs = []
    ftp.dir(".", dirs.append)
    for i in dirs:
        if(key in i):
            try:
                # 識別為目錄進行遞歸
                if("<DIR>" in i):
                    dir_name = get_dir_name(i)
                    local_path_new = local_path + "/" + dir_name
                    # 本地創(chuàng)建文件夾
                    pathlib.Path(local_path_new).mkdir(parents=True, exist_ok=True)
                    # 調(diào)用下載目錄方法
                    download_dir(ftp, dir_name, local_path_new)
                else:
                    file_name = get_dir_name(i)
                    local_filename = local_path + "/" + file_name
                    f = open(local_filename, "wb")
                    # 下載ftp文件
                    ftp.retrbinary('RETR ' + file_name, f.write)
                    f.close()
            except Exception as e:
                print(e)

# 設(shè)置編碼，解決上傳的文件包含中文的問題
ftp.encoding = 'GBK'
key = "C2021.11_ZDHJC"
path = "/CaseData/nc.vo.sdp.testcase.testcase.TestcaseHVO/"
local_path = "D:/ftp下載"
download_file(ftp, key, path, local_path)

# 下載ftp文件
ftp.retrbinary('RETR ' + file_name, f.write)