参考文章
装饰器
用过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: {}