Python - 使用 decorator.py 保存方法文档字符串

Question

我有兴趣将实例方法作为类方法和实例方法来调用。这可以通过使用 class_or_instance 装饰器来完成，如下所示：

class class_or_instance(object):
    def __init__(self, fn):
        self.fn = fn

    def __get__(self, obj, cls):
        if obj is not None:
            return lambda *args, **kwds: self.fn(obj, *args, **kwds)
        else:
            return lambda *args, **kwds: self.fn(cls, *args, **kwds)

class A(object):
    @class_or_instance
    def func1(self,*args):
         # method body

现在我可以将 func1 称为 A.func1(*args) 或A().func1(*args)。 然而，当这样做时，func1 的文档字符串消失了。解决这个问题的一种方法是使用来自decorator.py 的装饰器，但我无法让它与作为类而不是函数的装饰器一起使用。有什么建议可以解决这个问题吗？

编辑：functools.wraps() 在这种情况下将无法正常工作。请参阅 stackoverflow 上的 related question

Answer 1

基本描述符/装饰器

你只需要记住应该装饰哪个函数。您的函数是在__get__ 中创建的，因此将包装器用作装饰器无济于事，相反，您需要在__get__ 方法中应用它。顺便说一句，您可以为此使用functools.update_wrapper 或decorators.decorator。它们的工作方式非常相似，只是您必须保留decorators.decorator 的结果，而functools.update_wrapper 返回None。两者都有签名f(wrapper, wrapped)。

from functools import update_wrapper
class class_or_instance(object):
    def __init__(self, fn):
        self.fn = fn

    def __get__(self, obj, cls):
        if obj is not None:
            f = lambda *args, **kwds: self.fn(obj, *args, **kwds)
        else:
            f = lambda *args, **kwds: self.fn(cls, *args, **kwds)
        # update the function to have the correct metadata
        update_wrapper(f, self.fn)
        return f

class A(object):
    @class_or_instance
    def func1(self,*args):
        """some docstring"""
        pass

现在如果你这样做：

print A.func1.__doc__

你会看到“一些文档字符串”。耶！

---

缓存属性装饰器

这里的关键是您只能影响返回的内容。由于class_or_instance 实际上并没有用作函数，所以你用它做什么并不重要。请记住，此方法会导致函数每次都被反弹。我建议你添加一点魔法，并在第一次调用后绑定/缓存函数，这实际上只涉及添加一个setattr 调用。

from functools import update_wrapper
import types

class class_or_instance(object):
    # having optional func in case is passed something that doesn't have a correct __name__
    # (like a lambda function)
    def __init__(self, name_or_func):
        self.fn = fn
        self.name = fn.__name__

    def __get__(self, obj, cls):
        print "GET!!!"
        if obj is not None:
            f = lambda *args, **kwds: self.fn(obj, *args, **kwds)
            update_wrapper(f, self.fn)
            setattr(obj, self.name, types.MethodType(f, obj, obj.__class__))
        else:
            f = lambda *args, **kwds: self.fn(cls, *args, **kwds)
            update_wrapper(f, self.fn)
        return f

然后我们可以测试一下...neato：

A.func1 #GET!!!
obj = A()
obj.func1 #GET!!!
obj.func1 is obj.func1 # True
A.func1 # GET!!!
obj2 = A()
obj2.func1 is not obj.fun1 # True + GET!!!