Python-自定义装饰器

来源:互联网 发布:商业数据分析报告 编辑:程序博客网 时间:2024/06/06 19:27

什么是装饰器?装饰器本质是一个函数,它可以在不改变原来的函数的基础上额外的增加一些功能。如常见的@classmethod,@staticmethod等都是装饰器,接下来记录下如何自定义个装饰器:

刚刚说过了,装饰器的本质就是一个函数,所有想要自定义一个装饰器,首先自定义一个函数

def decorate(func):    def wrapper(*args,**kwargs):        print("定义一个装饰器")        func(*args,**kwargs)    return wrapper

此时就已经定义好了一个基本的装饰器,那该如何调用呢?

@decoratedef text1():    print("text1")text1()

输出的结果为:

定义一个装饰器text1

分析:此时的@decorate相当于将text1函数的内存地址传入decorate函数,并返回wrapper函数的内存地址。因此在代码结尾中调用text1()本质上是执行wrapper函数。因为执行的是wrapper函数,所以会打印”定义一个装饰器”,又因为func函数是text1的内存地址,所以调用func,会打印”text1”。

带参数的装饰器

上面介绍了一个简单的装饰器如何定义,可是我们常常看到一个装饰器@xxxxxxx(abc=”python”),这种装饰器是如何封装的,原理又是怎么样的呢

def decorate(name):    def wrapper(func):        def sub_wrapper(*args,**kwargs):            print("定义一个带参数的装饰器",name)            func(*args,**kwargs)        return sub_wrapper        return wrapper@decorate(name="python")def text1():    print("text1")text1()

输出结果:

定义一个带参数的装饰器 pythontext1

分析:带参数的装饰器与普通的装饰器多加了一层,其实就是讲“python”参数传入decorate函数,并返回wrapper函数的内存地址,再将text1函数内存地址传入wrapper函数,并返回了sub_wrapper函数的内存地址。而在代码末尾调用text1,其实本质是调用了sub_wrapper函数。

原创粉丝点击