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

Question

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

不幸的是，我在第一个障碍上就倒下了。我正在尝试将模型组合到一个 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})'

这里是 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 %}

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

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

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

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

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

谁能看出我哪里出错了？

Answer 1

原来答案隐藏在 Flask-SQLAlchemy docs 的“One-to-May Relationships”部分中.为了引用一对一关系，您需要在关系中设置 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}'