首页 > 解决方案 > Flask-SQLAlchemy 设置关系默认值

问题描述

我有一个由 Flask 制作的应用程序,我使用 Flask-admin 和 Flask-SQLAlchemy 进行数据库管理。在我的应用程序中,有三个角色,即:

  1. 管理员
  2. 学校
  3. 父母

这是我桌子上的代码片段:

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'))

代码已经可以正常工作了。

我可以使用 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'

所以,我想要的是在定义角色用户的关系列上设置一个默认值。

任何帮助都感激不尽。

标签: pythonflaskflask-sqlalchemyflask-loginflask-admin

解决方案


在您的代码中,当您应该分配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]

推荐阅读