浅析装饰器的那些事儿

浏览: 86

一、装饰器的简单定义

外层函数返回里层函数的引用,里层函数引用外层函数的变量。

二、装饰器的作用

通俗来讲装饰器的作用就是在不改变已有函数代码前提下,为该函数增加新的功能。

def run():
print('我会跑')
fun()

现在我想在原有函数的基础上新增一个功能:我会唱歌。这个时候利用装饰器则轻松可以帮我们实现这个功能。

三、实例理解

(1)不传参的装饰器

def outer(fun):
def inner():
fun() //fun是外层函数的变量,在inner里面用
return inner //inner就是里层函数的引用

(2)传递参数的装饰器:

def func(fun):
def add(*args,**kwarge):
return fun(*args,**kwargs)
return add

现在对于装饰器的基本格式有一定的了解,就可以直接写函数了。下面实现文章开头的 我会唱歌 的功能

def outer(fun):
def inner(*args, **kwarge):
print("我会唱歌")
return fun(*args, **kwarge)
return inner

四、如何使用装饰器

方法一:使用@符号+装饰器的名字   把它放在想要装饰函数的上一行即可
@outer
def run():
print('我会跑')

run()


方法二:
def run():
print('我会跑')

run=outer(run) #就等价于@outer
run()

最终打印结果是:
我会唱歌
我会跑

如果我想知道fun 传递的参数是什么,在装饰器内部可以使用如下方式:

def outer(fun):
a = 1
def inner(*args, **kwarge): # args是一个数组,kwargs一个字典
print(fun.__name__) #打印fun接收的函数的名字
print("我会唱歌")
return fun(*args, **kwarge)
return inner

但是如果我们 print(run.name,6666666) 输出的结果是inner,并不是我们想要的run,这里的函数被warpTheFunction替代了。它重写了我们函数的名字和注释文档(docstring)。解决方法如下:

from functools import wraps

def outer(fun):
@wraps(fun)
def inner(*args, **kwargs):
print(fun.__name__,11111111111)
print("我会唱歌")
return fun(*args, **kwargs)
return inner

@outer
def run():
print('我会跑')

print(run.__name__,6666666) //输出结果为 run 666666

五、自己实现装饰器

def subuser_keymanage(view_func):
'''功能是实现用户管理权限的判定'''
def _wrapper_view(request, *args, **kwargs):
user = request.user #一个Customer对象,包含了用户名/密码等信息
customer = user.customer.customer_id #用户的id
select_status = get_curuser_permission(user=user, customer=customer)#调用函数返回的值有两种0和1
if not select_status:#如果返回0表示没有权限,返回错误码
return render_response(request, ErrorCode.FAILED)
return view_func(request, *args, **kwargs)
return _wrapper_view
@subuser_keymanage
def generate_subuser_ak_sk(request):
params = json.loads(request.body) #获取卡前端传递的参数
user_id_only = params.get("user_id") #获取用户表示id值
中间代码就忽略了......
return render_response(request, ErrorCode.FAILED)

六、装饰器小结

通过装饰器很大程度上可以减少代码的复用,在代码规范中这一点是很重要的。

以上就是装饰器的基本知识,即便没有任何基础,按照作者的思路,套用固定的格式,不需要完全理解,只要按照流程一步一步就能写出高端大气上档次的装饰器了,恭喜你!

前方高能请注意:装饰器传参,三层嵌套函数一般用的比较少,其实也不难,一层一层看,跟上文讲的一样,仅作为知识的拓宽。

import logging
def use_logging(level):
def decorator(func):
def wrapper(*args, **kwargs):
if level == "warn":
logging.warn("%s is running" % func.__name__)
elif level == "info":
logging.info("%s is running" % func.__name__)
return func(*args)
return wrapper

return decorator

@use_logging(level="warn")
def foo(name='foo'):
print("i am %s" % name)

foo()
i am foo
WARNING:root:foo is running

如果在操作过程中有任何问题,记得下面留言,我们看到会第一时间解决问题。

推荐 0
本文由 dcpeng 创作,采用 知识共享署名-相同方式共享 3.0 中国大陆许可协议 进行许可。
转载、引用前需联系作者,并署名作者且注明文章出处。
本站文章版权归原作者及原出处所有 。内容为作者个人观点, 并不代表本站赞同其观点和对其真实性负责。本站是一个个人学习交流的平台,并不用于任何商业目的,如果有任何问题,请及时联系我们,我们将根据著作权人的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。

0 个评论

要回复文章请先登录注册