examples.association.dict_of_sets_with_default — SQLAlchemy 1.1 Documentation

Release: 1.1.18 | Release Date: March 6, 2018

# SQLAlchemy 1.1 Documentation

# Source code for examples.association.dict_of_sets_with_default

    """dict_of_sets_with_default.py

    an advanced association proxy example which
    illustrates nesting of association proxies to produce multi-level Python
    collections, in this case a dictionary with string keys and sets of integers
    as values, which conceal the underlying mapped classes.

    This is a three table model which represents a parent table referencing a
    dictionary of string keys and sets as values, where each set stores a
    collection of integers. The association proxy extension is used to hide the
    details of this persistence. The dictionary also generates new collections
    upon access of a non-existent key, in the same manner as Python's
    "collections.defaultdict" object.

    """

    from sqlalchemy import String, Integer, Column, create_engine, ForeignKey
    from sqlalchemy.orm import relationship, Session
    from sqlalchemy.orm.collections import MappedCollection
    from sqlalchemy.ext.declarative import declarative_base
    from sqlalchemy.ext.associationproxy import association_proxy
    import operator

    class Base(object):
        id = Column(Integer, primary_key=True)

    Base = declarative_base(cls=Base)

    class GenDefaultCollection(MappedCollection):
        def __missing__(self, key):
            self[key] = b = B(key)
            return b

    class A(Base):
        __tablename__ = "a"
        associations = relationship(
            "B",
            collection_class=lambda: GenDefaultCollection(operator.attrgetter("key"))
        )

        collections = association_proxy("associations", "values")
        """Bridge the association from 'associations' over to the 'values'
        association proxy of B.
        """

    class B(Base):
        __tablename__ = "b"
        a_id = Column(Integer, ForeignKey("a.id"), nullable=False)
        elements = relationship("C", collection_class=set)
        key = Column(String)

        values = association_proxy("elements", "value")
        """Bridge the association from 'elements' over to the
        'value' element of C."""

        def __init__(self, key, values=None):
            self.key = key
            if values:
                self.values = values

    class C(Base):
        __tablename__ = "c"
        b_id = Column(Integer, ForeignKey("b.id"), nullable=False)
        value = Column(Integer)

        def __init__(self, value):
            self.value = value

    if __name__ == '__main__':
        engine = create_engine('sqlite://', echo=True)
        Base.metadata.create_all(engine)
        session = Session(engine)

        # only "A" is referenced explicitly.  Using "collections",
        # we deal with a dict of key/sets of integers directly.

        session.add_all([
            A(collections={
                "1": set([1, 2, 3]),
            })
        ])
        session.commit()

        a1 = session.query(A).first()
        print(a1.collections["1"])
        a1.collections["1"].add(4)
        session.commit()

        a1.collections["2"].update([7, 8, 9])
        session.commit()

        print(a1.collections["2"])