feat: Szenario-Engine (Projektion, Modifikatoren, Zahlungsstrom)
Implementiert Task 5 mit TDD-Ansatz: projection.py mit Projection-Klasse und project()-Funktion für Cashflow-Simulation; scenario.py mit PlainRecurring, PlainPlanned, PlainModifier Dataclasses und build_cashflows() für Szenarien-Zahlungsströme mit Modifier-Support (remove, percent, absolute). Alle 15 Tests grün. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
39
finance/app/engine/projection.py
Normal file
39
finance/app/engine/projection.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from collections import defaultdict
|
||||
from dataclasses import dataclass
|
||||
from datetime import date, timedelta
|
||||
from decimal import Decimal
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class Projection:
|
||||
series: list[tuple[date, Decimal]]
|
||||
low_point: tuple[date, Decimal]
|
||||
first_below_zero: date | None
|
||||
first_below_threshold: date | None
|
||||
|
||||
|
||||
def project(start_balance: Decimal, start_date: date,
|
||||
cashflows: list[tuple[date, Decimal]], horizon_days: int,
|
||||
threshold: Decimal = Decimal("0")) -> Projection:
|
||||
end = start_date + timedelta(days=horizon_days)
|
||||
by_day: dict[date, Decimal] = defaultdict(lambda: Decimal("0"))
|
||||
for d, amount in cashflows:
|
||||
if start_date < d <= end:
|
||||
by_day[d] += amount
|
||||
series: list[tuple[date, Decimal]] = []
|
||||
balance = start_balance
|
||||
low = (start_date, start_balance)
|
||||
below0: date | None = None
|
||||
below_t: date | None = None
|
||||
d = start_date + timedelta(days=1)
|
||||
while d <= end:
|
||||
balance += by_day.get(d, Decimal("0"))
|
||||
series.append((d, balance))
|
||||
if balance < low[1]:
|
||||
low = (d, balance)
|
||||
if below0 is None and balance < 0:
|
||||
below0 = d
|
||||
if below_t is None and balance < threshold:
|
||||
below_t = d
|
||||
d += timedelta(days=1)
|
||||
return Projection(series, low, below0, below_t)
|
||||
82
finance/app/engine/scenario.py
Normal file
82
finance/app/engine/scenario.py
Normal file
@@ -0,0 +1,82 @@
|
||||
from dataclasses import dataclass
|
||||
from datetime import date
|
||||
from decimal import ROUND_HALF_UP, Decimal
|
||||
|
||||
from app.engine.loans import Installment
|
||||
from app.engine.recurrence import occurrences
|
||||
|
||||
CENT = Decimal("0.01")
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class PlainRecurring:
|
||||
id: int
|
||||
name: str
|
||||
amount: Decimal
|
||||
rhythm: str
|
||||
due_day: int
|
||||
start_date: date | None
|
||||
end_date: date | None
|
||||
category_id: int | None
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class PlainPlanned:
|
||||
name: str
|
||||
amount: Decimal
|
||||
due: date
|
||||
category_id: int | None
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class PlainModifier:
|
||||
target_type: str
|
||||
target_id: int
|
||||
kind: str
|
||||
value: Decimal
|
||||
|
||||
|
||||
def _modified(amount: Decimal, category_id: int | None, recurring_id: int | None,
|
||||
modifiers: list[PlainModifier]) -> Decimal | None:
|
||||
for m in modifiers:
|
||||
hit = ((m.target_type == "category" and category_id == m.target_id)
|
||||
or (m.target_type == "recurring" and recurring_id == m.target_id))
|
||||
if not hit:
|
||||
continue
|
||||
if m.kind == "remove":
|
||||
return None
|
||||
if m.kind == "percent":
|
||||
amount = (amount * (Decimal(100) - m.value) / Decimal(100)).quantize(CENT, ROUND_HALF_UP)
|
||||
elif m.kind == "absolute":
|
||||
if amount < 0:
|
||||
amount = min(Decimal("0"), amount + m.value)
|
||||
else:
|
||||
amount = max(Decimal("0"), amount - m.value)
|
||||
return amount
|
||||
|
||||
|
||||
def build_cashflows(recurring: list[PlainRecurring], planned: list[PlainPlanned],
|
||||
loan_schedules: list[list[Installment]],
|
||||
loan_payouts: list[tuple[date, Decimal]],
|
||||
modifiers: list[PlainModifier],
|
||||
window_start: date, window_end: date) -> list[tuple[date, Decimal]]:
|
||||
flows: list[tuple[date, Decimal]] = []
|
||||
for r in recurring:
|
||||
amount = _modified(r.amount, r.category_id, r.id, modifiers)
|
||||
if amount is None:
|
||||
continue
|
||||
for d in occurrences(r.rhythm, r.due_day, window_start, window_end,
|
||||
r.start_date, r.end_date):
|
||||
flows.append((d, amount))
|
||||
for p in planned:
|
||||
amount = _modified(p.amount, p.category_id, None, modifiers)
|
||||
if amount is not None and window_start <= p.due <= window_end:
|
||||
flows.append((p.due, amount))
|
||||
for d, payout in loan_payouts:
|
||||
if window_start <= d <= window_end:
|
||||
flows.append((d, payout))
|
||||
for sched in loan_schedules:
|
||||
for inst in sched:
|
||||
if window_start <= inst.due <= window_end:
|
||||
flows.append((inst.due, inst.payment))
|
||||
return sorted(flows)
|
||||
Reference in New Issue
Block a user