python - 在 Pony ORM 中增量构建查询

Question

我正在评估从 peewee 到 Pony ORM 的转换。 peewee 中提供的一件好事是能够从这样的部分组成查询:

def Entry(BaseModel):
    # peewee fields go here

def where_entry_category(category, recurse=False):
    """ Generate a where clause for a particular category """

    if category or not recurse:
        cat_where = (Entry.category == str(category))

        if recurse:
            # We're recursing and aren't in /, so add the prefix clause
            cat_where = cat_where | (
                Entry.category.startswith(str(category) + '/'))
    else:
        cat_where = True

    return cat_where

query = Entry.select().where(where_entry_category("test"))

其工作方式是 peewee 模型类型上的各种运算符重载仅返回查询组件树，这些子树可以由进一步的运算符重载组成。使用 & 或 | 运算符链接在一起的多个查询组件也是一件简单的事情，例如model.Entry.select().where(some_test() & some_other_test())。这非常有用，因为我的许多过滤查询都是以模块化方式组合的，而且大多数底层查询部分经常被重用，而且很多都是重要的(例如上面的示例)。

然而，在 Pony ORM 中，似乎只有(非常聪明!)AST generator parser和原始 SQL。由于原始 SQL 表单无法让我直接传递必要的查询部分，因此如果可能的话，我更愿意使用一些更高级别的查询构建功能。

如果我尝试将查询部分定义为模型上的方法，例如:

class Entry(db.Entity):
    ...
    def in_category(self, category, recurse=False):
        # return a test on the parameters

orm.select(entry for entry in model.Entry if entry.in_category('foo', True)) 

不出所料，我得到了 NotImplementedError。

是否有从现有部分构建查询表达式并传递到 SQL 查询构建器的机制？ (也许通过自己构建 AST 并将其传递到 Pony 的相关部分，或者有一种机制，我可以传递一个查询以被另一个子查询过滤。)

Answer 1

在 PonyORM 中，您可以通过两种方式逐步编写查询。第一个是 filter查询方法:

def where_entry_category(query, category, recourse)
    if category:
        category = str(category)
        if recurse:
            query = query.filter(lambda x: x.category == category or
                                 x.category.startswith(category + '/')
        else:
            query = query.filter(lambda x: x.category == category)
    return query

query = Entry.select()
query = where_entry_category(query, "test")

从版本 0.7.6 开始，也可以使用以前的查询作为新查询的来源:

def where_entry_category(query, category, recourse)
    if category:
        category = str(category)
        if recurse:
            query = select(x for x in query
                           if x.category == category or
                              x.category.startswith(category + '/'))
        else:
            query = select(x for x in query if x.category == category)
    return query

你可能遇到的唯一问题是，如果你想逐渐构建具有可变数量子表达式的 或 子句，此时 Pony 没有 API。可能我们会在未来的版本中逐渐增加向 或 子句添加子表达式的可能性。