如何根据使用 Sqlalchemy 计算的 json 值选择 Postgresql 记录？

Question

我有一个包含 2 列的 postgres 表，用于查询：

start_date_detail: date 
flex: jsonb

弹性域示例：

{
    "communication": [
        {
            "remind_on": "mxmw@cxla.nl",
            "type": "email",
            "remind_date": -150
        },
        {
            "remind_on": "+31612345678",
            "type": "sms",
            "remind_date": -360
        }
    ]
}

我需要选择所有提醒日期在上周的记录，所以（伪代码）：

(now() - 1-week) < (start_date_detail + remind_date) < now()

我如何使用 sqlalchemy 实现这一点？

因为它是查询中的计算值，所以我不知道该怎么做。 在 postgres 中，我想出了这个，并且有效：

SELECT * FROM time_item 
WHERE 
    (start_date_detail + INTERVAL '1 second' * (flex->'communication'->0->>'remind_date')::numeric <= NOW())
    OR (start_date_detail + INTERVAL '1 second' * (flex->'communication'->1->>'remind_date')::numeric <= NOW())

如何将其放入 sqlalchemy 中？

还有一点： 在上面的查询中，我将每个通信项添加到 where 子句。我怎样才能使它更灵活？那就是我不需要为每个通信项都添加 where 子句。

Answer 1

您可以使用jsonb_array_elements() 将数组扩展为一组jsonb 元素，然后您可以在谓词中使用这些元素。 SQLAlchemy 通过使用alias 支持将函数表达式作为可选择项。使用诸如

之类的模型

In [4]: class TimeItem(Base):
   ...:     __tablename__ = 'time_item'
   ...:     id = Column(Integer, primary_key=True)
   ...:     start_date_detail = Column(Date)
   ...:     flex = Column(JSONB)
   ...:     

查询可能看起来像

In [39]: session.query(TimeItem).\
    ...:     select_from(TimeItem,
    ...:                 func.jsonb_array_elements(TimeItem.flex['communication']).
    ...:                     alias('comm')).\
    ...:     filter((TimeItem.start_date_detail +
    ...:             timedelta(seconds=1) *
    ...:             column('comm', type_=JSONB)['remind_date'].
    ...:                 astext.
    ...:                 cast(Integer)).
    ...:                     between(func.now() - timedelta(weeks=1),
    ...:                             func.now())).\
    ...:     all()

然后您可以调整谓词以满足您的需求——我尝试按照您的示例将remind_date 解释为start_date_detail 的偏移秒数，与前一周的现在和现在相比。在查询 TimeItem 等实体时，SQLAlchemy 会根据对象身份进行自己的重复数据删除，因此查询可以省略 SQL 端 DISTINCT、在 EXISTS 子查询表达式中移动数组元素等。