Flask中的一对一关系SQLAlchemy仅单向工作



我在一个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}'

最新更新