通过 python的 __call__ 函数与元类 实现单例模式

简单一句话,当一个类实现__call__方法时,这个类的实例就会变成可调用对象。

直接上测试代码

class ClassA:

    def __call__(self, *args, **kwargs):
        print('call ClassA instance')


if __name__ == '__main__':
    # ClassA实现了__call__方法
    a = ClassA()
    '''
    这个时候,ClassA的实例a,就变成可调用对象
    调用a(),输出call ClassA instance,说明是调用了
    __call__函数
    '''
    a()
    # 其实a()等同于a.__call__(),它本质上就是后者的缩写
    a.__call__()
    # 判断是否可调用,输出True
    print(callable(a))

注意,是这个类的实例变成可调用对象,类的实例变成可调用对象,类的实例变成可调用对象,而不是改变这个类的实例化行为。

那么,如果要改变一个类的被实例化行为呢?

当然要用上黑魔法元类了,因为类本身就是元类的实例,当我们在元类中定义__call__的函数时,会改变类的实例化行为(或者说被调用的行为?感觉类和函数的界限有些模糊了)。

利用元类和__call__,可以在不使用工厂函数的情况,轻松实现单例模式,同时保持不错的可读性。

(以下代码来自《Python Cookbook》,进行部分修改,同时注释部分为个人理解)。

完整源码:https://github.com/blackmatrix7/python-learning/blob/master/class_/singleton.py

先定义一个名为Singleton的元类,实现如下

class Singleton(type):

    def __init__(cls, *args, **kwargs):
        cls.__instance = None
        super().__init__(*args, **kwargs)

    # __call__ 是对于类实例有效,比如说Spam类,是type类的实例
    def __call__(cls, *args, **kwargs):
        print('Singleton __call__ running')
        if cls.__instance is None:
            '''
            元类定义__call__方法,可以抢在类运行 __new__ 和 __init__ 之前执行,
            也就是创建单例模式的前提,在类实例化前拦截掉。
            type的__call__实际上是调用了type的__new__和__init__
            '''
            cls.__instance = super().__call__(*args, **kwargs)
            return cls.__instance
        else:
            return cls.__instance

使用元类创建类Spam

class Spam(metaclass=Singleton):

    def __new__(cls):
        print('Spam __new__ running')
        return super().__new__(cls)

    def __init__(self):
        print('Spam __init__ running')

 

解释下上面的代码,在元类Singleton的__init__函数中,给类增加了一个叫__instance的类属性。

需要注意的是,__init__的第一个参数是cls,其实等同于我们平时在类中定义__init__的self,因为对于元类来说,类是它的实例。

之所以写成cls,是便于理解 cls.__instance = None 是给类属性 __instance 赋值为 None。

 

接着在元类Singleton中重写__call__方法,__call__会抢在类(元类的实例)执行__new__和__init__之前执行,也就为拦截类的实例化提供了可能。

在元类Singleton的__call__方法对类属性__instance进行判断,如果__instance为None,说明类还未进行实例化,那么调用元类的父类(元类是type的子类)type的__call__方法,同时赋值给 cls.__instance。如果 cls.__instance 不为None,说明类已经进行过实例化,直接返回之前存储在类属性cls.__instance 中的类实例,即实现单例模式。

 

测试代码:

if __name__ == '__main__':
    a = Spam()
    b = Spam()
    print(a is b)
    c = Spam()
    print(a is c)

执行结果:

Singleton __call__ running
Spam __new__ running
Spam __init__ running
Singleton __call__ running
True
Singleton __call__ running
True

从运行结果上可以看出,每次尝试实例化Spam时,会被__call__函数拦截,所以会打印出:Singleton __call__ running

接着判断实例是否存在,在第一次运行时,实例不存在,创建类实例,并赋值给类属性__instance。

后续的几次尝试实例化Spam,因为Spam已经有实例存在,不在创建实例,实现了单例模式。

 

一个错误的例子

 

如果,我们把Spam的__new__改成下面这样,不返回任何结果,会有什么问题??

class Spam(metaclass=Singleton):

    def __new__(cls):
        print('Spam __new__ running')

    def __init__(self):
        print('Spam __init__ running')

还是执行之前的测试代码,得到下面的运行结果

Singleton __call__ running
Spam __new__ running
Singleton __call__ running
Spam __new__ running
True
Singleton __call__ running
Spam __new__ running
True

初看似乎没有什么问题,print(a is b) 和 print(a is c)都返回了True。

 

细心的话,会发现上面的输出结果有个很奇怪的问题,__init__方法是从未被执行!!!

对于此,官方文档是如此说明的

If __new__() does not return an instance of cls, then the new instance’s __init__() method will not be invoked.

当__new__函数没有返回这个类的实例时,__init__函数不会被调用。上面的示例函数,只是打印了文字,没有返回任何结果,所以不会执行__init__。

其实也容易理解,__init__需要类实例self参数,而__new__没有返回一个类实例,这样的话__init__自然无法运行。

 

但是,还有个问题,为什么每次Spam的__new__方法都会被运行??

因为 __new__ 不返回任何结果,那么__init__方法不会执行,实例从头到尾都没有被创建过(cls.__instance永远是None)。

在 执行 a = Spam() b = Spam() c = Spam()的过程中,每次cls.__instance为None,就调用type.__call__试图创建实例。

而每次执行到Spam的__new__方法时,会没有创建出任何实例,这样每次都不能成功创建实例。

所以,会有每次都执行Spam __new__ running的情况。

 

最后,因为 a、b、c 三个实例都是None,所以在做比较时,永远的True。

posted @ 2017-05-25 21:21  BlackMatrix  阅读(2915)  评论(2编辑  收藏  举报