55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""Scenario evaluation services including pricing integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Iterable
|
|
|
|
from models.scenario import Scenario
|
|
from services.pricing import (
|
|
PricingInput,
|
|
PricingMetadata,
|
|
PricingResult,
|
|
calculate_pricing,
|
|
)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class ScenarioPricingConfig:
|
|
"""Configuration for pricing evaluation within a scenario."""
|
|
|
|
metadata: PricingMetadata | None = None
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class ScenarioPricingSnapshot:
|
|
"""Captured pricing results for a scenario."""
|
|
|
|
scenario_id: int
|
|
results: list[PricingResult]
|
|
|
|
|
|
class ScenarioPricingEvaluator:
|
|
"""Evaluate scenario profitability inputs using pricing services."""
|
|
|
|
def __init__(self, config: ScenarioPricingConfig | None = None) -> None:
|
|
self._config = config or ScenarioPricingConfig()
|
|
|
|
def evaluate(
|
|
self,
|
|
scenario: Scenario,
|
|
*,
|
|
inputs: Iterable[PricingInput],
|
|
metadata_override: PricingMetadata | None = None,
|
|
) -> ScenarioPricingSnapshot:
|
|
metadata = metadata_override or self._config.metadata
|
|
results: list[PricingResult] = []
|
|
for pricing_input in inputs:
|
|
result = calculate_pricing(
|
|
pricing_input,
|
|
metadata=metadata,
|
|
currency=scenario.currency,
|
|
)
|
|
results.append(result)
|
|
return ScenarioPricingSnapshot(scenario_id=scenario.id, results=results)
|