34e7748f23
* make sure number of month is actually a int * improve rule explain like in pr * remove redundant if in can_complete_bundle * assert number is int so cache is not bloated
85 lines
4.1 KiB
Python
85 lines
4.1 KiB
Python
from functools import cached_property
|
|
from typing import Union, List
|
|
|
|
from .base_logic import BaseLogicMixin, BaseLogic
|
|
from .fishing_logic import FishingLogicMixin
|
|
from .has_logic import HasLogicMixin
|
|
from .money_logic import MoneyLogicMixin
|
|
from .quality_logic import QualityLogicMixin
|
|
from .quest_logic import QuestLogicMixin
|
|
from .received_logic import ReceivedLogicMixin
|
|
from .region_logic import RegionLogicMixin
|
|
from .skill_logic import SkillLogicMixin
|
|
from .time_logic import TimeLogicMixin
|
|
from ..bundles.bundle import Bundle
|
|
from ..stardew_rule import StardewRule, True_
|
|
from ..strings.ap_names.community_upgrade_names import CommunityUpgrade
|
|
from ..strings.currency_names import Currency
|
|
from ..strings.machine_names import Machine
|
|
from ..strings.quality_names import CropQuality, ForageQuality, FishQuality, ArtisanQuality
|
|
from ..strings.quest_names import Quest
|
|
from ..strings.region_names import Region
|
|
|
|
|
|
class BundleLogicMixin(BaseLogicMixin):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.bundle = BundleLogic(*args, **kwargs)
|
|
|
|
|
|
class BundleLogic(BaseLogic[Union[ReceivedLogicMixin, HasLogicMixin, TimeLogicMixin, RegionLogicMixin, MoneyLogicMixin, QualityLogicMixin, FishingLogicMixin,
|
|
SkillLogicMixin, QuestLogicMixin]]):
|
|
# Should be cached
|
|
def can_complete_bundle(self, bundle: Bundle) -> StardewRule:
|
|
item_rules = []
|
|
qualities = []
|
|
time_to_grind = 0
|
|
can_speak_junimo = self.logic.region.can_reach(Region.wizard_tower)
|
|
for bundle_item in bundle.items:
|
|
if Currency.is_currency(bundle_item.get_item()):
|
|
return can_speak_junimo & self.logic.money.can_trade(bundle_item.get_item(), bundle_item.amount)
|
|
|
|
item_rules.append(bundle_item.get_item())
|
|
if bundle_item.amount > 50:
|
|
time_to_grind = bundle_item.amount // 50
|
|
qualities.append(bundle_item.quality)
|
|
quality_rules = self.get_quality_rules(qualities)
|
|
item_rules = self.logic.has_n(*item_rules, count=bundle.number_required)
|
|
time_rule = self.logic.time.has_lived_months(time_to_grind)
|
|
return can_speak_junimo & item_rules & quality_rules & time_rule
|
|
|
|
def get_quality_rules(self, qualities: List[str]) -> StardewRule:
|
|
crop_quality = CropQuality.get_highest(qualities)
|
|
fish_quality = FishQuality.get_highest(qualities)
|
|
forage_quality = ForageQuality.get_highest(qualities)
|
|
artisan_quality = ArtisanQuality.get_highest(qualities)
|
|
quality_rules = []
|
|
if crop_quality != CropQuality.basic:
|
|
quality_rules.append(self.logic.quality.can_grow_crop_quality(crop_quality))
|
|
if fish_quality != FishQuality.basic:
|
|
quality_rules.append(self.logic.fishing.can_catch_quality_fish(fish_quality))
|
|
if forage_quality != ForageQuality.basic:
|
|
quality_rules.append(self.logic.skill.can_forage_quality(forage_quality))
|
|
if artisan_quality != ArtisanQuality.basic:
|
|
quality_rules.append(self.logic.has(Machine.cask))
|
|
if not quality_rules:
|
|
return True_()
|
|
return self.logic.and_(*quality_rules)
|
|
|
|
@cached_property
|
|
def can_complete_community_center(self) -> StardewRule:
|
|
return (self.logic.region.can_reach_location("Complete Crafts Room") &
|
|
self.logic.region.can_reach_location("Complete Pantry") &
|
|
self.logic.region.can_reach_location("Complete Fish Tank") &
|
|
self.logic.region.can_reach_location("Complete Bulletin Board") &
|
|
self.logic.region.can_reach_location("Complete Vault") &
|
|
self.logic.region.can_reach_location("Complete Boiler Room"))
|
|
|
|
def can_access_raccoon_bundles(self) -> StardewRule:
|
|
if self.options.quest_locations < 0:
|
|
return self.logic.received(CommunityUpgrade.raccoon, 1) & self.logic.quest.can_complete_quest(Quest.giant_stump)
|
|
|
|
# 1 - Break the tree
|
|
# 2 - Build the house, which summons the bundle racoon. This one is done manually if quests are turned off
|
|
return self.logic.received(CommunityUpgrade.raccoon, 2)
|