SQLAlchemy，PostgreSQL：查询具有数组中特定项的行

Question

我试图根据数组中存在的整数变量使用SQLALchemy ORM对象过滤PostgreSQL记录，但我找不到正确的方法。

数据库设置

我有一个带array of integers的PostgreSQL表：

my_db=> \d test_arr;
                           Table "public.test_arr"
  Column  |   Type    |                       Modifiers
----------+-----------+-------------------------------------------------------
 id       | integer   | not null default nextval('test_arr_id_seq'::regclass)
 partners | integer[] |

该表包含一些值：

my_db=> SELECT * FROM test_arr;
 id | partners
----+----------
 12 | {1,2,3}
 13 | {2,3,4}
 14 | {3,4,5}
 15 | {4,5,6}
(4 rows)

使用2关键字在PostgreSQL中查询partners数组中包含ANY数字的行的表：

my_db=> SELECT * FROM test_arr WHERE 2 = ANY(partners);
 id | partners
----+----------
 12 | {1,2,3}
 13 | {2,3,4}
(2 rows)

ORM与SQLAlchemy

SQLAlchemy supports PostgreSQL arrays，表格由：

表示

class TestArr(Base):

    __tablename__ = 'test_arr'

    # override schema elements like Columns
    id = Column('id', Integer, Sequence('test_arr_id_seq'), primary_key=True)
    partners = Column(ARRAY(Integer))

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

    def __str__(self):
        return '<%d for %s>' % (self.id, str(self.partners))

问题

如何使用Pythonic对象运行等效的SELECT * FROM test_arr WHERE 2 = ANY(partners);？

我尝试了什么

我尝试使用func，如下所示：

print session.query(TestArr).filter(func.any(TestArr.partners, 2)).all()

但语法错误失败：

sqlalchemy.exc.ProgrammingError: (ProgrammingError) syntax error at or near "ANY"
LINE 3: WHERE ANY(test_arr.partners, 332)

Answer 1

您可以使用sqlalchemy.dialects.postgresql.Any

from sqlalchemy.dialects.postgresql import Any
session.query(TestArr).filter(Any(2, TestArr.partners)).all()

或@>数组运算符

session.query(TestArr).filter(TestArr.partners.op('@>')([2])).all()