利用装饰器遍历文件
经常会对某个表格进行操作,然后要对文件夹内的所有表格都进行同样的操作。此时我们可以使用装饰器。
import time
import os
my_format = '.kva' # 对于某种文件,要进行相应的操作
def iter_files(func):
def wrapper():
for file in os.listdir():
if file.endswith(my_format):
func(file)
return wrapper
@iter_files
def foo(file_name):
print(file_name) # 具体操作
foo() # 直接调用即可