如何区分装饰器中的函数和方法

Question

我正在尝试编写装饰器来完成不同的工作，具体取决于它是装饰方法还是函数。

我尝试过这样的事情：

def dec(f):
    def wrapper(*v, **kv):
        if '__class__' in dir(f):
            text = "Its a method"
        else:
            text = "Its a function"
        print(text)
        return f(*v, **kv)
    return wrapper


class A:

    @dec
    def f(self):
        pass

@dec
def f():
    pass

a = A()

a.f()
f()

返回：

Its a method
Its a method

但这不起作用，因为所有函数也都有__class__ 属性:) 我也尝试在 A.f 上执行inspect.ismethod，但 A.f 就像函数一样被对待，但在 A 命名空间中，它当然返回 false。如果我们完成了inspect.ismmethod(a.f)，它将返回True，其中a 是A 的实例，但我等不及要实例了。

Answer 1

正如 Martijn 在他的 cmets 中已经解释的那样，在类语句的主体中定义并不会使函数成为方法 - 方法对象是在查找时创建的，正如 here 所解释的那样 - 所以你正在装饰的是并且将会总是（几乎是下面的cf）是一个函数。 FWIW，Python 的“方法”只是函数、类和（最终）实例的薄包装。

除了函数之外，你可能会得到其他东西的唯一情况是类方法和静态方法（在这种情况下，你会得到一个 classmethod 或 staticmethod 对象），即使那样，它也取决于装饰器的顺序.

如果您想在装饰器中获取方法的当前实例（或类），您知道它将始终是第一个位置参数 - 但这可能没有太大帮助...长话短说，您必须明确告诉您的装饰器是否应该将函数视为普通函数或方法，或者用自定义描述符装饰它，在这种情况下，您将知道该函数是否已被查找为方法。 .. 一个简单的实现可能如下所示：

class MyDecorator(object):
    def __init__(self, func, instance=None):
        self.func = func
        self.instance = instance

    def __call__(self, *args, **kw):
        if self.instance:
            print "is method - instance : %s" % self.instance
            args = (self.instance,) + args
        else:
            print "is_function"

        return self.func(*args, **kw)

    def __get__(self, instance, cls):
        return type(self)(self.func, instance)