Python中的装饰器

浅浅的花香味﹌ 2022-04-24 09:02 171阅读 0赞

# Python中的装饰器 #

*  装饰器：本质就是一个函数，主要是为其他函数添加附加功能
    
     *  原则：不修改被修饰函数的源代码，不修改被修饰函数的调用方式。

#### 简单示例 ####

1.  加的原来一个加法器如下：

def add(x,y):
        print("开始计算{}+{}".format(x,y))
        hh = x + y
        print("结果为：{}".format(hh))
        return hh

1.  现在需要在不改变原代码的情况下，不修改调用方式的情况下，对加法运算前后增加相应功能  
    简单示实现如下：

#原加法器
    def add(x,y):
        print("开始计算{}+{}".format(x,y))
        hh = x + y
        print("结果为：{}".format(hh))
        return hh
    
    def strengthen(fn):
        def _add(*args,**kwargs):
            print("加法前执行")
            fn(*args,**kwargs) #这里使用原理的加法器做运算
            print("加法后执行")
        return _add
    
    add = strengthen(add)
    add(5,4)

![decorator003][]  
decorator

*  上面代码中strengthen方法就对add加法器进行了装饰。

#### 装饰器语法 ####

*  在python中提供了一种语法糖，来方便使用者实现上述这种装饰功能。  
    简单代码如下：

def strengthen(fn):
        def _add(*args,**kwargs):
            print("加法前执行")
            fn(*args,**kwargs) #这里使用原理的加法器做运算
            print("加法后执行")
        return _add 
    
    #表示用strengthen装饰add方法
    @strengthen    #注意@strengthen 等价于 add = strengthen(add) 
    def add(x,y):
        print("开始计算{}+{}".format(x,y))
        hh = x + y
        print("结果为：{}".format(hh))
        return hh
    
    add(5,4)

![decorator004][]

#### 装饰器语法糖，【@+表达式】 ####

@表达式  
函数  
等价于 函数名 = 表达式执行结果(函数名)

*  注意：表达式的返回值一定要是一个函数。  
    例如：

import datetime,time
    
    #定义修改注解函数，并将修改注解函数柯里化
    def logg(fn):
        def copy_property(tofn):
            tofn.__name__ = fn.__name__
            tofn.__doc__ = fn.__doc__
            return tofn  #注意，这里需要将原函数返回。不然元函数值为none
        return copy_property
    
    def strengthen(fn):
        """我是strengthen方法的注解"""
        @logg(fn) #等效于 wrapper = logg(fn)(wrapper)
        def wrapper(*args,**kwargs):
            """我是wraper方法的注解"""
            starttime = datetime.datetime.now()
            fn(*args,**kwargs) #这里使用原理的加法器做运算
            elapsedtime = datetime.datetime.now() - starttime
            print("耗时：{}".format(elapsedtime.total_seconds()))
        return wrapper 
    
    @strengthen    #等效于add = strengthen(add)
    def add(x,y):
        """我是add方法的注解"""
        print("开始计算{}+{}".format(x,y))
        time.sleep(1)
        hh = x + y
        print("结果为：{}".format(hh))
        return hh
    
    add(5,4)
    print(add.__name__)
    # datetime.datetime.now()

*  运行结果为：  
    ![decorator005][]
 *  上面示例中，不仅对add函数做了装饰，而且还改变了装饰器的默认属性与add属性保持一致。

[decorator003]: /images/20220217/c35048a28bc14d3daf1b215aa9c5cc90.png
[decorator004]: /images/20220217/bba0737dec124247957b34cf0461055b.png
[decorator005]: /images/20220217/eacaf018face4a2ba760b8a9b07d5720.png