Flask 使用 SqlAlchemy 实现同表的两列关联

在本文中，我们将介绍如何使用 Flask 和 SqlAlchemy 实现同一表中的两列的关联。考虑一个场景，我们有一个 User 表，其中有两列：follower_id 和 followed_id，分别表示用户和其关注的用户。我们希望能够方便地查询一个用户关注的所有用户，以及关注某个用户的所有用户。

阅读更多：Flask 教程

创建 User 表

首先，我们需要创建一个 User 表，用于存储用户信息和关联关系。在这个表中，我们需要两个外键，分别对应关注者和被关注者的用户。

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'your_database_url'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))

    followers = db.relationship('User',
                                secondary='follower',
                                primaryjoin=('follower.c.follower_id == User.id'),
                                secondaryjoin=('follower.c.followed_id == User.id'),
                                backref=db.backref('followed_by', lazy='dynamic'),
                                lazy='dynamic')

follower = db.Table(
    'follower',
    db.Column('follower_id', db.Integer, db.ForeignKey('user.id')),
    db.Column('followed_id', db.Integer, db.ForeignKey('user.id'))
)

在上面的代码中，我们定义了一个 User 类，它表示用户的模型。表中有三个字段：id、name 和 followers。followers 字段是通过 db.relationship 方法创建的，它指定了与 User 表关联的表和关联条件。

查询关注用户

在我们创建了 User 表之后，我们可以轻松地查询一个用户关注的所有用户。以下是一个示例：

user = User.query.get(1)
followed_users = user.followers.all()

for followed_user in followed_users:
    print(followed_user.name)

在上面的代码中，我们首先通过 User.query.get(1) 获取一个用户实例。然后，我们使用 user.followers.all() 查询该用户所关注的所有用户，并将结果打印出来。

查询被关注用户

如果我们想要查找关注某个用户的所有用户，也很容易实现。以下是一个示例：

user = User.query.get(2)
followers = user.followed_by.all()

for follower in followers:
    print(follower.name)

与查询关注用户类似，我们首先获取一个用户实例，然后使用 user.followed_by.all() 查询该用户的所有关注者，并将结果打印出来。

更新关注关系

除了查询关注关系之外，我们还可以更新关注关系。例如，我们可以将一个用户关注另一个用户：

user1 = User.query.get(1)
user2 = User.query.get(2)

user1.followers.append(user2)
db.session.commit()

在上面的代码中，我们首先获取两个用户实例 user1 和 user2。然后，我们使用 user1.followers.append(user2) 将 user1 关注 user2。最后，使用 db.session.commit() 提交更改。

删除关注关系

同样地，我们可以删除关注关系。例如，我们可以取消一个用户对另一个用户的关注：

user1 = User.query.get(1)
user2 = User.query.get(2)

user1.followers.remove(user2)
db.session.commit()

在上面的代码中，我们首先获取两个用户实例 user1 和 user2。然后，我们使用 user1.followers.remove(user2) 取消 user1 对 user2 的关注。最后，使用 db.session.commit() 提交更改。

总结

本文介绍了如何使用 Flask 和 SqlAlchemy 实现同一表中的两列的关联。通过使用 db.relationship 和 secondary 参数，我们可以方便地查询一个用户关注的所有用户，以及关注某个用户的所有用户。还可以通过 append 和 remove 方法更新和删除关注关系。这种关联方式在社交网络应用程序等需要管理用户关系的场景中非常有用。

希望本文能帮助你理解和使用 Flask 和 SqlAlchemy 的关联功能。