102 lines
4.8 KiB
Python
102 lines
4.8 KiB
Python
from typing import Union, List
|
|
|
|
from Utils import cache_self1
|
|
from .base_logic import BaseLogicMixin, BaseLogic
|
|
from .received_logic import ReceivedLogicMixin
|
|
from .region_logic import RegionLogicMixin
|
|
from .season_logic import SeasonLogicMixin
|
|
from .skill_logic import SkillLogicMixin
|
|
from .tool_logic import ToolLogicMixin
|
|
from ..data import FishItem, fish_data
|
|
from ..locations import LocationTags, locations_by_tag
|
|
from ..options import ExcludeGingerIsland, Fishsanity
|
|
from ..options import SpecialOrderLocations
|
|
from ..stardew_rule import StardewRule, True_, False_, And
|
|
from ..strings.fish_names import SVEFish
|
|
from ..strings.quality_names import FishQuality
|
|
from ..strings.region_names import Region
|
|
from ..strings.skill_names import Skill
|
|
|
|
|
|
class FishingLogicMixin(BaseLogicMixin):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fishing = FishingLogic(*args, **kwargs)
|
|
|
|
|
|
class FishingLogic(BaseLogic[Union[FishingLogicMixin, ReceivedLogicMixin, RegionLogicMixin, SeasonLogicMixin, ToolLogicMixin, SkillLogicMixin]]):
|
|
def can_fish_in_freshwater(self) -> StardewRule:
|
|
return self.logic.skill.can_fish() & self.logic.region.can_reach_any((Region.forest, Region.town, Region.mountain))
|
|
|
|
def has_max_fishing(self) -> StardewRule:
|
|
skill_rule = self.logic.skill.has_level(Skill.fishing, 10)
|
|
return self.logic.tool.has_fishing_rod(4) & skill_rule
|
|
|
|
def can_fish_chests(self) -> StardewRule:
|
|
skill_rule = self.logic.skill.has_level(Skill.fishing, 6)
|
|
return self.logic.tool.has_fishing_rod(4) & skill_rule
|
|
|
|
def can_fish_at(self, region: str) -> StardewRule:
|
|
return self.logic.skill.can_fish() & self.logic.region.can_reach(region)
|
|
|
|
@cache_self1
|
|
def can_catch_fish(self, fish: FishItem) -> StardewRule:
|
|
quest_rule = True_()
|
|
if fish.extended_family:
|
|
quest_rule = self.logic.fishing.can_start_extended_family_quest()
|
|
region_rule = self.logic.region.can_reach_any(fish.locations)
|
|
season_rule = self.logic.season.has_any(fish.seasons)
|
|
if fish.difficulty == -1:
|
|
difficulty_rule = self.logic.skill.can_crab_pot
|
|
else:
|
|
difficulty_rule = self.logic.skill.can_fish(difficulty=(120 if fish.legendary else fish.difficulty))
|
|
if fish.name == SVEFish.kittyfish:
|
|
item_rule = self.logic.received("Kittyfish Spell")
|
|
else:
|
|
item_rule = True_()
|
|
return quest_rule & region_rule & season_rule & difficulty_rule & item_rule
|
|
|
|
def can_start_extended_family_quest(self) -> StardewRule:
|
|
if self.options.exclude_ginger_island == ExcludeGingerIsland.option_true:
|
|
return False_()
|
|
if self.options.special_order_locations != SpecialOrderLocations.option_board_qi:
|
|
return False_()
|
|
return self.logic.region.can_reach(Region.qi_walnut_room) & And(*(self.logic.fishing.can_catch_fish(fish) for fish in fish_data.legendary_fish))
|
|
|
|
def can_catch_quality_fish(self, fish_quality: str) -> StardewRule:
|
|
if fish_quality == FishQuality.basic:
|
|
return True_()
|
|
rod_rule = self.logic.tool.has_fishing_rod(2)
|
|
if fish_quality == FishQuality.silver:
|
|
return rod_rule
|
|
if fish_quality == FishQuality.gold:
|
|
return rod_rule & self.logic.skill.has_level(Skill.fishing, 4)
|
|
if fish_quality == FishQuality.iridium:
|
|
return rod_rule & self.logic.skill.has_level(Skill.fishing, 10)
|
|
|
|
raise ValueError(f"Quality {fish_quality} is unknown.")
|
|
|
|
def can_catch_every_fish(self) -> StardewRule:
|
|
rules = [self.has_max_fishing()]
|
|
exclude_island = self.options.exclude_ginger_island == ExcludeGingerIsland.option_true
|
|
exclude_extended_family = self.options.special_order_locations != SpecialOrderLocations.option_board_qi
|
|
for fish in fish_data.get_fish_for_mods(self.options.mods.value):
|
|
if exclude_island and fish in fish_data.island_fish:
|
|
continue
|
|
if exclude_extended_family and fish in fish_data.extended_family:
|
|
continue
|
|
rules.append(self.logic.fishing.can_catch_fish(fish))
|
|
return And(*rules)
|
|
|
|
def can_catch_every_fish_in_slot(self, all_location_names_in_slot: List[str]) -> StardewRule:
|
|
if self.options.fishsanity == Fishsanity.option_none:
|
|
return self.can_catch_every_fish()
|
|
|
|
rules = [self.has_max_fishing()]
|
|
|
|
for fishsanity_location in locations_by_tag[LocationTags.FISHSANITY]:
|
|
if fishsanity_location.name not in all_location_names_in_slot:
|
|
continue
|
|
rules.append(self.logic.region.can_reach_location(fishsanity_location.name))
|
|
return And(*rules)
|