86 lines
3.8 KiB
Python
86 lines
3.8 KiB
Python
from typing import Union
|
|
|
|
from Utils import cache_self1
|
|
from .action_logic import ActionLogicMixin
|
|
from .base_logic import BaseLogic, BaseLogicMixin
|
|
from .has_logic import HasLogicMixin
|
|
from .received_logic import ReceivedLogicMixin
|
|
from .region_logic import RegionLogicMixin
|
|
from .time_logic import TimeLogicMixin
|
|
from .tool_logic import ToolLogicMixin
|
|
from .. import options
|
|
from ..data.museum_data import MuseumItem, all_museum_items, all_museum_artifacts, all_museum_minerals
|
|
from ..stardew_rule import StardewRule, False_
|
|
from ..strings.metal_names import Mineral
|
|
from ..strings.region_names import Region
|
|
from ..strings.tool_names import Tool, ToolMaterial
|
|
|
|
|
|
class MuseumLogicMixin(BaseLogicMixin):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.museum = MuseumLogic(*args, **kwargs)
|
|
|
|
|
|
class MuseumLogic(BaseLogic[Union[ReceivedLogicMixin, HasLogicMixin, TimeLogicMixin, RegionLogicMixin, ActionLogicMixin, ToolLogicMixin, MuseumLogicMixin]]):
|
|
|
|
def can_donate_museum_items(self, number: int) -> StardewRule:
|
|
return self.logic.region.can_reach(Region.museum) & self.logic.museum.can_find_museum_items(number)
|
|
|
|
def can_donate_museum_artifacts(self, number: int) -> StardewRule:
|
|
return self.logic.region.can_reach(Region.museum) & self.logic.museum.can_find_museum_artifacts(number)
|
|
|
|
@cache_self1
|
|
def can_find_museum_item(self, item: MuseumItem) -> StardewRule:
|
|
if item.locations:
|
|
region_rule = self.logic.region.can_reach_all_except_one(item.locations)
|
|
else:
|
|
region_rule = False_()
|
|
if item.geodes:
|
|
geodes_rule = self.logic.and_(*(self.logic.action.can_open_geode(geode) for geode in item.geodes))
|
|
else:
|
|
geodes_rule = False_()
|
|
# monster_rule = self.can_farm_monster(item.monsters)
|
|
time_needed_to_grind = (20 - item.difficulty) / 2
|
|
time_rule = self.logic.time.has_lived_months(time_needed_to_grind)
|
|
pan_rule = False_()
|
|
if item.item_name == Mineral.earth_crystal or item.item_name == Mineral.fire_quartz or item.item_name == Mineral.frozen_tear:
|
|
pan_rule = self.logic.tool.has_tool(Tool.pan, ToolMaterial.iridium)
|
|
return (pan_rule | region_rule | geodes_rule) & time_rule # & monster_rule & extra_rule
|
|
|
|
def can_find_museum_artifacts(self, number: int) -> StardewRule:
|
|
rules = []
|
|
for artifact in all_museum_artifacts:
|
|
rules.append(self.logic.museum.can_find_museum_item(artifact))
|
|
|
|
return self.logic.count(number, *rules)
|
|
|
|
def can_find_museum_minerals(self, number: int) -> StardewRule:
|
|
rules = []
|
|
for mineral in all_museum_minerals:
|
|
rules.append(self.logic.museum.can_find_museum_item(mineral))
|
|
|
|
return self.logic.count(number, *rules)
|
|
|
|
def can_find_museum_items(self, number: int) -> StardewRule:
|
|
rules = []
|
|
for donation in all_museum_items:
|
|
rules.append(self.logic.museum.can_find_museum_item(donation))
|
|
|
|
return self.logic.count(number, *rules)
|
|
|
|
def can_complete_museum(self) -> StardewRule:
|
|
rules = [self.logic.region.can_reach(Region.museum)]
|
|
|
|
if self.options.museumsanity == options.Museumsanity.option_none:
|
|
rules.append(self.logic.received("Traveling Merchant Metal Detector", 2))
|
|
else:
|
|
rules.append(self.logic.received("Traveling Merchant Metal Detector", 3))
|
|
|
|
for donation in all_museum_items:
|
|
rules.append(self.logic.museum.can_find_museum_item(donation))
|
|
return self.logic.and_(*rules) & self.logic.region.can_reach(Region.museum)
|
|
|
|
def can_donate(self, item: str) -> StardewRule:
|
|
return self.logic.has(item) & self.logic.region.can_reach(Region.museum)
|