1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
# Copyright (C) 2026 Danilo M. <danix@danix.xyz> GPL-2.0-only
import calendar
from datetime import date
from firefly_cli import registry, output
def _current_month():
"""(first, last) ISO dates for the current calendar month."""
today = date.today()
last_day = calendar.monthrange(today.year, today.month)[1]
first = today.replace(day=1).isoformat()
last = today.replace(day=last_day).isoformat()
return first, last
def _spent_scalar(budget_obj):
"""Sum a budget's nested `spent` array into a single number (for --human).
Firefly sends spent as a per-currency list of {sum: "-12.34", ...}; we sum
the sums. JSON output is untouched; this only feeds the table view."""
total = 0.0
for entry in budget_obj.get("spent") or []:
try:
total += float(entry.get("sum", 0))
except (TypeError, ValueError):
pass
return f"{total:.2f}"
def _list_args(p):
p.add_argument("--start", default=None, help="YYYY-MM-DD (default: 1st of this month)")
p.add_argument("--end", default=None, help="YYYY-MM-DD (default: last of this month)")
@registry.command("budget list", help="list budgets with spent for a period (default: current month)", args=_list_args)
def cmd_list(args, ctx):
first, last = _current_month()
params = {"start": args.start or first, "end": args.end or last}
resp = ctx.client.request("GET", "/api/v1/budgets", params=params)
rows = output.unwrap(resp)
if ctx.human and isinstance(rows, list):
# Replace the nested spent array with a scalar so the table shows it.
rows = [{**r, "spent": _spent_scalar(r)} for r in rows]
output.emit(rows, human=ctx.human)
return 0
def _create_args(p):
p.add_argument("name", help="budget name (must be unique)")
g = p.add_mutually_exclusive_group()
g.add_argument("--active", dest="active", action="store_true", default=True,
help="create active (default)")
g.add_argument("--inactive", dest="active", action="store_false",
help="create inactive")
p.add_argument("--currency", default=None, help="auto-budget currency code, e.g. EUR")
p.add_argument("--auto-budget-amount", dest="auto_budget_amount", default=None,
help="recurring auto-budget amount")
p.add_argument("--auto-budget-period", dest="auto_budget_period", default=None,
help="daily|weekly|monthly|quarterly|half_year|yearly")
p.add_argument("--auto-budget-type", dest="auto_budget_type", default=None,
help="reset|rollover|adjusted|none")
@registry.command("budget create", help="create a budget", args=_create_args)
def cmd_create(args, ctx):
body = {"name": args.name, "active": args.active}
if args.auto_budget_amount is not None:
body["auto_budget_amount"] = str(args.auto_budget_amount)
if args.auto_budget_period is not None:
body["auto_budget_period"] = args.auto_budget_period
if args.auto_budget_type is not None:
body["auto_budget_type"] = args.auto_budget_type
if args.currency:
body["currency_code"] = args.currency
resp = ctx.client.request("POST", "/api/v1/budgets", body=body)
output.emit(output.unwrap(resp), human=ctx.human)
return 0
|
