函数装饰器和类装饰器的使用方法

这篇文章主要为大家详细介绍了函数装饰器和类装饰器的使用方法,文中示例代码介绍的非常详细,零基础也能参考此文章,感兴趣的小伙伴们可以参考一下。

创新新互联,凭借10多年的网站设计、网站建设经验,本着真心·诚心服务的企业理念服务于成都中小企业设计网站有1000多家案例。做网站建设,选创新互联

函数装饰器实现

装饰器算是类里面比较难的内容之一,但是实际上它的思想并不复杂。简单点说,就是在你原来内容的基础上,在外面给你加点东西,实现类似装饰的效果。但是它是怎么实现的呢?一般来说,都是通过拦截函数调用来实现的,比如:用装饰器装饰函数的时候,它拦截函数调用,装饰类的时候,它拦截类实例的创建调用,即拦截类初始化__init__函数。
知道这个原理以后,我们就可以来尝试实现了。

首先来看通过函数装饰器拦截类的创建过程,代码如下:

instance = {}
def createInstance(cls, *args):
    if cls not in instance:
        instance[cls] = cls(*args)
    return instance[cls]

def singleIns(cls):
    def onCall(*args):
        return createInstance(cls, *args)
    return onCall

上面就是这个函数装饰器singleIns的实现,它返回一个函数调用,当用它来装饰一个类,创建类实例的时候,就会用onCall方法拦截类的__init__方法。我们再来看一下它怎么使用的。

@singleIns
class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def __str__(self):
        return "{} 的年龄是{}".format(self.name, self.age)
zhangsan = Person('zhangsan', 30)
lisi = Person('lisi', 29)
print(zhangsan)
print(lisi)

最终的输出结果是:

zhangsan 的年龄是30
zhangsan 的年龄是30

为什么结果是一样的?因为在创建实例的过程中,__init__函数被onCall函数拦截,此时会进入到createInstance函数的流程中,会对这个类实例进行判断,如果不存在这个类的实例,那么就初始化一个后返回,如果存在,直接返回第一个创建的类实例。因此最终只有一个类实例存在,就实现类单例类。

类装饰器实现

上面最开始的位置我们说了,函数装饰器和类装饰器都是拦截函数调用,在函数装饰器实现类调用拦截的地方我们看到,它是通过函数装饰器内部的函数来实现拦截的。如果是类装饰器呢,它通过什么来拦截呢?
答案是call函数来拦截,我们来看一下类装饰器的实现代码:

class singleIns:
    def __init__(self, cls):
        self.cls = cls
        self.ins = None
    def __call__(self, *args):
        if self.ins is None:
            self.ins = self.cls(*args)
        return self.ins

代码和函数装饰器相比,其实功能没有太多变化,通过__call__方法来接收被拦截类的初始化函数参数args,然后用args来初始化类实例。但是只在这个类还没有实例的情况下进行初始化,否则直接返会初始化好的类。
我们来看一下应用的代码:

@singleIns
class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def __str__(self):
        return "{} 的年龄是{}".format(self.name, self.age)
zhangsan = Person('zhangsan', 30)
lisi = Person('lisi', 29)
print(zhangsan)
print(lisi)

最终的结果和上面函数装饰器的一样,如下所示:

zhangsan 的年龄是30
zhangsan 的年龄是30

看完上述内容,你们对函数装饰器和类装饰器的使用方法大概了解了吗?如果想了解更多相关文章内容,欢迎关注创新互联行业资讯频道,感谢各位的阅读!


名称栏目:函数装饰器和类装饰器的使用方法
网页网址:http://scyanting.com/article/jepheg.html