定义装饰器（使用@wraps与不使用的区别）

最新推荐文章于 2024-12-12 15:40:22 发布

Chens_2333

最新推荐文章于 2024-12-12 15:40:22 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

文章标签： python 装饰器

本文链接：https://blog.youkuaiyun.com/qq_42716602/article/details/81903737

本文介绍如何使用Python装饰器实现用户验证及权限控制。通过@wraps保持原函数属性不变，实现登录状态检查和API接口权限验证。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

装饰器的作用: 在不改变原有功能代码的基础上,添加额外的功能,如用户验证等
@wraps(view_func)的作用:
不改变使用装饰器原有函数的结构(如name, doc)

不使用wraps可能出现的ERROR: view_func…endpoint…map..

一、用户验证装饰器，使用@wraps

 # 定义登录判断装饰器
def admin_login(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        # session不存在时请求登录
        if "admin" not in session:
            return redirect(url_for("admin.login", next=request.url))
        return f(*args, **kwargs)
    return decorated_function

二、用户权限设置装饰器，api接口

# 权限 1查看/2可下载/4可添加   [限制接口的权限]
def check_permission_control(permission):
    def check_permission(func):
        def check(*args, **kwargs):
            parse = parser.parse_args()
            token = parse.get('token')
            returndata = {}
            if token:  # 有token
                users = User.query.filter(User.token == token)
                if users.count() > 0:  # 存在用户
                    user = users.first()
                    # 有无对应操作的权限
                    # 用户权限user: 4    100
                    # 接口限制权限permission: 2    010
                    # 接口限制权限permission: 4    100
                    if user.permissions & permission == permission: # 该用户有权限使用此接口
                        return func(*args, **kwargs)
                    else:
                        abort(403, message='你没有该操作权限,请联系管理员!')
                else:  # 不存在
                    abort(401, message='你还没有登录,请登录后操作')
            else:  # 无token
                abort(401, message='你还没有登录,请登录后操作')
        return check
    return check_permission