python传参装饰器的拓展妙用

开门见山，不扯那么远了，直接上代码：

def setTag(tag):  # 由于此装饰器需要参数，所以要再套一层
    def myDecorator(func):  # 装饰器的核心，接受函数对象做参数，返回包装后的函数对象
        def myWrapper(*arg, **kvargs):  # 包装的具体过程
            sign = "<" + tag + ">"
            return sign + func(*arg, **kvargs) + sign

        return myWrapper

    return myDecorator


@setTag("div")  # 用@标签在定义函数时套上装饰器
def hello(name):
    return 'hello' + name


print(hello('world'))

我们都知道python装饰器是对现有已经写好的函数的一些功能拓展，（如果对于python中这个独有的功能还没有彻彻底底的了解的话，请稍后翻看我的另一篇文章《Python装饰器小谈》）。
以上代码是对既有函数的传参装饰，其实就是通过添加一层闭包来实现传入函数的同时，用外部传入的参数影响装饰的结果。
对于此来说，之前用django比较多，权限的认证和token的判断都有封装好的函数，虽然有些功能看起来有些笨重，但也能快速完成需求。最近重返flask，没有太多现成的东西以供使用，就只能自己思考开发。之前的白名单、token验签的方式更多是在主函数中调用钩子函数整体判断，亦或者加入白名单解除验签操作。但是对于这点来说未免也没有好好利用python这个独特的工具。单独装饰亦或者整体装饰，无论是token还是权限都可以轻松实现呀！
装饰器无非是在函数开头或者是函数的结尾进行新的功能的添加或新的条件的判断。对于一个api来说，除非我们将其拆分封装成几个不同各司其职的功能函数来看的话，装饰器传参好像无法正常进行下去。我想要获取request对象中的一些数据，但作为参数传参，不转个弯来看，就只能url传参获取了。
但是！为何不整个传入request对象来看呢，参数不一定是一个参数，也可以是一个对象啊！于是疯狂改写，传入参数的装饰器也可以通用api了!

token验签装饰器：

def token_v(r):
    """
    token验证装饰器
    :param user_type: 传入用户类型
    :param r: request对象
    :return:
    """
    def decorator(func):
        def wrapper():
            if r.form.get(user_id_name):
                user_id = r.form.get('user_id')
            else:
                user_id = r.args.get('user_id')
            token = get_uid_token(user_type=user_type, user_id=user_id)  # 连接redis获取token函数
            if not token:
                return visitFail(code=-300)  # 返回值函数
            else:
                return func()
        return wrapper
    return decorator

装饰函数：

@app.route('/task/apply', methods=['post'])
@func.token_v('f', request)
def task_apply():
    """
        :return:
    """
    return visitSuccess()  # 返回值函数

对此来说，其实类装饰器对比来说也大同小异，研究研究其实用起来还挺顺手的！