Python Flask:如何正确查询选择字段?

时间:2017-05-10 01:21:59

标签: python flask flask-sqlalchemy flask-admin

我想构建可选择的下拉按钮,根据用户选择数据库中的用户'角色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),
   )

1 个答案:

答案 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()]
    )
}

这将使字段Reviewer1Reviewer2成为必选字段,并在图片中显示: enter image description here