python切面/装饰器

参考文章

装饰器

装饰器

用过java都知道，java的AOP非常方便，它可以在调用方法之前和之后做一下操作，如打印日志。python 也有类似的功能，就是 装饰器。我们就以打印日志为例。

没有打印日志之前

def buy(good_name):
    print("购买商品 %s " % good_name)


def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

运行结果：

购买商品 10斤苹果 
支付金额 30 

在函数调用前自动打印日志

打印日志函数

import functools

# 在函数调用前自动打印日志
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print("call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        return func(*args, **kw)

    return wrapper

在调用函数上添加注解

@log
def buy(good_name):
    print("购买商品 %s " % good_name)


@log
def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

运行结果：

call  function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
call  function: payment args: (30,) kw: {} 
支付金额 30 

在函数调用前后都打印日志

# 在函数调用前后自动打印日志
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print("start call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        result = func(*args, **kw)
        print("end call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        return result

    return wrapper

运行结果：

start call  function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
end call  function: buy args: ('10斤苹果',) kw: {} 
start call  function: payment args: (30,) kw: {} 
支付金额 30 
end call  function: payment args: (30,) kw: {} 

自定义参数，并在函数调用前后打印日志

def log(description="", version="1.0"):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print("start %s version=%s . call function: %s args: %s kw: %s " % (
                description, version, func.__name__, args, kw))
            result = func(*args, **kw)
            print("end %s version=%s . call function: %s args: %s kw: %s " % (
                description, version, func.__name__, args, kw))
            return result

        return wrapper

    return decorator

@log(description="购买商品", version="2.2")
def buy(good_name):
    print("购买商品 %s " % good_name)


@log(description="支付金额")
def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

运行结果：

start 购买商品 version=2.2 . call function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
end 购买商品 version=2.2 . call function: buy args: ('10斤苹果',) kw: {} 
start 支付金额 version=1.0 . call function: payment args: (30,) kw: {} 
支付金额 30 
end 支付金额 version=1.0 . call function: payment args: (30,) kw: {}