0

私はこのようなモデルを持っています:

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 ($)")

各カテゴリとショーのすべての予算の合計を見つけたい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)

一部のカテゴリにはまだタスクがない場合があります。したがって、次のようにcategories_costなります。

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

したがって、減算しようとすると、次のエラーが発生します。

>>> 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'

これを解決する方法は？

4

1 に答える 1

0

多くの解決策があります。これは最も簡単です：

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

またはマップを使用：

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

またはフィルターを使用：

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

rem = filter(no_nones, rem)
于 2013-10-06T11:20:02.580 に答える