Appliquer condition dans cette requête dans django

Question

j'ai modèles comme ceci:

class Category(models.Model): 

    user = models.ForeignKey(User) 
    name = models.CharField(max_length=128) 
    budget = models.DecimalField(default=0.0, decimal_places=2, max_digits=8, help_text="Amount in dollars ($)") 

class Task(models.Model): 

    description = models.CharField(max_length=128) 
    owner = models.ForeignKey(User, default="bride") 
    category = models.ForeignKey(Category) 
    cost = models.DecimalField(default=0.0, decimal_places=2, max_digits=6, help_text="Amount in dollars ($)") 

Je veux trouver la somme de tous les budgets de chaque catégorie et montrer remaining budget:

categories = Category.objects.filter(user_id=request.user.id).order_by("name") 
categories_cost = [Task.objects.filter(category=i).aggregate(Sum('cost')) for i in categories] 

category_budget = [i.budget for i in categories] 

categories = zip(categories, remaining_budget) 

Certaines catégories ne peuvent pas avoir encore une tâche. Ainsi, categories_cost ressemblerait à ceci:

[{'cost__sum': None}, {'cost__sum': Decimal('0.00')}, {'cost__sum': None}, {'cost__sum': Decimal('300.00')}] 

Alors, quand je tente de soustraire je reçois cette erreur:

>>> remaining_budget = [i.budget - j['cost__sum'] for i, j in zip(categories, rem)] 
Traceback (most recent call last): 
    File "<console>", line 1, in <module> 
TypeError: unsupported operand type(s) for -: 'Decimal' and 'NoneType' 

Comment résoudre ce problème?

Answer 1

Il existe de nombreuses solutions. C'est la plus simple:

remaining_budget = [i.budget - (j['cost__sum'] or 0) for i, j in zip(categories, rem)] 

ou en utilisant la carte:

rem = map(lambda x: (x or 0), rem) 

ou en utilisant un filtre:

def no_nones(item): 
    if item is None: 
     return 0 
    return item 

rem = filter(no_nones, rem)