我想构建可选择的下拉按钮,根据用户选择数据库中的用户'角色ID。
这些课程定义如下。
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)
def __str__(self):
return self.name
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
roles = db.relationship('Role', secondary=roles_users,
backref=db.backref('users', lazy='dynamic'))
email = db.Column(db.String(255), unique=True)
def __str__(self):
return self.email
class Project(db.Model):
id = db.Column(db.Integer, primary_key=True)
reviewer = db.Column(db.Unicode(128))
def __unicode__(self):
return self.name
由于我想在项目类中列出Role.id==4
作为审阅者的用户,我尝试了以下两种方法,但徒劳无功:
1。 QuerySelectField
它不会在列表中返回任何内容:
def reviewer_choices():
return User.query.join(User.roles).filter(Role.id == 4).all().query
class Project(db.Model):
reviewer = QuerySelectField(query_factory = reviewer_choices)
2。 form_args
它返回错误:TypeError: __init__() got an unexpected keyword argument 'query_factory'
def reviewer_choices():
return User.query.join(User.roles).filter(Role.id == 4).all().query
class Project(db.Model):
form_args = dict(
reviewer=dict(query_factory =reviewer_choices),
)
答案 0 :(得分:1)
query_factory
回调应该返回一个查询,您可以像这样更改reviewer_choices
:
def reviewer_choices():
return User.query.join(User.roles).filter(Role.id == 4)
<强>更新强>
将以下代码添加到class SWProjectView(sqla.ModelView)
:
form_extra_fields = {
'reviewer1': sqla.fields.QuerySelectField(
label='Reviewer1',
query_factory=reviewer1_choices,
validators=[DataRequired()]
),
'reviewer2': sqla.fields.QuerySelectField(
label='Reviewer2',
query_factory=reviewer2_choices,
validators=[DataRequired()]
)
}