import datetime from typing import LiteralString, Set, List, Tuple from token_bot.history_manager.update_trigger import UpdateTrigger from token_bot.persistant_database import AlertType, Alert from token_bot.token_database.flavor import Flavor from token_bot.token_database.region import Region class History: def __init__(self, flavor: Flavor, region: Region): self._flavor : Flavor = flavor self._region : Region = region self._history : List[Tuple[datetime, int]] = [] self._last_price_movement : int = 0 self._latest_price_datum : Tuple[datetime.datetime, int] | None = None self._update_triggers : List[UpdateTrigger] = [] for alert_type in AlertType: self._update_triggers.append(UpdateTrigger(Alert(alert_type, flavor, self._region))) @property def flavor(self) -> Flavor: return self._flavor @property def region(self) -> Region: return self._region @property def last_price_datum(self) -> Tuple[datetime.datetime, int] | None: return self._latest_price_datum @property def history(self) -> List[Tuple[datetime.datetime, int]]: return self._history async def _process_update_triggers(self) -> List[Alert]: alerts = [] for trigger in self._update_triggers: if trigger.check_and_update(self._latest_price_datum, self._history): alerts.append(trigger.alert) return alerts async def add_price(self, datum: Tuple[datetime.datetime, int]) -> List[Alert]: if self._latest_price_datum is None: self._latest_price_datum = datum self._last_price_movement = datum[1] - self._latest_price_datum[1] self._latest_price_datum = datum self._history.append(datum) return await self._process_update_triggers()