我在一个flask应用程序中有两个模型,它们形成一对一的关系,其中"Staff"模型可以与一个"User"关联。可以有"职员"而不是"用户",但不能有"用户"而没有"职员"。
不幸的是,我在第一个栏上摔倒了。我正在尝试将这些模型组合到一个HTML表中,该表是通过Jinja2模板化的。该表应列出所有"员工",并确定他们是否为"用户"。如果他们是"用户",则应该识别他们是否处于活动状态。
以下是我的模型:
class Staff(db.Model):
__tablename__ = 'staff'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), nullable=False)
job_title = db.Column(db.String(255), nullable=True)
is_user = db.Column(db.Boolean, nullable=False, default=False)
user = db.relationship('User',
backref=db.backref('staff', lazy=True))
def __repr__(self):
return f'Staff Member: {self.id}, {self.name}, {self.job_title}'
class User(db.Model, UserMixin):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(120), unique=True, nullable=False)
image_file = db.Column(db.String(20), nullable=False,
default='default.jpg')
password = db.Column(db.String(255), nullable=False)
active = db.Column(db.Boolean, nullable=False, default=False)
staff_id = db.Column(db.Integer, db.ForeignKey('staff.id'),
nullable=False)
def __repr__(self):
return f'User({self.name}, {self.email}, {self.image_file})'
这里有一个例子,说明金贾的关系。此代码段按预期工作:
{% for user in users %}
<tr>
<td>{{ user.staff.name }}</td>
<td>{{ user.staff.job_title }}</td>
<td>{{ user.email }}</td>
{% if user.active == True %}
<td>Yes</td>
{% else %}
<td>No</td>
{% endif %}
{% endfor %}
然而,在这个代码中,我试图在循环所有"工作人员"的同时获得"用户"的详细信息,但它根本不起作用:
{% for staff in all_staff %}
<tr>
<td>{{ staff.name }}</td>
<td>{{ staff.job_title }}</td>
{% if staff.is_user == True %}
{% if staff.user.active == True %}
<td class="text-success">Yes (Active)</td>
{% elif staff.user.active == False %}
<td class="text-warning">Yes (Inactive)</td>
{% endif %}
{% elif staff.is_user == False %}
<td class="text-danger">No</td>
{% endif %}
{% endfor %}
虽然这种关系以一种方式起作用,但在第二表的例子中却不起作用;它根本不返回HTML代码,除非staff.is_user==False语句为true。
我试图在"User"类中创建一个新的db.relationship。我还同时在两个班上尝试了数据库关系。我也在同一行上尝试过Jinja2if语句:
{% if staff.is_user == True and staff.user.active == True %}
然而,这些都不起作用,而且在我希望根据if语句的结果返回正确的html的地方,当通过"Staff"表引用"Users"表时,我根本没有得到任何结果。
我肯定我错过了一些简单的东西,但我想不通。
有人能看出我哪里错了吗?
事实证明,答案隐藏在Flask SQLAlchemy文档的"一对五关系"部分。为了引用一对一关系,您需要在关系中设置uselist=False
。正如它听起来的那样,它将连接加载为标量而不是列表。
我还犯了一个错误,从主要的SQLAlchemy文档中获取了一些信息,并且没有正确地在关系中加载backref。在Flask SQLAlchemy中,它只需要声明为一个字符串。在这种情况下,backref='staff'
。
此代码的工作方式与预期完全一致:
class Staff(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), unique=True, nullable=False)
job_title = db.Column(db.String(80))
is_user = db.Column(db.Boolean, default=False)
user = db.relationship('User', backref='staff', lazy=True, uselist=False)
def __repr__(self):
return f'Staff Member: {self.id}, {self.name}, {self.job_title}'
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(120), unique=True, nullable=False)
active = db.Column(db.Boolean, nullable=False, default=False)
staff_id = db.Column(db.Integer, db.ForeignKey('staff.id'), nullable=False)
def __repr__(self):
return f'User Account: {self.id}, {self.email}'