python - Flask-SQLAlchemy 设置关系默认值
问题描述
我有一个由 Flask 制作的应用程序,我使用 Flask-admin 和 Flask-SQLAlchemy 进行数据库管理。在我的应用程序中,有三个角色,即:
- 管理员,
- 学校
- 父母
这是我桌子上的代码片段:
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(80), unique=True)
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(255))
roles = db.relationship('Role', secondary=roles_users,
backref=db.backref('users', lazy='dynamic'))
class School(User):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
class Parent(User):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
school_id = db.Column(db.Integer, db.ForeignKey('school.id'))
代码已经可以正常工作了。
- admin可以创建学校帐户并将用户的角色设置为school。
- 具有学校角色的用户可以创建父母帐户。
我可以使用 Flask-admin 界面手动为所有用户设置角色,我可以使用具有管理员角色的用户来设置角色。
但是,对于具有学校角色的用户,我不会像管理员那样授予设置角色的权限。
我想要的是当具有学校角色的用户创建帐户时,它会自动将角色设置为parent。
在我的 ModelView 中,我尝试使用以下代码插入它:
class ParentModelView(sqla.ModelView):
def on_model_change(self, form, model, is_created):
if is_created:
model.school_id = current_user.id
model.roles = 'parent'
然后我得到了这个错误:
TypeError: Incompatible collection type: str is not list-like
然后我尝试用列表包装它:
def on_model_change(self, form, model, is_created):
if is_created:
model.school_id = current_user.id
model.roles = ['parent']
那么错误是这样的:
AttributeError: 'str' object has no attribute '_sa_instance_state'
当我尝试在我的关系列上设置默认值时:
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(255))
roles = db.relationship('Role', secondary=roles_users,
backref=db.backref('users', lazy='dynamic'),
default='parent') # or default=3 (which is '3' is the parent id in roles_user association table.
然后我得到了这个错误:
TypeError: relationship() got an unexpected keyword argument 'default'
所以,我想要的是在定义角色用户的关系列上设置一个默认值。
任何帮助都感激不尽。
解决方案
在您的代码中,当您应该分配Role
实例列表时,您试图分配字符串“父级”。
在您的on_model_change
方法中,您需要从数据库中获取名称为“Parent”的角色,然后将其作为列表分配给角色关系。
示例(未经测试)
def on_model_change(self, form, model, is_created):
if is_created:
# find the role with name 'Parent'
_parent_role = Role.query.filter(Role.name == 'Parent').first()
if _parent_role:
model.school_id = current_user.id
model.roles = [_parent_role]
推荐阅读
- flutter - 防止谷歌地图在 SingleChildScrollView 内滚动
- vega-lite - 使用 Vega-Lite 计算同一字段的负值和正值
- javascript - chess.com API 不会带回一些请求
- c# - 防止 XmlDocument 解释 HTML 代码
- php - using arabic words in pdf
- unity3d - C# Unity 我怎么能每秒都做点什么?
- java - 使用 itextpdf 时在 pdf 文件中定位坐标文本
- java - 基于 Java 的 azure 函数的 Swagger 生成
- flutter - 在 Flutter 中,如何删除小部件的一部分。我想删除容器小部件的下半部分
- css - 如果 scss 中的输入为空,如何禁用或忽略代码