python装饰器系列(六)

可调用对象

在python中可以用callable函数查看一个对象是不是可调用

def fn():
    print('ha ha ha')

callable(fn)
True

输出为True，函数当然是一个可调用对象，如果是一个类呢？

class Fun:
    def __init__(self,name):
        self.name = name

f = Fun('zhaochj')

callable(f)
False

如上，一个类被实例化后，这个实例对象是一个不可调用对象

有什么方法能让一个实例对象变成可调用对象呢？做如下修改：

class Fun_1:
    def __init__(self,name):
        self.name = name
        
    def __call__(self):
        print('my name is {0}'.format(self.name))

f1 = Fun_1('zhaochj')

callable(f1)
True

f1这个实例现在已是一个可调用对象了。所以只要一个类中实现了__call__方法，那么类实例就是一个可调用对象

f1()
my name is zhaochj

事实证明，调用此实例对象也就是执行了__call__方法。既然是可调用对象，就可以向调用函数一样传递参数来调用，只要在__call__方法中定义可接收参数即可。

类通过实现__call__方法可以让实例变成一个可调用对象，如果我们向这个可调用对象传递一个函数作为其参数，那__call__函数就可以写成一个装饰器

import functools
class InjectUser:
    def __init__(self,default_user):
        self.user = default_user
        
    def __call__(self,fn):
        @functools.wraps(fn)
        def wrap(*args,**kwargs):
            if 'user' not in kwargs.keys():
                kwargs['user'] = self.user
            return fn(*args,**kwargs)
        return wrap

@InjectUser('zhaochj')
def do_somthings(*args,**kwargs):
    print(kwargs.get('user'))

do_somthings()
zhaochj

对上述的代码进行分析：

def __call__(self,fn):
        @functools.wraps(fn)
        def wrap(*args,**kwargs):
            if 'user' not in kwargs.keys():
                kwargs['user'] = self.user
            return fn(*args,**kwargs)
        return wrap

上边的代码是定义一个装饰器

@InjectUser('zhaochj')
def do_somthings(*args,**kwargs):
    print(kwargs.get('user'))

这里的魔法等价执行了InjectUser('zhaochj')(do_somthings)，实质是返回装饰器的wrap函数。执行do_somthings()时，实质是执行了wrap()，并返回fn(*args,**kwargs)，到这里才真正执行函数中的print语句

def do_somthings(*args,**kwargs):
    print(kwargs.get('user'))