通用装饰器;
可以装饰任意参数的函数:(无参数也可以)
重点:
1. 闭包中内部函数的参数类型
2.内部函数的返回值上
需求:给函数func1增加日志功能(不能修改源码)
import time
def funcOut(func):
def funcInfo(*args, **kwargs):
log(func)
return func(*args, **kwargs)
return funcInfo
# 重点!!!通用装饰器
def log(func):
# 增加日志功能
f = open("log.txt", mode="a", enconding="utf-8")
# time.asctime()得到函数执行时的时间
time_str = time.asctime()
# 通过函数得到函数名__name__
func_name = func.__name__
# 追加内容至txt文档
f.write(time_str + "\t" + func_name + "\n")
# 关闭文档
f.close()
@funcOut
# 等同于func1 = funcOut(func1)
def func1(a):
print("a = {}".format(a))
func1(10)
# a = 10
# log.txt 文档中增加了时间日志
@funcOut
def func2(a, b, c):
print(a, b, c)
return 1000
func2(10, 100, {"name":"尼古拉斯"})
# 10 100 {"name":"尼古拉斯"}
# 1000
# log.txt 文档中增加了时间日志