Djangoでデータベースからデータをとってくる

Question

開発環境

• Python 3.6.5
• Django 2.1.2

直面している問題

次のようなモデルを作って、利用しています。

from django.db import models

# Create your models here.
class Money(models.Model):
    use_date = models.DateField('日付')
    detail = models.CharField(max_length=200)
    cost = models.IntegerField(default=0)
    category = models.CharField(max_length=10)

    def __str__(self):
        return self.detail + ' ￥' + str(self.cost)

ビューを作るときにここからデータをとってこようと考えています。問題が生じているのは、特定の月の支出の合計額をデータベースから計算するコードです。

def index(request, year=TODAY[0], month=TODAY[1]):
    money = Money.objects.filter(use_date__year=year,
            use_date__month=month).order_by('use_date')
    total = 0
    for m in money:
        total += m.cost
#...

これだとうまくいきます。しかし合計額を計算する部分を関数として抽出すると途端にエラーが出てしまいます。

def calc_month_pay(money):
    total = 0
    for m in money:
        total += m.cost

    return cost


def index(request, year=TODAY[0], month=TODAY[1]):
    money = Money.objects.filter(use_date__year=year,
            use_date__month=month).order_by('use_date')
    total = calc_month_pay(money)

#...

NameError: name 'cost' is not defined

なぜでしょうか？

Answer 1

return cost

この cost という変数が未定義です

Answer 2

……エラー文を読んだそのままですが、 cost という未定義の変数を導入されたから、かと思います。

def calc_month_pay(money):
    total = 0
    for m in money:
        total += m.cost

-    return cost
+    return total

Comments

[gh640]

ちなみに、単純に int の合計値を取りたい場合は `QuerySet` の `aggregate()` メソッドを使うこともできるかと思います（そして、その方が一般的かと思います）。ご参考になさってください。

- https://stackoverflow.com/questions/6481279/django-sum-query

[gh640]

さらにちなみに、コーディング中に未定義変数等に気づけるように、おそらく多くのプログラマは IDE / エディタの linting 機能を利用しています。お使いの IDE またはエディタの名前 + linting 等のキーワードでぐぐるとそういったツールが見つかりますので、ぜひお探しになってみてください :)

[Yhaya]

calc_month_payという関数に渡している引数は、辞書のようなオブジェクトで、その中にはモデルで定義しているような「use_date」、「cost」等をkeyのような形で持っていると考えているのですが違うのでしょうか？

[Yhaya]

実際に上のうまく動いているほうのコードでは「m.cost」でアクセスできているのに、同じものを引数に渡した瞬間にNameErrorが出る理由が知りたいです。

[Yhaya]

完全に勘違いしていました。returnのほうですね。解決しました。ありがとうございます。

[gh640]

そうですか。すぐに反応ができませんでしたが……ご解決されたとのことでよかったです！

余談ですが、エラーが出たときには

> NameError: name 'cost' is not defined

の近くに、エラーが出た行（今回の場合ですと return cost の行）を示す出力が出たのではないかと思います。エラーが出たときはエラー内容とあわせて該当行を必ずチェックするようにされると、問題の絞り込みがしやすくなるかと思います。ご参考になさってください。