获取装饰类方法的类名

Question

考虑这种情况：

import functools

def wrapmethod(f):
    @functools.wraps(f)
    def wrap(*args, **kwargs):
        print '>> %s' % (f.func_name)

        # Here I'll do pre-processing
        r = f(*args, **kwargs)
        # Here I'll do post-processing

        return r

    return wrap

@wrapmethod
def foo():
    pass

class Test(object):
    @wrapmethod
    def foo(self):
        pass

test = Test()
test.foo()
foo()

它会输出这个，正如你在http://codepad.org/Y4xXyjJO中看到的那样：

>> 富
>> 富

我想知道一种在第一行打印出Test.foo 的方法，表明该方法链接到的类。

有什么想法吗？有可能吗？

提前谢谢你。

Answer 1

其实你可以使用inspect模块来获取函数的签名，假设你遵循第一个参数'self'引用类对象的约定，你可以这样做：

import inspect  
def print_name(*_args):
    def _print_name(fn):
        def wrapper(*args, **kwargs):
            try :
                is_method   = inspect.getargspec(fn)[0][0] == 'self'
            except :
                is_method   = False

            if is_method :
                name    = '{}.{}.{}'.format(fn.__module__, args[0].__class__.__name__, fn.__name__)
            else :
                name    = '{}.{}'.format(fn.__module__, fn.__name__)

            print (name)
            return  fn(*args,**kwargs)
        return wrapper
    return _print_name

如果这是一个函数，这将打印方法模块、类和名称或者只是模块和名称

从 python 3.3 开始，fn.__qualname__ 可用于获取函数/方法的限定名。

def print_name(*args):
     def _print_name(fn):
         def wrapper(*args, **kwargs):
             print('{}.{}'.format(fn.__module__, fn.__qualname__))
             return fn(*args, **kwargs)
         return wrapper
     return _print_name

这适用于函数和方法：

In [1]: class A():
...:     @print_name()
...:     def a():
...:         print('Hi from A.a')
In [2]: A.a()
    __main__.A.a
    Hi from A.a

In [25]: @print_name()
    ...: def b():
    ...:     print('Hi from b')
    ...: 
In [26]: b()
    __main__.b
    Hi from b

Answer 2

这并不容易。 如果您将self 添加为内部函数的第一个参数，您可以使用self.__class__.__name__ 来访问类名，但是在装饰没有参数的无类函数时它会中断（如果它有参数，它会考虑第一个参数为self)。

因此，除非有办法确定函数是否已在对象上下文中调用，否则您想要做的事情是不可能的。

顺便说一句..你需要什么？这听起来像是可以用更好的方式解决的问题。

Answer 3

一个简单的版本：

def print_my_location(func):
    def wrapper(*args, **kwargs):
        print(f'[{func.__module__}::{func.__qualname__}]: ')
        return func(*args, **kwargs)

    return wrapper