SQLAlchemy ManyToMany вторичная таблица с дополнительными полями
У меня есть 3 таблицы: User, Community, community_members (для отношений many2many пользователей и сообщества).
Я создаю эти таблицы с помощью Flask-SQLAlchemy:
community_members = db.Table('community_members',
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('community_id', db.Integer, db.ForeignKey('community.id')),
)
class Community(db.Model):
__tablename__ = 'community'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False, unique=True)
members = db.relationship(User, secondary=community_members,
backref=db.backref('community_members', lazy='dynamic'))
Теперь я хочу добавить дополнительное поле в community_members следующим образом:
community_members = db.Table('community_members',
db.Column('id', db.Integer, primary_key=True),
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('community_id', db.Integer, db.ForeignKey('community.id')),
db.Column('time_create', db.DateTime, nullable=False, default=func.now()),
)
И теперь в оболочке python я могу сделать это:
создать сообщество:
> c = Community()
> c.name = 'c1'
> db.session.add(c)
> db.session.commit()
добавить членов сообщества:
> u1 = User.query.get(1)
> u2 = User.query.get(2)
> c.members.append(u1)
> c.members.append(u2)
> db.session.commit()
> c.members
[<User 1>, <User 2>]
Хорошо, это работает.
Но как теперь я могу получить time_create из таблицы community_members?
Ответы
Ответ 1
Вам придется перейти от использования простого отношения "многие ко многим" к использованию "объекта ассоциации", который в основном просто берет таблицу ассоциации и дает ей правильное отображение классов. Затем вы определите отношения "один ко многим" с User
и Community
:
class Membership(db.Model):
__tablename__ = 'community_members'
id = db.Column('id', db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
community_id = db.Column(db.Integer, db.ForeignKey('community.id'))
time_create = db.Column(db.DateTime, nullable=False, default=func.now())
community = db.relationship(Community, backref="memberships")
user = db.relationship(User, backref="memberships")
class Community(db.Model):
__tablename__ = 'community'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False, unique=True)
Но вас могут интересовать только время создания; Вы хотите вернуть старые отношения! ну, вы не хотите настраивать relationship
дважды; потому что sqlalchemy будет думать, что вы как-то хотите две ассоциации; что должно означать что-то другое! Вы можете сделать это, добавив прокси-сервер ассоциации.
from sqlalchemy.ext.associationproxy import association_proxy
Community.members = association_proxy("memberships", "user")
User.communities = association_proxy("memberships", "community")
Ответ 2
Если вам нужна только таблица запросов community_members
и community
с помощью известного user_id (например, user_id = 2), в SQLAlchemy вы можете выполнить:
session.query(community_members.c.time_create, Community.name).filter(community_members.c.user_id==2)
чтобы получить результат.