python - 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()

我想知道是否可以用一个返回语句来重构这段代码？ 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)

应该可以满足您的问题，但我会说它很可能与书面上的一样。