refactor: add rule engine, risk service, quote state machine, and replay tests
This commit is contained in:
47
core/quote_state_machine.py
Normal file
47
core/quote_state_machine.py
Normal file
@@ -0,0 +1,47 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from typing import Protocol
|
||||
|
||||
|
||||
class QuoteStateLike(Protocol):
|
||||
pending_image_urls: list
|
||||
pending_requirements: list
|
||||
quote_phase: str
|
||||
quote_ready_turns: int
|
||||
|
||||
|
||||
@dataclass
|
||||
class QuoteStateMachine:
|
||||
delay_turns: int = 1
|
||||
|
||||
def refresh(self, state: QuoteStateLike, phase_hint: str = "") -> None:
|
||||
if phase_hint in {"idle", "collecting", "ready_to_quote", "waiting_result"}:
|
||||
state.quote_phase = phase_hint
|
||||
if phase_hint == "idle":
|
||||
state.quote_ready_turns = 0
|
||||
return
|
||||
|
||||
if not state.pending_image_urls:
|
||||
state.quote_phase = "idle"
|
||||
state.quote_ready_turns = 0
|
||||
return
|
||||
|
||||
if state.quote_phase in {"ready_to_quote", "waiting_result"}:
|
||||
return
|
||||
|
||||
state.quote_phase = "collecting"
|
||||
|
||||
def mark_ready(self, state: QuoteStateLike) -> None:
|
||||
if state.quote_phase != "ready_to_quote":
|
||||
state.quote_phase = "ready_to_quote"
|
||||
state.quote_ready_turns = max(0, int(self.delay_turns))
|
||||
|
||||
def should_defer_batch_quote(self, state: QuoteStateLike, mark_ready: bool = False) -> bool:
|
||||
if mark_ready and state.quote_phase != "ready_to_quote":
|
||||
self.mark_ready(state)
|
||||
|
||||
if state.quote_phase == "ready_to_quote" and state.quote_ready_turns > 0:
|
||||
state.quote_ready_turns -= 1
|
||||
return True
|
||||
return False
|
||||
Reference in New Issue
Block a user