django - Django - 在 Q 过滤器中使用字符串变量

Question

我有一个非常相似的查询，适用于两种不同类型的输入，并且会在正确的列上匹配。除了一个词，即列名之外，这两个查询基本相同。

def trade_company_group(company_group):
    # can filter both name or alias as name will always contain 'Companies'
    if "COMPANIES" in company_group.upper():
        return (
            // Same query as below except for "name", below is "alias"
            Q(buy_book__entity__company_groups__name__iexact=company_group) &
            Q(sell_book__entity__company_groups__name__iexact=company_group) &
            (
                ~Q(buy_book__entity__type=ENTITY.INTERNAL) | 
                (
                    Q(buy_book__entity__primary_company_group__name__iexact=company_group) |
                    Q(sell_book__entity__primary_company_group__name__iexact=company_group)
                )
            )) 
    
    return (
        Q(buy_book__entity__company_groups__alias__iexact=company_group) &
        Q(sell_book__entity__company_groups__alias__iexact=company_group) &
        (
            ~Q(buy_book__entity__type=ENTITY.INTERNAL) | 
            (
                Q(buy_book__entity__primary_company_group__alias__iexact=company_group) |
                Q(sell_book__entity__primary_company_group__alias__iexact=company_group)
            )
        ))

我不想重复代码，所以我希望有一种方法可以根据我的 if 语句替换查询中的列名。

这可能吗？

标签： djangodjango-q

Answer 1

您可以改用 dict

.eg 这个：

Q(buy_book__entity__type=ENTITY.INTERNAL)

相当于这个：

q_filter = {"buy_book__entity__type": ENTITY.INTERNAL}
Q(**q_filter)