Django：按值过滤或返回所有记录

Question

我有以下代码（django，python）：def function(self, some_id):    if some_id:        return Model1.objects.filter(model2__id=some_id)    else:        return Model1.objects.all()我想知道是否可以使用单个return语句来重构此代码？Model1在Model2.id上具有外键。

Answer 1

def function(self, some_id):

    filter_kwargs = {}

    if some_id:

        filter_kwargs['model2__id'] = some_id

    return Model1.objects.filter(**filter_kwargs)

应该可以满足您的问题，尽管我会说书面形式最有可能。

Answer 2

好吧，您可以像这样解决此问题：

def function(self, some_id):

    return Model1.objects.filter(*(some_id and [Q(model2__id=some_id)] or []))

但这是很神秘的。

我个人很喜欢实现一个filter_not_none，它只是过滤掉带有Noneas值的语句，例如：

def filter_not_none(qs, **kwargs):

    return qs.filter(**{k: v for k, v in kwargs.item() if v is not None})

然后我们可以简单地使用：

def function(self, some_id):

    return filter_not_none(Model1.objects, model2__id=some_id)

这适用于任意数量的命名参数。因此不是位置参数或Q对象。

这是有效的，因为filter(..)没有参数的a不会过滤任何内容。我们使用字典理解，从而检查哪个kwargs具有非None值，并且仅将它们传递给.filter(..)调用。

然后，您可以使用例如：

return filter_not_none(

    Model.objects,

    name=some_name,

    date__lt=some_date

)

如果some_name是None，我们做的名字没有过滤器，如果some_date是None我们没有过滤器时date__lt，如果两者都是None，我们不会过滤所有，并且如果两者都没有 None，我们过滤器上都。

我们可以对其进行扩展以使其也可以与Q对象一起使用*args，但是检查这些-Q对象和表达式将非常困难，并且还可能导致各种不良行为。但是，也需要Q对象的扩展是：

def filter_not_none(qs, *args, **kwargs):

    # warning, does not perform manipulations on Q-objects, Q-expressions

    # and unnamed parameters in general

    return qs.filter(

        *args,

        **{k: v for k, v in kwargs.item() if v is not None}

    )