python - 如何将烧瓶 RBAC 模型添加到现有的 SQLAlchemy 数据库
问题描述
我正在尝试将RBAC添加到我现有的烧瓶应用程序中,其中我已经有 2 个分别描述 User 和 Post 模型的模型。这是我的代码:
# models.py
from datetime import datetime
from rpd_site import db, login_manager
from flask_login import UserMixin
# []
@login_manager.user_loader
def load_user(user_id):
return User.query.get(int(user_id))
# Main site account table
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
image_file = db.Column(db.String(20), nullable=False, default='default.jpg')
password = db.Column(db.String(60), nullable=False)
confirmed = db.Column(db.Boolean, nullable=False, default=0)
posts = db.relationship('Post', backref='author', lazy=True)
def __repr__(self):
return f"User('{self.username}', '{self.email}', '{self.confirmed}')"
# Posts table
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
date_posted = db.Column(db.DateTime, nullable=False, default=datetime.now) # current local time instead of .utcnow
content = db.Column(db.Text, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
image_file = db.Column(db.String(20), nullable=False, default='default_post.png')
def __repr__(self):
return f"Post('{self.title}', '{self.date_posted}', '{self.content[:15]}')"
当我尝试从这里添加所有缺少的代码时,我遇到了很多错误。特别是我不确定是否应该从flask_rbac 或flask_login 导入UserMixin。
帮助我了解如何使用 RBAC 功能升级我的数据库。
解决方案
这是一个非常广泛的问题,我会尽量给你一个最低限度的代码,这样你就可以实现 RBAC。下面的示例使用 Flask-security。
from app import db
from flask_security import RoleMixin, UserMixin
# may to many association table between User and Role
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):
__tablename__ = 'role'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), unique=True)
def __str__(self):
return self.name
class User(db.Model, UserMixin):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(50), unique=True)
password = db.Column(db.String(255))
roles = db.relationship('Role', secondary=roles_users,
backref=db.backref('users', lazy='joined'))
def __str__(self):
return self.email
您可以根据上述模型迁移或创建数据库。以上内容足以让您在 RDBC 或视图级别执行后端操作。然后,您可以使用以下链接轻松地为每个用户分配角色。
Flask-security 创建角色、用户并将 user_id 链接到 role_id
如果您想在视图中执行 RBAC,请执行以下操作。
from flask_security import login_required, roles_accepted
@app.route('/a_restricted_view/')
@login_required
@roles_accepted('role_one', 'role_two')
def a_restricted_view():
return "I am only visible to users with role_one and role_two"
推荐阅读
- r - 使用 label_bquote 为构面的每一行分配不同的标签
- apache-kafka - Kafka Streams RocksDB 抛出 IOError(Stalefile) 异常
- google-cloud-platform - 使用 Python 从经过训练的 Vertex AI 表格回归模型中访问特征重要性
- java - 如何将 Eclipse 从 Java 切换到 C/C++?
- laravel - 受到圣殿机械师的干扰
- xcode - XCode 上的错误:致命:无法读取“https://github.com”的用户名:未配置设备
- ios - 如何通过 WhatsApp 分享图片+文字?
- angular - pTableCheckbox 更改分页/排序 (PrimeNG) 的值
- python-3.x - 为什么在 Python 中使用 @classmethods 而不是没有实例的普通方法?
- c# - oauth 状态丢失或无效,但有时仅在 azure 中发生