Python装饰器通用样式

Posted 2020-07-27 MacoLee

装饰器本质上是一个函数，该函数用来处理其他函数，它可以让其他函数在不需要修改代码的前提下增加额外的功能，装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等应用场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。概括的讲，装饰器的作用就是为已经存在的对象添加额外的功能。-----该段内容摘自知乎

我们可以定义一个通用的装饰器，当需要使用装饰器时，可以直接套用这个样式。

下面是一个通用的装饰器样式：

def Decoroator(befor_func,after_func):
    def outer(main_func):
        def wrapper(*args,**kwargs):
            #主函数之前执行的函数
            befor_result = befor_func(*args,**kwargs)
            if befor_result != None:
                return  befor_result

            #主函数
            main_result = main_func(*args,**kwargs)
            if main_result != None:
                return main_result

            #主函数之后执行的函数
            after_result = after_func(*args,**kwargs)
            if after_result != None:
                return after_result
        return wrapper
    return outer

#主函数之前执行的函数
def befor_func(*args,**kwargs):
    pass
#主函数之后执行的函数
def after_func(*args,**kwargs):
    pass

@Decoroator(befor_func,after_func)
def test(*args,**kwargs):   #主函数
    pass