封装函数隐藏其属性？

这是我先前的问题Change an attribute of a function inside its own body的继续.

如果我包装了一个函数,以便它保留使用以下装饰器调用该函数的次数：

def keep_count(f):
    @wraps(f)
    def wrapped_f(*args, **kwargs):
        f(*args, **kwargs)
        wrapped_f.count += 1
    wrapped_f.count = 0
    return wrapped_f

然后我想用其他东西再次包裹它：

def decorator2(fn):
    @wraps(fn)
    def fn_wrapper(*args, **kwargs):
        if my_condition(fn):
            fn(*args, **kwargs)
    return fn_wrapper

test_f = decorator2(test_f)

我无法再按期望访问该函数的count属性.
count属性的当前值通过@wraps(fn)复制,但是如果我再次调用该函数,则count将在原始函数内递增,但是新值将不会复制到新的装饰函数中.

>>> test_f()
() {}
>>> test_f.count
1
>>> test_f = decorator2(test_f)
>>> test_f.count  # The 1 gets copied here
1
>>> test_f()  # Only the inner function's count increments...
() {}
>>> test_f.count  # Still one, tho it should be two
1

有什么解决办法吗？
像“不断地”重新包装,还是更好的东西？

解决方法:

functools.wraps()仅跨属性复制.然后在包装函数上增加计数器时,您正在为属性分配一个新的整数值,包装器仍将引用旧值.

与其在属性之间进行环绕复制,不如在函数的整个__dict__属性上复制：

from functools import wraps, WRAPPER_ASSIGNMENTS

def decorator2(fn):
    @wraps(fn, assigned=WRAPPER_ASSIGNMENTS + ('__dict__',), updated=())
    def fn_wrapper(*args, **kwargs):
        if my_condition(fn):
            fn(*args, **kwargs)
    return fn_wrapper

现在,包装的函数fn和fn_wrapper对象共享可变的__dict__名称空间字典,并且对该字典所做的任何更改在两个函数中均可见.

指定的是要复制的属性序列(通常复制诸如文档字符串,函数名称和模块名称之类的东西),而更新的则是应视为字典的属性序列,其中这些字典是从包装函数中更新的.后者通常设置为__dict__,但是既然我们复制了整个对象,则不再需要从原始对象进行更新.

标签：python-decorators,python-3-x,wrapper,decorator,python
来源： https://codeday.me/bug/20191028/1954964.html