Отношения SQLAlchemy с Postgresql ARRAY - PullRequest
Новая
       11

Отношения SQLAlchemy с Postgresql ARRAY

9 голосов
/ 16 марта 2012

Предположим, у нас есть база данных PostgreSQL с двумя таблицами A, B. 

table A columns: id, name
table B columns: id, name, array_a

Столбец array_a в таблице B содержит массив идентификаторов переменной длины из таблицы A. В SQLAlchemy у нас есть два классакоторые моделируют эти таблицы, скажем, класс A и B.

Следующее прекрасно работает, чтобы получить все объекты A, на которые есть ссылки в объекте B: 

session.query(A).join(B, A.id == func.any(B.array_a)).filter(B.id == <id>).all()

Как мы можем создать отношенияв B ссылаясь на объекты A, соответствующие массиву?Попытки сравнения столбцов с использованием func.any выше, но он жалуется, что ANY(array_a) не является столбцом в модели.Указание условий первичного соединения, как указано выше, похоже, также не сокращает его.

1 Ответ

8 голосов
/ 16 марта 2012

Этот анти-паттерн называется "Jaywalking" ; и мощная система типов PostgreSQL делает ее очень заманчивой. Вы должны использовать другую таблицу: 

CREATE TABLE table_a (
    id SERIAL PRIMARY KEY,
    name VARCHAR
);

CREATE TABLE table_b (
    id SERIAL PRIMARY KEY,
    name VARCHAR
);

CREATE TABLE a_b (
    a_id INTEGER PRIMARY KEY REFERENCES table_a(id),
    b_id INTEGER PRIMARY KEY REFERENCES table_b(id)
)

Который сопоставлен: 

from sqlalchemy import *
from sqlalchemy.dialects import postgresql
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import *

Base = declarative_base()

a_b_table = Table("a_b", Base.metadata,
    Column("a_id", Integer, ForeignKey("table_a.id"), primary_key=True),
    Column("b_id", Integer, ForeignKey("table_b.id"), primary_key=True))

class A(Base):
    __tablename__ = "table_a"
    id = Column(Integer, primary_key=True)
    name = Column(String)

class B(Base):
    __tablename__ = "table_b"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    a_set = relationship(A, secondary=a_b_table, backref="b_set")

пример: 

>>> print Query(A).filter(A.b_set.any(B.name == "foo"))
SELECT table_a.id AS table_a_id, table_a.name AS table_a_name 
FROM table_a 
WHERE EXISTS (SELECT 1 
FROM a_b, table_b 
WHERE table_a.id = a_b.a_id AND table_b.id = a_b.b_id AND table_b.name = :name_1)

Если вы застряли в столбце ARRAY, лучше всего использовать альтернативный выбор, который «выглядит» как правильная таблица ассоциаций. 

from sqlalchemy import *
from sqlalchemy.dialects import postgresql
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import *

Base = declarative_base()


class A(Base):
    __tablename__ = "table_a"
    id = Column(Integer, primary_key=True)
    name = Column(String)

class B(Base):
    __tablename__ = "table_b"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    array_a = Column(postgresql.ARRAY(Integer))

a_b_selectable = select([func.unnest(B.array_a).label("a_id"),
                         B.id.label("b_id")]).alias()

A.b_set = relationship(B, secondary=a_b_selectable,
                          primaryjoin=A.id == a_b_selectable.c.a_id,
                          secondaryjoin=a_b_selectable.c.b_id == B.id,
                          viewonly=True,)

B.a_set = relationship(A, secondary=a_b_selectable,
                          primaryjoin=A.id == a_b_selectable.c.a_id,
                          secondaryjoin=a_b_selectable.c.b_id == B.id,
                          viewonly=True)

, что дает вам: 

>>> print Query(A).filter(A.b_set.any(B.name == "foo"))
SELECT table_a.id AS table_a_id, table_a.name AS table_a_name 
FROM table_a 
WHERE EXISTS (SELECT 1 
FROM (SELECT unnest(table_b.array_a) AS a_id, table_b.id AS b_id 
FROM table_b) AS anon_1, table_b 
WHERE table_a.id = anon_1.a_id AND anon_1.b_id = table_b.id AND table_b.name = :name_1)

И, очевидно, поскольку там нет реального стола, viewonly=True является необходимым, и вы не можете получить прекрасное, динамичное объектное совершенство, которое вы получили бы, если бы избегали пешеходов.

...