Question

我有一个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'))

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

管理员可以创建学校帐户，并将用户的角色设置为学校。
和具有学校角色的用户可以注册父母帐户。

我可以使用Flask-admin界面为所有用户手动设置角色，也可以使用具有 admin 角色的用户进行设置。

但是，对于具有 school 角色的用户，我没有权限设置 admin 那样的角色。

我想要的是当具有学校角色的用户注册帐户时，它会自动将该角色设置为父母。

在我的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'

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

任何帮助将不胜感激。

Answer 1

在您的代码中，当您应分配Role实例列表时，您正在尝试分配字符串“ parent”。

在您的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]

Flask-SQLAlchemy设置关系默认值

1 个答案: