实现带有权限验证的文件托管服务

最新推荐文章于 2025-11-25 14:54:57 发布

原创最新推荐文章于 2025-11-25 14:54:57 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

python web开发实战专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种在数据库中实现用户和角色多对多关系的方法，通过定义一个中间表'roles_users'来关联用户表和角色表，同时展示了如何使用Python的Flask-Security库中的Role和User模型来操作这些关系。

用户表和角色表是多对多的关系，需要定义一个用于关系的中间表，对于这种表强烈建议不要使用模型，而采用一个实际的表：

roles_users = db.Table(
    'roles_users',
    db.Column('user_id', db.Integer,
              db.ForeignKey('user.id')),
    db.Column('role_id', db.Integer, db.ForeignKey('role.id')))


class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), unique=True)
    permissions = db.Column(db.Integer, default=Permission.LOGIN)
    description = db.Column(db.String(255))


class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))

    def can(self, permissions):
        if self.roles is None:
            return False
        #获取用户所有权限
        all_perms = reduce(or_, map(lambda x: x.permissions, self.roles))
        return all_perms & permissions == permissions

    def can_admin(self):
        return self.can(Permission.ADMINISTER)