一、装饰器的简单定义
外层函数返回里层函数的引用,里层函数引用外层函数的变量。
二、装饰器的作用
通俗来讲装饰器的作用就是在不改变已有函数代码前提下,为该函数增加新的功能。
1 2def run(): 3 print('我会跑') 4fun()
现在我想在原有函数的基础上新增一个功能:我会唱歌。这个时候利用装饰器则轻松可以帮我们实现这个功能。
三、实例理解
(1)不传参的装饰器
1 2def outer(fun): 3 def inner(): 4 fun() //fun是外层函数的变量,在inner里面用 5 return inner //inner就是里层函数的引用
(2)传递参数的装饰器:
1 2def func(fun): 3 def add(*args,**kwarge): 4 return fun(*args,**kwargs) 5 return add
现在对于装饰器的基本格式有一定的了解,就可以直接写函数了。下面实现文章开头的 我会唱歌 的功能
1 2def outer(fun): 3 def inner(*args, **kwarge): 4 print("我会唱歌") 5 return fun(*args, **kwarge) 6 return inner
四、如何使用装饰器
1方法一:使用@符号+装饰器的名字 把它放在想要装饰函数的上一行即可 2@outer 3def run(): 4 print('我会跑') 5 6run() 7 8 9方法二: 10def run(): 11 print('我会跑') 12 13run=outer(run) #就等价于@outer 14run() 15 16最终打印结果是: 17我会唱歌 18我会跑
如果我想知道fun 传递的参数是什么,在装饰器内部可以使用如下方式:
1def outer(fun): 2 a = 1 3 def inner(*args, **kwarge): # args是一个数组,kwargs一个字典 4 print(fun.__name__) #打印fun接收的函数的名字 5 print("我会唱歌") 6 return fun(*args, **kwarge) 7 return inner
但是如果我们 print(run.__name__,6666666) 输出的结果是inner,并不是我们想要的run,这里的函数被warpTheFunction替代了。它重写了我们函数的名字和注释文档(docstring)。解决方法如下:
1from functools import wraps 2 3def outer(fun): 4 @wraps(fun) 5 def inner(*args, **kwargs): 6 print(fun.__name__,11111111111) 7 print("我会唱歌") 8 return fun(*args, **kwargs) 9 return inner 10 11@outer 12def run(): 13 print('我会跑') 14 15 print(run.__name__,6666666) //输出结果为 run 666666
五、自己实现装饰器
1 2def subuser_keymanage(view_func): 3 '''功能是实现用户管理权限的判定''' 4 def _wrapper_view(request, *args, **kwargs): 5 user = request.user #一个Customer对象,包含了用户名/密码等信息 6 customer = user.customer.customer_id #用户的id 7 select_status = get_curuser_permission(user=user, customer=customer)#调用函数返回的值有两种0和1 8 if not select_status:#如果返回0表示没有权限,返回错误码 9 return render_response(request, ErrorCode.FAILED) 10 return view_func(request, *args, **kwargs) 11 return _wrapper_view
1 2@subuser_keymanage 3def generate_subuser_ak_sk(request): 4 params = json.loads(request.body) #获取卡前端传递的参数 5 user_id_only = params.get("user_id") #获取用户表示id值 6 中间代码就忽略了...... 7 return render_response(request, ErrorCode.FAILED)
六、装饰器小结
通过装饰器很大程度上可以减少代码的复用,在代码规范中这一点是很重要的。
以上就是装饰器的基本知识,即便没有任何基础,按照作者的思路,套用固定的格式,不需要完全理解,只要按照流程一步一步就能写出高端大气上档次的装饰器了,恭喜你!
前方高能请注意:装饰器传参,三层嵌套函数一般用的比较少,其实也不难,一层一层看,跟上文讲的一样,仅作为知识的拓宽。
1 2import logging 3def use_logging(level): 4 def decorator(func): 5 def wrapper(*args, **kwargs): 6 if level == "warn": 7 logging.warn("%s is running" % func.__name__) 8 elif level == "info": 9 logging.info("%s is running" % func.__name__) 10 return func(*args) 11 return wrapper 12 13 return decorator 14 15@use_logging(level="warn") 16def foo(name='foo'): 17 print("i am %s" % name) 18 19foo()
1 2i am foo 3WARNING:root:foo is running
如果在操作过程中有任何问题,记得下面留言,我们看到会第一时间解决问题。
看完本文有收获?请转发分享给更多的人
IT共享之家
入群请在微信后台回复【入群】

-------------------********************************** End **********-------------**-----********-**********************************
往期精彩文章推荐:
欢迎各位大佬点击链接加入群聊【helloworld开发者社区】:https://jq.qq.com/?_wv=1027&k=mBlk6nzX进群交流IT技术热点。
本文转自 https://mp.weixin.qq.com/s/-9ZS1B0IHW-tDVggm_0oqQ,如有侵权,请联系删除。
