《Python cookbook》 “定义一个属性可由用户修改的装饰器” 笔记

Posted 2020-09-18 BlackMatrix

看《Python cookbook》的时候，第9.5部分，“定义一个属性可由用户修改的装饰器”，有个装饰器理解起来花了一些时间，做个笔记免得二刷这本书的时候忘了

完整代码：https://github.com/blackmatrix7/python-learning/blob/master/python_cookbook/chapter_9/section_5/attach_wrapper.py

书中的装饰器（书中称之为访问器函数）

def attach_wrapper(obj, func=None):
    if func is None:
        return partial(attach_wrapper, obj)
    setattr(obj, func.__name__, func)
    return func

这个访问器函数，接受两个参数，obj是需要处理的对象，func是被装饰的函数。实现的效果是将被装饰的对象，附加到obj中，使obj这个对象具有func这个方法。

当访问器函数，执行到func is None时，实际上是执行的是attach_wrapper(obj=wrapper)，刚刚将需要处理的对象obj传入。

        @attach_wrapper(wrapper)
        def set_message(newmsg):
            print(‘set message‘)
            nonlocal logmsg
            logmsg = newmsg

用过装饰器的话，一定可以知道，上面的set_message的装饰器，实际上等于下面的方法

attach_wrapper(wrapper)(set_message)

对上面这个语句，还可以继续拆分

访问器函数中判断，当func为None时，返回一个偏函数 return partial(attach_wrapper, obj)

这个偏函数实际上是生成了一个包装器，包装器本身接受被装饰的函数。

所以我们得到一个偏函数，也是包装器

partial_func = attach_wrapper(wrapper)

因为偏函数已经带有一个obj参数， 所以再次调用时，执行 setattr(obj, func.__name__, func)

此时给装饰器传入的对象，附加上被装饰的函数

partial_func(set_message)

如果还不明白，可以看下面的装饰器。

这个装饰器的功能，等价于上面的访问器函数

def my_attach_wrapper(obj):
    """
    如果这样写就非常容易理解了，把这个方法附加到被传入的对象中
    :param obj: 
    :return: 
    """
    def _my_attach_wrapper(func):
            setattr(obj, func.__name__, func)
    return _my_attach_wrapper