1 import os
  2
  3
  4 def get_all_path(fun):
  5     '''装饰器。功能：获取全路径文件名。如：D:/tmp/12.txt
  6     :param fun:
  7     :return:file_path_list  全路径文件名列表
  8     '''
  9
 10     def inner(r):
 11         # print("获取全路径文件名")
 12         file_name_list = fun(r)
 13         file_path_list = []
 14         for file in file_name_list:
 15             path_file = refer_path + os.sep + file
 16             file_path_list.append(path_file)
 17         return file_path_list
 18
 19     return inner
 20
 21
 22 def get_pdf(fun):
 23     '''装饰器。功能：获取后缀为pdf、PDF的文件名列表
 24     :param fun:
 25     :return: file_pdf_list后缀为pdf、PDF的文件名列表
 26     '''
 27
 28     def inner(r):
 29         # print("获取后缀为.pdf的文件名列表")
 30         file_name_list = fun(r)
 31         file_pdf_list = []
 32         for file in file_name_list:
 33             ext = os.path.splitext(file)[1]
 34             if ext.lower() == '.pdf':  # 判断后缀是否为.pdf、.PDF
 35                 file_pdf_list.append(file)
 36         return file_pdf_list
 37
 38     return inner
 39
 40
 41 def get_suffix_out(suf="txt"):
 42     '''装饰器。功能：获取后缀为指定后缀（suf）的文件名列表
 43     :param suf:指定后缀，如：txt、jpg
 44     :return:suf_list符合指定后缀的文件列表
 45     '''
 46
 47     def get_suffix(fun):
 48         def inner(*args, **kwargs):
 49             # print("inner 后缀限制是：", suf)
 50             # # print("获取后缀为.pdf的文件名列表")
 51             file_name_list = fun(*args, **kwargs)
 52             suf_list = []
 53             for file in file_name_list:
 54                 src_suf = os.path.splitext(file)[1]  # 获取文件的后缀，如：.pdf、.txt
 55                 if src_suf.lower() == '.' + str(suf).lower():  # 判断后缀是否为.suf、.SUF
 56                     suf_list.append(file)
 57             return suf_list
 58
 59         return inner
 60
 61     return get_suffix
 62
 63
 64 def get_papers_only(fun):
 65     '''装饰器。功能：只获取列表中的文件名。不含文件夹。如：1.txt、123.pdf
 66     :param fun:
 67     :return: file_name_only_list 只有文件名，不含文件夹的列表
 68     '''
 69
 70     def inner(r):
 71         # print("只获取文件名，不含文件夹")
 72         file_name_list = fun(r)
 73         # 只获取文件名，不含文件夹列表。如：1.txt、123.pdf
 74         file_name_only_list = []
 75         for file in file_name_list:
 76             if os.path.isabs(file):  # 1、如果为绝对路径，r表示绝对路径全路径列表
 77                 if not os.path.exists(file):  # 检查全路径的有效性
 78                     print("路径不存在")
 79                     exit()
 80                 if os.path.isfile(file):  # 判断是否为文件（非文件夹）
 81                     file_name_only_list.append(file)
 82             else:  # 2、如果为相对路径，r表示相对路径统一路径字符串
 83                 path_file = r + os.sep + file
 84                 if not os.path.exists(path_file):  # 检查全路径的有效性
 85                     print("路径不存在")
 86                     exit()
 87                 if os.path.isfile(path_file):  # 判断是否为文件（非文件夹）
 88                     file_name_only_list.append(file)
 89         return file_name_only_list
 90
 91     return inner
 92
 93
 94 # 巧用装饰器做过滤条件
 95 @get_all_path  # 获取全路径文件列表。如：[D:\tmp\123.pdf]
 96 @get_pdf  # 获取pdf后缀列表。如：[123.pdf]
 97 @get_papers_only  # 只获取文件列表，不含文件夹。如：[a、b、1.txt、123.pdf]
 98 def get_list(refer_path):
 99     '''获取指定路径下的所有文件列表（含文件夹）。如：a、b、1.txt、123.pdf
100     :param refer_path: 指定存放文件的路径
101     :return:路径下的所有文件名
102     '''
103     try:
104         file_name_list = os.listdir(refer_path)
105         return file_name_list
106     except Exception as e:
107         print(e)
108         exit()
109
110
111 @get_papers_only  # 只获取文件列表，不含文件夹。如：[a、b、1.txt、123.pdf]
112 @get_suffix_out("txt") # 只获取txt后缀列表。如：[D:\tmp\123.txt]
113 # @get_pdf  # 只获取pdf后缀列表。如：[D:\tmp\123.pdf]
114 def all_path(all_path_list):
115     return all_path_list
116
117
118 if __name__ == '__main__':
119     # 指定路径
120     refer_path = r'D:\tmp'
121     # 文件名称列表
122     file_name_list = get_list(refer_path)
123     print(file_name_list)
124
125     path_list = ['D:\\tmp\\123.PDF',
126                  'D:\\tmp\\combinedminutes.txt',
127                  'D:\\tmp\\mark.pdf',
128                  'D:\\tmp\\meetingminutes.ps',
129                  'D:\\tmp\\meetingminutes2.123',
130                  'D:\\tmp\\rotatedpage.pdf',
131                  'D:\\tmp\\watermarkedCover.pdf']
132
133     print(path_list)
134     pdf_list = all_path(path_list)
135     print(pdf_list)

 1 def check(fun):
 2     def inner():
 3         print("权限检查--1")
 4         fun()
 5
 6     return inner
 7
 8
 9 @check
10 def foo1():
11     print("查看用户信息")
12
13
14 if __name__ == '__main__':
15     foo1()

 1 def check(fun):
 2     def inner(*args, **kwargs):
 3         print("权限检查--1")
 4         fun(*args, **kwargs)
 5
 6     return inner
 7
 8
 9 @check
10 def foo1():
11     print("查看用户信息")
12
13
14 if __name__ == '__main__':
15     foo1()

 1 def check_out(who):
 2     def check(fun):
 3         def inner(*args, **kwargs):
 4             print("权限检查--1，部门时：%s" % who)
 5             fun(*args, **kwargs)
 6
 7         return inner
 8
 9     return check
10
11
12 @check_out("天猫")
13 def foo1(a, b):
14     print("查看用户信息%s" % (a + b))
15
16
17 @check_out("闲鱼")
18 def foo2(a, b, c):
19     print("查看用户信息%s" % (a + b + c))
20
21
22 if __name__ == '__main__':
23     foo1(1, 2)
24     foo2(1, 2,3)

 1 class MyClass(object):
 2     def __init__(self, fun):
 3         print('__init__函数：fun=', fun)
 4         self.fun = fun
 5
 6     def __call__(self, *args, **kwargs):
 7         print("验证权限")
 8         return self.fun(*args, **kwargs)
 9
10
11 @MyClass
12 def foo1(a, b):
13     print("%s + %s = %s" % (a, b, (a + b)))
14
15
16 if __name__ == '__main__':
17     foo1(1,2)

 1 import functools
 2
 3 def check(fun):
 4     """检查权限的装饰器"""
 5     @functools.wraps(fun)
 6     def inner(*args, **kwargs):
 7         """这里是闭包函数"""
 8         print("权限检查--1")
 9         fun(*args, **kwargs)
10
11     return inner
12
13 @check
14 def foo1():
15     '''这里是原始函数foo1的说明文档'''
16     print("查看用户信息")
17
18 if __name__ == '__main__':
19     print(foo1.__doc__)