浅析装饰器的那些事儿

约定不等于承诺〃 2022-11-13 05:27 73阅读 0赞

## 一、装饰器的简单定义 ##

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

## 二、装饰器的作用 ##

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

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

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

想学习更多Python网络爬虫与数据挖掘知识，可前往专业网站：[http://pdcfighting.com/][http_pdcfighting.com] 想学习更多Python网络爬虫与数据挖掘知识，可前往专业网站：[http://pdcfighting.com/][http_pdcfighting.com]

[http_pdcfighting.com]: http://pdcfighting.com/