微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python-SQLAlchemy,PostgreSQL:查询数组中具有特定项目的行

我试图根据数组中整数变量的存在使用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中使用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))

问题

我如何运行相当于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)

解决方法:

你可以用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()

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐