外层函数返回里层函数的引用,里层函数引用外层函数的变量。
二、装饰器的作用
通俗来讲装饰器的作用就是在不改变已有函数代码前提下,为该函数增加新的功能。
defrun():print('我会跑')fun()
现在我想在原有函数的基础上新增一个功能:我会唱歌。这个时候利用装饰器则轻松可以帮我们实现这个功能。
三、实例理解
(1)不传参的装饰器
defouter(fun):definner():fun()//fun是外层函数的变量,在inner里面用returninner//inner就是里层函数的引用
(2)传递参数的装饰器:
deffunc(fun):defadd(*args,**kwarge):returnfun(*args,**kwargs)returnadd
defouter(fun):definner(*args,**kwarge):print("我会唱歌")returnfun(*args,**kwarge)returninner
四、如何使用装饰器
方法一:使用@符号+装饰器的名字把它放在想要装饰函数的上一行即可@outerdefrun():print('我会跑')run()
方法二:defrun():print('我会跑')
run=outer(run)#就等价于@outerrun()
最终打印结果是:我会唱歌我会跑
如果我想知道fun传递的参数是什么,在装饰器内部可以使用如下方式:
defouter(fun):a=1definner(*args,**kwarge):#args是一个数组,kwargs一个字典print(fun.__name__)#打印fun接收的函数的名字print("我会唱歌")returnfun(*args,**kwarge)returninner
但是如果我们print(run.__name__,6666666)输出的结果是inner,并不是我们想要的run,这里的函数被warpTheFunction替代了。它重写了我们函数的名字和注释文档(docstring)。解决方法如下:
fromfunctoolsimportwrapsdefouter(fun):@wraps(fun)definner(*args,**kwargs):print(fun.__name__,11111111111)print("我会唱歌")returnfun(*args,**kwargs)returninner
@outerdefrun():print('我会跑')print(run.__name__,6666666)//输出结果为run666666
五、自己实现装饰器
defsubuser_keymanage(view_func):'''功能是实现用户管理权限的判定'''def_wrapper_view(request,*args,**kwargs):user=request.user#一个Customer对象,包含了用户名/密码等信息customer=user.customer.customer_id#用户的idselect_status=get_curuser_permission(user=user,customer=customer)#调用函数返回的值有两种0和1ifnotselect_status:#如果返回0表示没有权限,返回错误码returnrender_response(request,ErrorCode.FAILED)returnview_func(request,*args,**kwargs)return_wrapper_view@subuser_keymanagedefgenerate_subuser_ak_sk(request):params=json.loads(request.body)#获取卡前端传递的参数user_id_only=params.get("user_id")#获取用户表示id值中间代码就忽略了......returnrender_response(request,ErrorCode.FAILED)
六、装饰器小结
通过装饰器很大程度上可以减少代码的复用,在代码规范中这一点是很重要的。
以上就是装饰器的基本知识,即便没有任何基础,按照作者的思路,套用固定的格式,不需要完全理解,只要按照流程一步一步就能写出高端大气上档次的装饰器了,恭喜你!
前方高能请注意:装饰器传参,三层嵌套函数一般用的比较少,其实也不难,一层一层看,跟上文讲的一样,仅作为知识的拓宽。
importloggingdefuse_logging(level):defdecorator(func):defwrapper(*args,**kwargs):iflevel=="warn":logging.warn("%sisrunning"%func.__name__)eliflevel=="info":logging.info("%sisrunning"%func.__name__)returnfunc(*args)returnwrapperreturndecorator@use_logging(level="warn")deffoo(name='foo'):print("iam%s"%name)foo()iamfooWARNING:root:fooisrunning