Initial commit: moon-household-budget

income.py

@@ -0,0 +1,79 @@
+import json
+import os
+from datetime import date
+from storage import load_json, save_json
+
+DATA_DIR = os.environ.get("DATA_DIR", os.path.join(os.path.dirname(__file__), "data"))
+INCOME_FILE = os.path.join(DATA_DIR, "income_log.json")
+
+INCOME_STREAMS = [
+    {"name": "Auto King",   "owner": "Tony",  "type": "paycheck"},
+    {"name": "Studio",      "owner": "Bonna", "type": "business"},
+    {"name": "Bakery",      "owner": "Bonna", "type": "business"},
+    {"name": "Car Flipping","owner": "Tony",  "type": "business"},
+    {"name": "Misc",        "owner": "Both",  "type": "misc"},
+]
+
+OWNER_COLORS = {
+    "Bonna": "#D4B8C5",
+    "Tony":  "#A0B8C5",
+    "Both":  "#B8A8D4",
+}
+
+
+def load_income_log():
+    return load_json(INCOME_FILE, [])
+
+
+def save_income_log(log):
+    save_json(INCOME_FILE, log)
+
+
+def month_key(year, month):
+    return f"{year}-{month:02d}"
+
+
+def income_by_month(year, month):
+    log = load_income_log()
+    mk = month_key(year, month)
+    entries = [e for e in log if e.get("month") == mk]
+    by_stream = {}
+    for e in entries:
+        s = e["stream"]
+        by_stream[s] = by_stream.get(s, 0) + float(e.get("amount", 0))
+    return by_stream, entries
+
+
+def income_ytd(year):
+    log = load_income_log()
+    by_stream = {}
+    by_month = {}
+    for e in log:
+        mk = e.get("month", "")
+        if not mk.startswith(str(year)):
+            continue
+        s = e["stream"]
+        amt = float(e.get("amount", 0))
+        by_stream[s] = by_stream.get(s, 0) + amt
+        by_month[mk] = by_month.get(mk, 0) + amt
+    return by_stream, by_month
+
+
+def log_income_entry(stream, amount, month_str, note="", entry_id=None):
+    import uuid
+    log = load_income_log()
+    log.append({
+        "id": entry_id or str(uuid.uuid4()),
+        "stream": stream,
+        "amount": float(amount),
+        "month": month_str,
+        "date": date.today().isoformat(),
+        "note": note,
+    })
+    save_income_log(log)
+
+
+def delete_income_entry(entry_id):
+    log = load_income_log()
+    log = [e for e in log if e.get("id") != entry_id]
+    save_income_log(log)