from __future__ import annotations
from collections import defaultdict
from decimal import Decimal
from typing import Any, Optional
from .policy import Policy
from .units import NewUnit
from .utils import SupportsDecimal
from .value import FinancialValue as FV
[docs]
class CalculationService:
"""Utility class for calculation services."""
def __init__(self, policy: Policy | None = None):
self._policy = policy or Policy()
def _fv(self, x: SupportsDecimal, *, unit: Optional[NewUnit] = None) -> FV:
"""Create a FinancialValue with the service's policy and optional unit.
Args:
x: Value to wrap in FinancialValue
unit: Optional unit for the value
Returns:
FinancialValue with the service's policy and specified unit
"""
return FV(x, policy=self._policy, unit=unit)
def _d(self, v) -> Decimal | None:
"""Coerce anything into FV with policy, then to Decimal or None."""
fv = self._fv(v)
return None if fv.is_none() else fv.as_decimal()
def _fv_dict(self, x: SupportsDecimal) -> dict[Any, FV]:
return defaultdict(lambda: self._fv(x))