web-dev-qa-db-fra.com

Table secondaire SQLAlchemy ManyToMany avec champs supplémentaires

J'ai 3 tables: utilisateur, communauté, community_members (pour la relation plusieurs2many d'utilisateurs et de la communauté). 

Je crée ces tables en utilisant 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):
    __table= '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'))

Maintenant, je veux ajouter un champ supplémentaire à community_members comme ceci:

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()),
                )

Et maintenant, dans Python Shell, je peux le faire:

créer une communauté:

> c = Community()
> c.name = 'c1'
> db.session.add(c)
> db.session.commit()

ajouter des membres à la communauté:

> 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>]

Ok, ça marche.

Mais comment puis-je obtenir time_create de la table community_members?

40
lestat

Vous devrez passer d'une relation simple, plusieurs à plusieurs à un "objet d'association", qui consiste simplement à prendre la table d'association et à lui attribuer un mappage de classe approprié. Vous définissez ensuite les relations un-à-plusieurs avec User et Community:

class Membership(db.Model):
    __table= '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):
    __table= 'community'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False, unique=True)

Mais vous ne pouvez être intéressé que de temps en temps par le temps de création; vous voulez retrouver l'ancienne relation! Eh bien, vous ne voulez pas configurer la relationship deux fois; parce que sqlalchemy pensera que vous voulez d'une manière ou d'une autre deux associations; ce qui doit signifier quelque chose de différent! Vous pouvez le faire en ajoutant un proxy association.

from sqlalchemy.ext.associationproxy import association_proxy

Community.members = association_proxy("memberships", "user")
User.communities = association_proxy("memberships", "community")
53

Si vous n'avez besoin que de la table de requête community_members et community par un user_id connu (tel que user_id = 2), dans SQLAlchemy, vous pouvez effectuer:

session.query(community_members.c.time_create, Community.name).filter(community_members.c.user_id==2)

pour obtenir le résultat.

0
carton.swing