import math import typing as t from datetime import datetime from functools import cached_property as functools_cached_property, partial from pydantic import BaseModel, ValidationError, WrapValidator, computed_field, field_validator, model_validator from module.base.decorator import cached_property, del_cached_property from module.config.stored.classes import now from module.config.utils import DEFAULT_TIME from module.exception import ScriptError from module.logger import logger from tasks.base.ui import UI from tasks.dungeon.keywords import DungeonList from tasks.planner.keywords import ITEM_TYPES from tasks.planner.keywords.classes import ItemBase class PlannerResultRow(BaseModel): """ A row of data from planner result page """ item: ITEM_TYPES total: int synthesize: int demand: int def __eq__(self, other): return self.item == other.item class ObtainedAmmount(BaseModel): """ A row of data from DungeonObtain detection """ item: ITEM_TYPES value: int def _fallback_to_default_validator( get_default: t.Callable[[], t.Any], v: t.Any, next_: t.Callable[[t.Any], t.Any], ) -> t.Any: try: return next_(v) except ValueError as e: logger.error(e) return get_default() class BaseModelWithFallback(BaseModel): """ Pydantic model that fallbacks to default on error https://github.com/pydantic/pydantic/discussions/8579 """ @classmethod def __pydantic_init_subclass__(cls, **kwargs: t.Any) -> None: for field in cls.model_fields.values(): if not field.is_required(): validator = WrapValidator(partial(_fallback_to_default_validator, field.get_default)) field.metadata.append(validator) cls.model_rebuild(force=True) class MultiValue(BaseModelWithFallback): purple: int = 0 blue: int = 0 green: int = 0 def add(self, other: "MultiValue"): self.green += other.green self.blue += other.blue self.purple += other.purple def __sub__(self, other): green = max(self.green - other.green, 0) blue = max(self.blue - other.blue, 0) purple = max(self.purple - other.purple, 0) return MultiValue(green=green, blue=blue, purple=purple) def equivalent_green(self): return self.green + self.blue * 3 + self.purple * 9 def clear(self): self.green = 0 self.blue = 0 self.purple = 0 SET_ROW_EXCLUDE = { 'drop_equivalent_green', 'combat_cost', 'progress_remain', 'progress_total', 'progress_current', } class InvalidPlannerRow(Exception): pass class StoredPlannerProxy(BaseModelWithFallback): item: ITEM_TYPES value: int | MultiValue = 0 total: int | MultiValue = 0 synthesize: int | MultiValue = 0 # progress: float = 0. # eta: float = 0. time: datetime = DEFAULT_TIME @field_validator('item', mode='before') def val_item(cls, v, info): if isinstance(v, str): v = ItemBase.find_name(v) return v @model_validator(mode='after') def val_value(self): if self.item.has_group_base: if not isinstance(self.value, MultiValue): logger.warning(f'Planner item {self.item} has_group_base ' f'but given value={self.value} is not a MultiValue') self.value = MultiValue() if not isinstance(self.total, MultiValue): logger.warning(f'Planner item {self.item} has_group_base ' f'but given total={self.total} is not a MultiValue') self.total = MultiValue() if not isinstance(self.synthesize, MultiValue): logger.warning(f'Planner item {self.item} has_group_base ' f'but given synthesize={self.synthesize} is not a MultiValue') self.synthesize = MultiValue() if self.total.equivalent_green() <= 0: raise InvalidPlannerRow(f'Planner item {self.item} has invalid total={self.total}, drop') else: if not isinstance(self.value, int): logger.warning(f'Planner item {self.item} has no group base ' f'but given value={self.value} is not an int') self.value = 0 if not isinstance(self.total, int): logger.warning(f'Planner item {self.item} has no group base ' f'but given total={self.total} is not an int') self.total = 0 if not isinstance(self.synthesize, int): logger.warning(f'Planner item {self.item} has no group base ' f'but given synthesize={self.synthesize} is not an int') self.synthesize = 0 if self.total <= 0: raise InvalidPlannerRow(f'Planner item {self.item} has invalid total={self.total}, drop') if self.synthesize != 0: logger.warning(f'Planner item {self.item} has no group base ' f'its synthesize={self.synthesize} should be 0') self.synthesize = 0 return self def clear(self): if self.item.has_group_base: self.value.clear() else: self.value = 0 def update_synthesize(self): if self.item.has_group_base: green = self.value.green - self.total.green blue = self.value.blue - self.total.blue purple = self.value.purple - self.total.purple syn_blue = 0 syn_purple = 0 if green >= 3 and blue < 0: syn = min(green // 3, -blue) syn_blue += syn green -= syn * 3 # blue += syn if blue >= 3 and purple < 0: syn = min(blue // 3, -purple) syn_purple += syn blue -= syn * 3 purple += syn if green >= 9 and purple < 0: syn = min(green // 9, -purple) syn_purple += syn syn_blue += syn * 3 green -= syn * 9 # purple += syn self.synthesize.green = 0 self.synthesize.blue = syn_blue self.synthesize.purple = syn_purple else: self.synthesize = 0 def revert_synthesize(self): if self.item.has_group_base: self.value.green += self.synthesize.blue * 3 if self.synthesize.blue > 0: self.value.green += self.synthesize.purple * 9 else: self.value.blue += self.synthesize.purple * 3 @computed_field(repr=False) @functools_cached_property def drop_equivalent_green(self) -> float: # Tracks_of_Destiny if self.item.dungeon is None: return 1 if self.item.dungeon.is_Calyx_Golden_Treasures: return 24000 if self.item.dungeon.is_Calyx_Golden_Memories: # purple, blue, green = 5, 1, 0 return 48 if self.item.dungeon.is_Calyx_Golden_Aether: # purple, blue, green = 1, 2, 2.5 return 17.5 if self.item.is_ItemAscension: return 3 if self.item.is_ItemTrace: # purple, blue, green = 0.155, 1, 1.25 return 5.645 if self.item.is_ItemWeekly: return 3 raise ScriptError(f'{self} has no drop_equivalent_green defined') @computed_field(repr=False) @functools_cached_property def combat_cost(self) -> int: # Tracks_of_Destiny if self.item.dungeon is None: return 30 if self.item.dungeon.is_Calyx_Golden: return 10 if self.item.is_ItemAscension: return 30 if self.item.is_ItemTrace: return 10 if self.item.is_ItemWeekly: return 30 raise ScriptError(f'{self} has no stamina_pre_combat defined') @computed_field(repr=False) @functools_cached_property def progress_remain(self) -> float: if self.item.has_group_base: remain = self.total - self.value - self.synthesize return remain.equivalent_green() else: remain = max(self.total - self.value, 0) return remain @computed_field(repr=False) @functools_cached_property def progress_total(self) -> float: if self.item.has_group_base: return self.total.equivalent_green() else: return self.total @computed_field(repr=False) @functools_cached_property def progress_current(self) -> float: if self.item.has_group_base: current = self.progress_total - self.progress_remain current = min(max(current, 0), self.progress_total) return current else: current = self.value current = min(max(current, 0), self.total) return current @computed_field @functools_cached_property def progress(self) -> float: try: progress = self.progress_current / self.progress_total * 100 return round(min(max(progress, 0), 100), 2) except ZeroDivisionError: return 100. def is_approaching_total(self, wave_done: int = 0): """ Args: wave_done: Returns: bool: True if the future value may >= total after next combat """ wave_done = max(wave_done, 0) # Items with a static drop rate will have `AVG * (wave_done + 1) remain = self.progress_remain cost = self.combat_cost drop = self.drop_equivalent_green if cost == 10: return remain <= drop * (wave_done + 12) else: return remain <= drop * (wave_done + 1) @computed_field @functools_cached_property def eta(self) -> float: """ Estimate remaining days to farm """ if not self.need_farm(): return 0. if self.item.dungeon is None: return 0. remain = self.progress_remain cost = self.combat_cost drop = self.drop_equivalent_green if self.item.is_ItemWeekly: weeks = math.ceil(remain / drop / 3) return weeks * 7 else: stamina = math.ceil(remain / drop) * cost return round(stamina / 240, 1) def update(self, time=False): for attr in SET_ROW_EXCLUDE: del_cached_property(self, attr) del_cached_property(self, 'progress') del_cached_property(self, 'eta') self.update_synthesize() _ = self.progress _ = self.eta if time: self.time = now() def load_value_total(self, item: ItemBase, value=None, total=None, synthesize=None): """ Update data from PlannerResultRow to self """ if self.item.has_group_base: if item.group_base != self.item: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but they are different items') else: if item != self.item: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but they are different items') if self.item.has_group_base: if not self.item.is_rarity_purple: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but self is not in rarity purple') if item.is_rarity_green: if value is not None: self.value.green = value if total is not None: self.total.green = total # Cannot synthesize green # if synthesize is not None: # self.synthesize.green = synthesize self.synthesize.green = 0 elif item.is_rarity_blue: if value is not None: self.value.blue = value if total is not None: self.total.blue = total if synthesize is not None: self.synthesize.blue = synthesize elif item.is_rarity_purple: if value is not None: self.value.purple = value if total is not None: self.total.purple = total if synthesize is not None: self.synthesize.purple = synthesize else: raise ScriptError( f'load_value_total: Trying to load {item} in to {self} but item is in invalid rarity') else: # Cannot synthesize if item doesn't have multiple rarity self.synthesize = 0 if value is not None: self.value = value if total is not None: self.total = total def add_planner_result(self, row: "StoredPlannerProxy"): """ Add data from another StoredPlannerProxy to self """ item = row.item if self.item.has_group_base: if item.group_base != self.item: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but they are different items') else: if item != self.item: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but they are different items') if self.item.has_group_base: if not self.item.is_rarity_purple: raise ScriptError( f'load_value_total: Trying to load {item} into {self} but self is not in rarity purple') # Add `total` only # `synthesize` will be updated later # `value` remains unchanged since you still having that many items self.total.add(row.total) else: self.value += row.value self.total += row.total self.synthesize += row.synthesize def need_farm(self): return self.progress < 100 def need_synthesize(self): if self.item.has_group_base: return self.synthesize.green > 0 or self.synthesize.blue > 0 or self.synthesize.purple > 0 else: return False def load_planner_result(self, row: PlannerResultRow): """ Update data from PlannerResultRow to self """ # Approximate value, accurate value can be update in DungeonObtain value = row.total - row.synthesize - row.demand self.load_value_total(item=row.item, value=value, total=row.total, synthesize=row.synthesize) def load_item_amount(self, row: ObtainedAmmount): """ Update data from ObtainedAmmount to self """ value = row.value self.load_value_total(item=row.item, value=value) class PlannerProgressParser: def __init__(self): self.rows: dict[str, StoredPlannerProxy] = {} def from_planner_results(self, results: list[PlannerResultRow]): self.rows = {} # Create objects of base items first # them load value and total for row in results: base = row.item.group_base if base.name not in self.rows: try: if row.item.has_group_base: obj = StoredPlannerProxy( item=base, value=MultiValue(), total=MultiValue(), synthesize=MultiValue()) else: obj = StoredPlannerProxy(item=base, value=0, total=0, synthesize=0) except ScriptError as e: logger.error(e) continue self.rows[base.name] = obj else: obj = self.rows[base.name] obj.load_planner_result(row) rows = {} for name, row in self.rows.items(): row.revert_synthesize() row.update() if row.need_farm() or row.need_synthesize(): rows[name] = row self.rows = rows return self def load_obtained_amount(self, results: list[ObtainedAmmount]): for row in results: base = row.item.group_base try: obj = self.rows[base.name] except KeyError: logger.warning( f'load_obtained_amount() drops {row} because no need to farm') continue obj.load_item_amount(row) obj.update() return self def from_config(self, data): self.rows = {} for name, row in data.items(): if not row: continue if name == 'PlannerOverall': continue try: row = StoredPlannerProxy(**row) except (ScriptError, ValidationError, InvalidPlannerRow) as e: logger.error(e) continue if not row.item.is_group_base: logger.error(f'from_config: item is not group base {row}') continue row.update(time=False) self.rows[row.item.name] = row return self def add_planner_result(self, planner: "PlannerProgressParser"): """ Add another planner result to self """ for name, row in planner.rows.items(): if name in self.rows: self_row = self.rows[name] self_row.add_planner_result(row) else: self.rows[name] = row for row in self.rows.values(): row.update(time=True) def to_config(self) -> dict: data = {} for row in self.rows.values(): name = f'Item_{row.item.name}' dic = row.model_dump(exclude=SET_ROW_EXCLUDE) dic['item'] = row.item.name data[name] = dic return data def get_overall(self): """ Calculate overall progress Note that this method will clear all values Returns: float: Progress percentage float: ETA in days """ eta = 0. progress_current = 0. progress_total = 0. for row in self.rows.values(): eta += row.eta progress_current += row.progress_current progress_total += row.progress_total try: progress = round(progress_current / progress_total * 100, 2) except ZeroDivisionError: progress = 100. return progress, eta def iter_row_to_farm(self, need_farm=True) -> t.Iterable[StoredPlannerProxy]: """ Args: need_farm: True if filter rows that need farm Yields: """ if need_farm: rows = [row for row in self.rows.values() if row.need_farm()] else: rows = self.rows.values() for row in rows: if row.item.is_ItemWeekly: yield row for row in rows: if row.item.is_ItemAscension: yield row for row in rows: if row.item.is_ItemTrace: yield row for row in rows: if row.item.is_ItemExp: yield row for row in rows: if row.item.is_ItemCurrency: yield row def get_dungeon(self, double_calyx=False) -> DungeonList | None: """ Get dungeon to farm, or None if planner finished or the remaining items cannot be farmed """ for row in self.iter_row_to_farm(): item = row.item if item.is_ItemWeekly: continue dungeon = item.dungeon if dungeon is None: logger.error(f'Item {item} has nowhere to be farmed') continue if double_calyx: if dungeon.is_Calyx: logger.info(f'Planner farm (double_calyx): {dungeon}') return dungeon else: logger.info(f'Planner farm: {dungeon}') return dungeon logger.info('Planner farm empty') return None def get_weekly(self) -> DungeonList | None: for row in self.iter_row_to_farm(): item = row.item if not item.is_ItemWeekly: continue dungeon = item.dungeon if dungeon is None: logger.error(f'Item {item} has nowhere to be farmed') continue logger.info(f'Planner weekly farm: {dungeon}') return dungeon logger.info('Planner weekly farm empty') return None def row_come_from_dungeon(self, dungeon: DungeonList | None) -> StoredPlannerProxy | None: """ If any items in planner is able to be farmed from given dungeon """ if dungeon is None: return None for row in self.iter_row_to_farm(need_farm=False): if row.item.dungeon == dungeon: logger.info(f'Planner {row} come from {dungeon}') return row return None class PlannerMixin(UI): def planner_write_results(self, results: list[PlannerResultRow]): """ Write planner detection results info user config """ add = self.config.PlannerScan_ResultAdd logger.attr('ResultAdd', add) planner = PlannerProgressParser().from_planner_results(results) if add: planner.add_planner_result(self.planner) self.planner_write(planner) @cached_property def planner(self) -> PlannerProgressParser: logger.hr('Planner') data = self.config.cross_get('Dungeon.Planner', default={}) model = PlannerProgressParser() model.from_config(data) for row in model.rows.values(): logger.info(row) return model def planner_write(self, planner=None): """ Write planner into user config, delete planner object """ if planner is None: planner = self.planner data = planner.to_config() progress, eta = planner.get_overall() with self.config.multi_set(): # Set value for key, value in data.items(): self.config.cross_set(f'Dungeon.Planner.{key}', value) # Remove other value remove = [] for key, value in self.config.cross_get('Dungeon.Planner', default={}).items(): if value != {} and key not in data: remove.append(key) for key in remove: self.config.cross_set(f'Dungeon.Planner.{key}', {}) # print(progress, eta) # Set overall self.config.stored.PlannerOverall.value = f'{progress:.2f}%' self.config.stored.PlannerOverall.comment = f'<{eta:.1f}d' del_cached_property(self, 'planner') if __name__ == '__main__': self = PlannerMixin('src') self.planner_write(self.planner)