一、需求
1 foo(): 2 print("执行方法")
遵循开闭原则，即关闭修改，开放扩展。显示函数执行时间。
1 import time 2 3 4 def show_time(func): 5 def wrapper(): 6 start = time.time() 7 func() 8 print("excute cost time %s" % (time.time() - start)) 9 return 10 11 return wrapper # 形成闭包函数 wrapper能够调用func 12 13 14 @show_time # foo = show_time(foo) 15 def foo(): 16 time.sleep(1) 17 print("执行方法") 18 19 20 foo()
二、被装饰函数传参
1 def detect(func): 2 def wrapper(x, y): # 为wrapper传参即为func传参 3 if type(x) == int and type(y) == int: 4 return func(x, y) # func也需接收参数 5 else: 6 return print("请使用数字类型数据") 7 8 return wrapper 9 10 11 @detect # add = detect(add) add即指向wrapper的内存地址 12 def add(x, y): 13 result = x + y 14 return result 15 16 17 add("s", 8)
三、装饰器传参
1 def show(flag): 2 if not flag == "True": 3 return print("该方法不可用") 4 5 def detect(func): 6 def wrapper(x, y, *args, **kwargs): 7 if type(x) == int and type(y) == int: 8 return func(x, y) 9 else: 10 return print("请使用数字类型数据") 11 return wrapper 12 return detect 13 14 15 @show("True") # @show("True")=@detect 16 def add(x, y): 17 result = x + y 18 return result 19 20 21 add("s", 8)