Pypika:是否可以动态更改where子句中的运算符

Question

我正在使用Pypika查询HDB。通常我会创建查询，然后像这样添加where子句：

query = query.where(table.ID == value)

现在我有一个情况，我不知道将使用哪个运算符。我可以是任何可能的运算符：

['==', '<>', '<', '>', '<=', '>=']

有没有办法为Pypika动态指定运算符？目前我正在做这件事：

if operator == '<=':
        query = query.where(table.ID == value)
elif operator == '>=':
    ...

但我想这样做：

operator = '=='
query = query.where(table.ID {operator} value)

感谢您的任何意见！

PS:不确定这是否是你在这里做的:也许有更多分数的人可以为Pypika图书馆添加一个标签

Answer 1

我看到你在大约30天前发布了这篇文章，所以我希望你在此期间已经找到了解决方案。但是，如果你没有-我遇到了同样的问题，下面的方法对我有效:编写一个函数，从operator模块中获取一个运算符，并生成查询：

from pypika import Table, Field, Query
from typing import List, Callable
import operator


def gen_my_query(varname: str, 
             my_operator: Callable, 
             permitted_value: int, 
             my_table: str, 
             selected_vars: List[str]):

    my_query = Query.from_(Table(my_table)
                          ).select(','.join(selected_vars)
                          ).where(
                                  my_operator(Field(varname), permitted_value)
                          ).get_sql(quote_char=None)

    return my_query

该函数的使用方法如下：

gen_my_query(varname='id', 
             my_operator=operator.eq, 
             permitted_value=1, 
             my_table='table', 
             selected_vars=["var1", "var2", "var3"])

这将产生以下结果：

'SELECT var1,var2,var3 FROM table WHERE id=1'

如果您想继续使用定义为字符串的运算符，您可以例如在字典中定义一个映射，并重构上面的函数以从字典中获取正确的运算符，例如：

my_operators = {'==': operator.eq,
                '<': operator.le} # ... etc.

希望这能有所帮助！