Flask-SQLAlchemy 中的一对一关系仅以一种方式工作

Question

我在一个烧瓶应用程序中有两个模型，它们形成一对一的关系，其中“员工”模型可以与一个“用户”相关联。可以有“员工”而不是“用户”，但不能有“用户”而不是“员工”。

不幸的是，我在第一个障碍时跌倒了。我正在尝试将模型组合在通过 Jinja2 模板化的 HTML 表中。该表应该列出所有“员工”并确定他们是否是“用户”。如果他们是“用户”，则应该识别他们是否处于活动状态。

以下是我的模型：

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})'

这里是 Jinja2 中关系的一个例子。此代码段按预期工作：

{% 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 %}

虽然关系以一种方式起作用，但在第二个表示例中不起作用；除非 staff.is_user == False 语句为真，否则它不会返回任何 HTML 代码。

我尝试在“用户”类中创建一个新的 db.relationship。我还同时在两个类中尝试了 db.relationship。我也在同一行尝试了 Jinja2 if 语句：

{% if staff.is_user == True and staff.user.active == True %}

但是，这些都不起作用，并且我希望根据 if 语句的结果返回正确的 html，但在通过'职员表。

我确定我错过了一些简单的东西，但我想不通。

谁能看出我哪里出错了？

Answer 1

原来答案隐藏在 Flask-SQLAlchemy docs 的“一对一关系”部分。为了引用一对一的关系，您需要在关系中设置uselist=False。就像听起来一样，它将连接加载为标量而不是列表。

我在从主要的 SQLAlchemy 文档中获取一些信息并且没有正确加载关系中的反向引用时也犯了一个错误。在 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}'