装饰器作用
代码运行期间动态增加功能，而不用修改原函数（被装饰的函数）的方式。使用@语法，把decorator置于函数的定义处。
装饰器原则
1.不改变被装饰函数的源代码
2.不改变被装饰函数的调用方式
装饰器的划分
装饰器可以划分为无参数装饰器与有参装饰器，带不带参数根据函数的需要自行设计。
装饰器演化过程
下面的代码是以无参数装饰器为例
1）新函数里调用原来函数
def old(x, y):
print(x, y)
def new(x, y):
print("new function")
old(x, y)
new(3, 4)
2）改进，将函数对象作为参数导入新函数中
def old(x, y):
print(x, y)
def new(func, x, y):
print("new function")
old = func(x, y)
return old
new(old, 3, 4)
3）继续改进，如果原函数需要任意个参数带入。则新函数需要定义可变位置参数与可变关键字参数。同时在调用原函数时，对输入的可变参数进行解构（参数符不符合需要在原函数中确认）。
def old(x, y):
print(x, y)
def new(func, *args, **kwargs):
print("new function")
old = func(*args, **kwargs)
return old
new(old, 3, 4)
4）对新函数进行柯里化，逐步形成闭包结构
def old(x, y):
print(x, y)
def new(func):
def inner(*args, **kwargs):
print("new function")
old = func(*args, **kwargs)
return old
return inner
old = new(old) # 后面old是原函数对象。通过新函数调用后返回inner对象给old标识符覆盖。inner函数封装着内存中的原函数的对象。 old = inner
old(3, 4)
5）改为装饰器
def new(func):
def inner(*args, **kwargs):
print("new function")
old = func(*args, **kwargs)
return old
return inner
@new #相当于old = new(old)或old=inner
def old(x, y):
print(x, y)
old(3, 4) # 相当于 new(old)(3, 4) 或 inner(3, 4)
装饰器练习
1）将下面字典d进行扁平化输出，并输出其最大节点深度
d = {'a':{'b':1, 'c':2}, 'd':{'e':3, 'f':{'g': 4}}}
d2 = {}
def outer(d2):
def fix(func):
def _fix(tmp = 0, tmp_index = "", *args, **kwargs):
ret = func(*args, **kwargs)
for i in d2:
if len(i.split(".")) > tmp:
tmp = len(i.split("."))
tmp_target = i
return ret, tmp, tmp_target
return _fix
return fix
@outer(d2)
def new(dic=d, key=""):
for k, v in dic.items():
if isinstance(v, dict):
new(dic=v, key=key+k+".")
else:
d2[key + k] = v
return d2
print(new())
