第九章用户角色（二）

最新推荐文章于 2024-04-16 09:53:05 发布

山药鱼儿

最新推荐文章于 2024-04-16 09:53:05 发布

阅读量372

点赞数

分类专栏： Flask 进阶文章标签： Flask 赋予角色 Flask角色检验

本文链接：https://blog.csdn.net/Geroge_lmx/article/details/100532099

版权

本文详细介绍了在Flask应用中如何实现用户角色的赋予和检验，包括在注册时根据邮箱设置用户角色，定义User模型以检查用户权限，实现自定义的匿名用户类以方便权限检查。此外，还展示了如何创建自定义装饰器来限制视图函数的访问权限，以及处理403错误的情况。最后提到了在模板中进行权限检查的方法。

摘要由CSDN通过智能技术生成

一. 赋予角色

用户在应用中注册账户时，应该赋予其适当的角色。多数用户在注册时赋予的角色是“普通用户”，因为这是默认角色。唯一例外的是管理员，在注册应用时就应该被赋予管理员。当前应用中，管理员由保存在环境变量APP_ADMIN中的电子邮件地址识别，只要改电子邮件地址出现在注册请求中，就会被赋予admin。

app/models.py：定义默认的用户角色

class User(UserMixin, db.Model):
    __tablename__ = "users"
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(64), unique=True, index=True)
    username = db.Column(db.String(64), unique=True, index=True)
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
    password_hash = db.Column(db.String(128))
    confirmed = db.Column(db.Boolean, default=False)

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        if self.role is None:
            if self.email == current_app.config['APP_ADMIN']:
                self.role = Role.query.filter_by(name='Administrator').first()
            if self.role is None:
                self.role = Role.query.filter_by(default=True).first()

User类的构造函数首先调用基类的构造函数。如果创建基类对象后还没有定义角色，则根据电子邮件将其设为管理员或普通用户。