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:\tmp\123.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:\tmp\123.txt]
 # @get_pdf  # 只获取pdf后缀列表。如：[D:\tmp\123.pdf]
 def all_path(all_path_list):
     return all_path_list

 if __name__ == '__main__':
     # 指定路径
     refer_path = r'D:\tmp'
     # 文件名称列表
     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