Python-装饰器-案例-获取文件列表

import os


def get_all_path(fun):
    '''装饰器。功能：获取全路径文件名。如：D:/tmp/12.txt
    :param fun:
    :return:file_path_list  全路径文件名列表
    '''

    def inner(r):
        # print("获取全路径文件名")
        file_name_list = fun(r)
        file_path_list = []
        for file in file_name_list:
            path_file = refer_path + os.sep + file
            file_path_list.append(path_file)
        return file_path_list

    return inner


def get_pdf(fun):
    '''装饰器。功能：获取后缀为pdf、PDF的文件名列表
    :param fun:
    :return: file_pdf_list后缀为pdf、PDF的文件名列表
    '''

    def inner(r):
        # print("获取后缀为.pdf的文件名列表")
        file_name_list = fun(r)
        file_pdf_list = []
        for file in file_name_list:
            ext = os.path.splitext(file)[1]
            if ext.lower() == '.pdf':  # 判断后缀是否为.pdf、.PDF
                file_pdf_list.append(file)
        return file_pdf_list

    return inner


def get_suffix_out(suf="txt"):
    '''装饰器。功能：获取后缀为指定后缀（suf）的文件名列表
    :param suf:指定后缀，如：txt、jpg
    :return:suf_list符合指定后缀的文件列表
    '''

    def get_suffix(fun):
        def inner(*args, **kwargs):
            # print("inner 后缀限制是：", suf)
            # # print("获取后缀为.pdf的文件名列表")
            file_name_list = fun(*args, **kwargs)
            suf_list = []
            for file in file_name_list:
                src_suf = os.path.splitext(file)[1]  # 获取文件的后缀，如：.pdf、.txt
                if src_suf.lower() == '.' + str(suf).lower():  # 判断后缀是否为.suf、.SUF
                    suf_list.append(file)
            return suf_list

        return inner

    return get_suffix


def get_papers_only(fun):
    '''装饰器。功能：只获取列表中的文件名。不含文件夹。如：1.txt、123.pdf
    :param fun:
    :return: file_name_only_list 只有文件名，不含文件夹的列表
    '''

    def inner(r):
        # print("只获取文件名，不含文件夹")
        file_name_list = fun(r)
        # 只获取文件名，不含文件夹列表。如：1.txt、123.pdf
        file_name_only_list = []
        for file in file_name_list:
            if os.path.isabs(file):  # 1、如果为绝对路径，r表示绝对路径全路径列表
                if not os.path.exists(file):  # 检查全路径的有效性
                    print("路径不存在")
                    exit()
                if os.path.isfile(file):  # 判断是否为文件（非文件夹）
                    file_name_only_list.append(file)
            else:  # 2、如果为相对路径，r表示相对路径统一路径字符串
                path_file = r + os.sep + file
                if not os.path.exists(path_file):  # 检查全路径的有效性
                    print("路径不存在")
                    exit()
                if os.path.isfile(path_file):  # 判断是否为文件（非文件夹）
                    file_name_only_list.append(file)
        return file_name_only_list

    return inner


# 巧用装饰器做过滤条件
@get_all_path  # 获取全路径文件列表。如：[D:	mp123.pdf]
@get_pdf  # 获取pdf后缀列表。如：[123.pdf]
@get_papers_only  # 只获取文件列表，不含文件夹。如：[a、b、1.txt、123.pdf]
def get_list(refer_path):
    '''获取指定路径下的所有文件列表（含文件夹）。如：a、b、1.txt、123.pdf
    :param refer_path: 指定存放文件的路径
    :return:路径下的所有文件名
    '''
    try:
        file_name_list = os.listdir(refer_path)
        return file_name_list
    except Exception as e:
        print(e)
        exit()


@get_papers_only  # 只获取文件列表，不含文件夹。如：[a、b、1.txt、123.pdf]
@get_suffix_out("txt") # 只获取txt后缀列表。如：[D:	mp123.txt]
# @get_pdf  # 只获取pdf后缀列表。如：[D:	mp123.pdf]
def all_path(all_path_list):
    return all_path_list


if __name__ == '__main__':
    # 指定路径
    refer_path = r'D:	mp'
    # 文件名称列表
    file_name_list = get_list(refer_path)
    print(file_name_list)

    path_list = ['D:\tmp\123.PDF',
                 'D:\tmp\combinedminutes.txt',
                 'D:\tmp\mark.pdf',
                 'D:\tmp\meetingminutes.ps',
                 'D:\tmp\meetingminutes2.123',
                 'D:\tmp\rotatedpage.pdf',
                 'D:\tmp\watermarkedCover.pdf']

    print(path_list)
    pdf_list = all_path(path_list)
    print(pdf_list)

==================优美的分割线==================

装饰器学习案例：

例子1：简单装饰器

def check(fun):
    def inner():
        print("权限检查--1")
        fun()

    return inner


@check
def foo1():
    print("查看用户信息")


if __name__ == '__main__':
    foo1()

例子2：被装饰函数待有参数的装饰器

def check(fun):
    def inner(*args, **kwargs):
        print("权限检查--1")
        fun(*args, **kwargs)

    return inner


@check
def foo1():
    print("查看用户信息")


if __name__ == '__main__':
    foo1()

例子3：装饰器带参数的装饰器

def check_out(who):
    def check(fun):
        def inner(*args, **kwargs):
            print("权限检查--1，部门时：%s" % who)
            fun(*args, **kwargs)

        return inner

    return check


@check_out("天猫")
def foo1(a, b):
    print("查看用户信息%s" % (a + b))


@check_out("闲鱼")
def foo2(a, b, c):
    print("查看用户信息%s" % (a + b + c))


if __name__ == '__main__':
    foo1(1, 2)
    foo2(1, 2,3)

例子4：类语法版本的装饰器

class MyClass(object):
    def __init__(self, fun):
        print('__init__函数：fun=', fun)
        self.fun = fun

    def __call__(self, *args, **kwargs):
        print("验证权限")
        return self.fun(*args, **kwargs)


@MyClass
def foo1(a, b):
    print("%s + %s = %s" % (a, b, (a + b)))


if __name__ == '__main__':
    foo1(1,2)

例子5：文档恢复为被装饰器的文档的装饰器

import functools

def check(fun):
    """检查权限的装饰器"""
    @functools.wraps(fun)
    def inner(*args, **kwargs):
        """这里是闭包函数"""
        print("权限检查--1")
        fun(*args, **kwargs)

    return inner

@check
def foo1():
    '''这里是原始函数foo1的说明文档'''
    print("查看用户信息")

if __name__ == '__main__':
    print(foo1.__doc__)

参考：https://www.cnblogs.com/andy9468/p/11137417.html