python装饰器

装饰器作用

        代码运行期间动态增加功能，而不用修改原函数（被装饰的函数）的方式。使用@语法，把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())

---