How can I make django query based on function input?

Question

I want to make django query based on function input.I don't want to check input with if and else in 2 separate line and then make query like this:

def customer_number(mall,store=None):
    if store :
        customer = models.Customer.objects.filter(mall=mall,store=store)
    else :
        customer = models.Customer.objects.filter(mall=mall)
    return customer.count

CJ4 · Accepted Answer · 2020-12-08 11:31:21Z

1

You can construct a lookup kwarg dictionary:

def customer_number(mall, store=None):
    lookup_kwargs = {'mail': mail}
    if store:
        lookup_kwargs['store'] = store
    return models.Customer.objects.filter(**lookup_kwargs).count()

Alternatively if you don't want any conditionals, you can construct the dict and remove None values:

def customer_number(mall, store=None):
    lookup_kwargs = {'mail': mail, 'store': store}
    lookup_kwargs = {k: v for k, v in lookup_kwargs.items() if v is not None}
    return models.Customer.objects.filter(**lookup_kwargs).count()

answered Dec 8, 2020 at 11:31

CJ4

2,5353 gold badges28 silver badges31 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How can I make django query based on function input?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related