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

Question

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

数据库设置

我有一个带有整数数组的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中partners排列使用PostgreSQL中完成ANY关键字：

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

带有 SQLAlchemy 的 ORM

SQLAlchemy支持 PostgreSQL 数组，表表示为：

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

问题

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

我试过什么

我试过使用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()