SQLAlchemy 表使用两个外键定义关系

Question

我有两个表，Users 和 ChatSessions。 ChatSessions 有两个字段，user_id 和friend_id，它们都是Users 表的外键。

user_id 始终包含发起聊天会话的用户，friend_id 是其他用户。由于某个用户可以有他自己或他的朋友发起的聊天会话，因此他可以在各种会话中将他的 id 作为 user_id 或friend_id。

是否可以在用户表中定义关系，我可以访问该用户的所有聊天会话，无论他的 id 是在 user_id 还是friend_id 中？ 像这样的：

chat_sessions = db.relationship('chat_sessions',
                primaryjoin="or_(User.id==ChatSession.user_id, User.id==ChatSession.friend_id)",
                backref="user")

当我尝试向用户表提交条目时收到以下错误：

ERROR   main.py:76      [10.0.2.2] Unhandled Exception [93e3f515-7dd6-4e8d-b096-8239313433f2]: relationship 'chat_sessions' expects a class or a mapper argument (received: <class 'sqlalchemy.sql.schema.Table'>)

模型：

class User(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(60), index=True, unique=True)
    password = db.Column(db.String(255))
    name = db.Column(db.String(100))
    active = db.Column(db.Boolean(), nullable=False)

    chat_sessions = db.relationship('chat_sessions',
                    primaryjoin="or_(User.id==ChatSession.user_id, User.id==ChatSession.friend_id)")



class ChatSession(db.Model):
    __tablename__ = 'chat_sessions'

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    friend_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    status = db.Column(db.String(50))


    user = db.relationship('User', foreign_keys=[user_id])
    friend = db.relationship('User', foreign_keys=[friend_id])

Answer 1

如果没有看到表的代码很难确定，但删除 backref 参数可能就足够了。

这是一个纯粹的 SQLAlchemy 实现，似乎可以满足您的需求：

import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import orm


Base = declarative_base()


class User(Base):
    __tablename__ = 'users'
    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.String)

    all_chats = orm.relationship('Chat',
        primaryjoin="or_(User.id==Chat.user_id, User.id==Chat.friend_id)")

    def __repr__(self):
        return f'User(name={self.name})'


class Chat(Base):
    __tablename__ = 'chats'

    id = sa.Column(sa.Integer, primary_key=True)
    user_id = sa.Column(sa.Integer, sa.ForeignKey('users.id'))
    friend_id = sa.Column(sa.Integer, sa.ForeignKey('users.id'))

    user = orm.relationship('User', foreign_keys=[user_id])
    friend = orm.relationship('User', foreign_keys=[friend_id])

    def __repr__(self):
        return f'Chat(user={self.user.name}, friend={self.friend.name})'


engine = sa.create_engine('sqlite:///')
Base.metadata.create_all(bind=engine)
Session = orm.sessionmaker(bind=engine)

usernames = ['Alice', 'Bob', 'Carol']
session = Session()
users = [User(name=name) for name in usernames]
session.add_all(users)
session.flush()

a, b, c = users

session.add(Chat(user_id=a.id, friend_id=b.id))
session.add(Chat(user_id=a.id, friend_id=c.id))
session.add(Chat(user_id=c.id, friend_id=a.id))
session.commit()
session.close()

session = Session()
users = session.query(User)
for user in users:
    for chat in user.all_chats:
        print(user, chat)
    print()
session.close()

这是输出：

User(name=Alice) Chat(user=Alice, friend=Bob)
User(name=Alice) Chat(user=Alice, friend=Carol)
User(name=Alice) Chat(user=Carol, friend=Alice)

User(name=Bob) Chat(user=Alice, friend=Bob)

User(name=Carol) Chat(user=Alice, friend=Carol)
User(name=Carol) Chat(user=Carol, friend=Alice)