Python实现连接FTP并下载文件夹

# -*- coding: UTF8 -*-
# 2022-3-8
# 作者：小蓝枣
# python连接ftp服务器
from ftplib import FTP

def conn_ftp():
    '''
     作用：连接ftp服务器
     参数：无
     返回：ftp服务器连接的对象
    '''
    
    # FTP连接信息
    ftp_ip = "xx.xx.xx.xx"
    # 默认端口21
    ftp_port = 21
    # 如果未指定，使用默认用户名为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):
    '''
     作用：进入并展示指定的目录内容
     参数1：ftp连接对象
     参数2：要展示的目录
     返回：无
    '''
    
    # 进入指定目录
    ftp.cwd(path)
    # 显示当前所在位置
    print("当前所在位置为：")
    print(ftp.pwd())
    # 展示目录内容
    print("\n显示目录内容：")
    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):
    '''
     作用：区分文件和文件夹
     参数1：ftp连接对象
     参数2：要展示的目录
     返回：无
    '''
    
    # 进入指定目录
    ftp.cwd(path)
    # 显示当前所在位置
    print("当前所在位置为：")
    print(ftp.pwd())
    # 展示目录内容
    print("\n显示目录内容：")
    dirs = []
    ftp.dir(".", dirs.append)
    for i in dirs:
        # 区分文件和文件夹
        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):
    '''
     作用：需要文件或文件夹名
     参数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

# 测试两条数据
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):
    '''
     作用: 下载目录
     参数1：ftp连接对象
     参数2：要展示的目录
     参数3：本地存放路径
     返回：无
    '''
    
    # 进入指定目录
    ftp.cwd(path)
    # 区分文件和文件夹
    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
                # 本地创建文件夹
                pathlib.Path(local_path_new).mkdir(parents=True, exist_ok=True)
                # 调用下载目录方法
                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)
    
    # 退出当前目录
    ftp.cwd("..")

def download_file(ftp, key, path, local_path):
    '''
     作用: 根据关键词下载文件
     参数1：ftp连接对象
     参数2：下载的关键词
     参数3：要展示的目录
     参数4：本地存放路径
     返回：无
    '''
    
    # 进入指定目录
    ftp.cwd(path)
    # 区分文件和文件夹
    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
                    # 本地创建文件夹
                    pathlib.Path(local_path_new).mkdir(parents=True, exist_ok=True)
                    # 调用下载目录方法
                    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)

# 设置编码，解决上传的文件包含中文的问题
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)