自由拡張｜コードで実践 - ゆめさくコード

このレッスンでできるようになること

家計簿に自分で欲しい機能を1つ足すときの、決め方と組み立て方が身に付きます。ここでは予算アラートを題材にします。

機能を足したくなったとき、いきなりコードを書き始めると手が止まります。先に決めるのは次の2つです。

今回の予算アラートなら、入力は「支出のリスト」と「カテゴリごとの予算」の2つ、出力は「予算を超えたカテゴリの警告文のリスト」です。これが決まれば、あとは中を埋めるだけになります。

def budget_alerts(expenses, budgets):
    ...

budgets はカテゴリ名から金額への辞書にします。{"食費": 30000, "交通費": 5000} のような形です。

ここで print してしまいたくなりますが、我慢してリストを返す形にします。値を返しておくと、画面に出すことも、ファイルに書くことも、件数だけ数えることも、呼ぶ側で選べます。第7章までのレポート関数を全部この形にしてきたのと同じ理由です。

そして値を返す関数は、前のレッスンで書いたチェックスクリプトで assert できます。print する関数は目で見るしかありません。

集計は第2章の Counter、絞り込みは第3章の内包表記、並べ替えは第4章の sorted と key です。

overs = [
    (category, totals[category] - limit)
    for category, limit in budgets.items()
    if totals[category] > limit
]

budgets.items() を回しているのがポイントです。支出のほうを回すと、予算を決めていないカテゴリまで対象になってしまいます。予算を決めたカテゴリだけを見たいので、予算の辞書を主役にします。

totals[category] は Counter なので、そのカテゴリの支出が1件も無くても 0 が返ります。第2章で defaultdict と Counter を学んだときの、KeyError を出さない性質がここで効きます。

超過額の大きい順に並べます。同じ超過額のときはカテゴリ名の昇順にして、いつ動かしても同じ順になるようにします。

overs.sort(key=lambda pair: (-pair[1], pair[0]))

あとは1行ずつの文にするだけです。ここまで、新しい文法は1つも出てきていません。7章ぶんの道具で、新しい機能が1つ増えました。