浅析装饰器的那些事儿

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

[[379814]]

一、装饰器的简单定义

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

二、装饰器的作用

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

  1. def run(): 
  2.    print('我会跑'
  3. fun() 

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

三、实例理解

(1)不传参的装饰器

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

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

  1. def func(fun): 
  2.    def add(*args,**kwarge): 
  3.       return fun(*args,**kwargs) 
  4.    return add 

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

  1. def outer(fun): 
  2.     def inner(*args, **kwarge): 
  3.         print("我会唱歌"
  4.         return fun(*args, **kwarge) 
  5.     return inner 

四、如何使用装饰器

  1. 方法一:使用@符号+装饰器的名字   把它放在想要装饰函数的上一行即可 
  2. @outer 
  3. def run(): 
  4.    print('我会跑'
  5.     
  6. run() 
  7.  
  8.  
  9. 方法二: 
  10. def run(): 
  11.     print('我会跑'
  12.  
  13. run=outer(run)   #就等价于@outer 
  14. run() 
  15.  
  16. 最终打印结果是: 
  17. 我会唱歌 
  18. 我会跑 

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

  1. def 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)。解决方法如下:

  1. from functools import wraps 
  2.  
  3. def 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 
  12. def run(): 
  13.    print('我会跑'
  14.     
  15.  print(run.__name__,6666666)  //输出结果为 run 666666 

五、自己实现装饰器

  1. def subuser_keymanage(view_func): 
  2.     '''功能是实现用户管理权限的判定''' 
  3.     def _wrapper_view(request, *args, **kwargs): 
  4.         user = request.user #一个Customer对象,包含了用户名/密码等信息 
  5.         customer = user.customer.customer_id #用户的id 
  6.         select_status = get_curuser_permission(user=user, customer=customer)#调用函数返回的值有两种0和1 
  7.         if not select_status:#如果返回0表示没有权限,返回错误码 
  8.             return render_response(request, ErrorCode.FAILED) 
  9.         return view_func(request, *args, **kwargs) 
  10.     return _wrapper_view 

  1. @subuser_keymanage  
  2. def generate_subuser_ak_sk(request): 
  3.     params = json.loads(request.body) #获取卡前端传递的参数 
  4.     user_id_only = params.get("user_id") #获取用户表示id值 
  5.     中间代码就忽略了...... 
  6.     return render_response(request, ErrorCode.FAILED) 

六、装饰器小结

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

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

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

  1. import logging 
  2. def use_logging(level): 
  3.     def decorator(func): 
  4.         def wrapper(*args, **kwargs): 
  5.             if level == "warn"
  6.                 logging.warn("%s is running" % func.__name__) 
  7.             elif level == "info"
  8.                 logging.info("%s is running" % func.__name__) 
  9.             return func(*args) 
  10.         return wrapper 
  11.  
  12.     return decorator 
  13.  
  14. @use_logging(level="warn"
  15. def foo(name='foo'): 
  16.     print("i am %s" % name
  17.  
  18. foo() 

  1. i am foo 
  2. WARNING:root:foo is running 

 

责任编辑:姜华 来源: IT共享之家
相关推荐

2018-02-02 13:58:59

数据存储

2012-12-06 14:34:16

2021-04-11 08:21:20

Python@property装饰器

2022-06-28 22:17:52

浏览器底层css

2023-04-11 07:34:40

分布式系统算法

2021-03-18 09:01:53

软件开发软件选型

2021-06-02 08:33:31

TPCTPC-H系统

2011-02-25 14:35:00

2022-02-08 17:39:04

MySQL服务器存储

2018-09-26 06:50:19

2021-06-09 13:28:40

密码安全身份认证数据安全

2013-12-26 14:23:03

定位系统GPS监测

2010-04-07 13:13:19

Visual Stud

2012-07-19 15:30:00

Linux

2013-07-09 13:50:05

2021-05-10 08:58:09

Harbor架构Registry 服务

2017-08-09 08:25:35

DBA数据库OLAP

2022-08-04 10:18:32

栈迁移​寄存器内存

2022-11-04 07:57:59

编程编码编译器

2021-10-13 08:53:53

Zookeeper存储系统
点赞
收藏

51CTO技术栈公众号