Как определить, включен ли один PGArray в другой с помощью сеансов SQLAlchemy?

Question

У меня есть таблица SqlAlchemy, вот так:

table = sql.Table('treeItems', META,
    sql.Column('id', sql.Integer(), primary_key=True),
    sql.Column('type', sql.String, nullable=False),
    sql.Column('parentId', sql.Integer, sql.ForeignKey('treeItems.id')),
    sql.Column('lineage', PGArray(sql.Integer)),
    sql.Column('depth', sql.Integer),
)

Который сопоставляется с объектом следующим образом:

orm.mapper(TreeItem, TreeItem.table, polymorphic_on=TreeItem.table.c.type, polymorphic_identity='TreeItem')

Я бы хотел выбрать любой дочерний узел данного узла, так что я ищу SQL, который выглядит следующим образом (для родителя с pk = 2):

SELECT *
FROM "treeItems"
WHERE ARRAY[2] <@ "treeItems".lineage AND "treeItems".id != 2
ORDER BY "treeItems".lineage

Вот код SqlAlchemy / Python, который я использую, чтобы попытаться добраться до вышеприведенного SQL без особой удачи:

arrayStr = 'ARRAY[%s]' % ','.join([str(i) for i in self.lineage])
lineageFilter = expr.text('%s <@ %s' % (arrayStr, TreeItem.table.c.lineage))
query = SESSION.query(TreeItem).filter(expr.and_(lineageFilter, TreeItem.table.c.id!=self.id))

Но вот SQL, с которым я столкнулся (обратите внимание на отсутствие кавычек вокруг имени таблицы treeItems в предложении where):

SELECT "treeItems".id AS "treeItems_id", "treeItems".type AS "treeItems_type", "treeItems"."parentId" AS "treeItems_parentId", "treeItems".lineage AS "treeItems_lineage", "treeItems".depth AS "treeItems_depth"
FROM "treeItems"
WHERE ARRAY[2] <@ treeItems.lineage AND "treeItems".id != %(id_1)s

Итак, теперь по вопросам:

Есть ли лучший способ сделать это, чем использовать выражение text () / В SqlAlchemy есть оператор или выражение, которое может выполнять <@ с PGArray? </strong>

Как я могу получить кавычки, которые будут отображаться вокруг имени моей таблицы, если я должен использовать выражение text ()?

Спасибо всем!

Answer 1

Элементы предложения SQLAlchemy имеют метод .op () для пользовательских операторов. Что недоступно, так это специальное предложение для литералов массива. Вы можете указать литерал массива с помощью literal_column:

print sql.literal_column('ARRAY[2]').op('<@')(table.c.lineage)
# ARRAY[2] <@ "treeItems".lineage

Если вам нужен лучший API для литералов массива, вы можете создать его с помощью модуля sqlalchemy.ext.compiler , добавленного в SQLAlchemy 0.5.4.

Answer 2

В этом конкретном случае я заметил, что цитирование в SQL было связано с тем, что я использовал имя таблицы в смешанном регистре. Преобразование имени таблицы из «treeItems» в «tree_items» решило проблему с цитированием, и я смог заставить свое текстовое выражение работать:

expr.text('%s <@ %s' % (arrayStr, TreeItem.table.c.lineage))

Это исправление, и приятно знать, что имена таблиц в смешанном регистре нужно заключать в кавычки, но ответ Антса остается правильным способом решения проблемы.