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