60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
from typing import Union
|
|
|
|
from .base_logic import BaseLogicMixin, BaseLogic
|
|
from .building_logic import BuildingLogicMixin
|
|
from .has_logic import HasLogicMixin
|
|
from .money_logic import MoneyLogicMixin
|
|
from ..stardew_rule import StardewRule, true_
|
|
from ..strings.animal_names import Animal, coop_animals, barn_animals
|
|
from ..strings.building_names import Building
|
|
from ..strings.forageable_names import Forageable
|
|
from ..strings.generic_names import Generic
|
|
from ..strings.region_names import Region
|
|
|
|
cost_and_building_by_animal = {
|
|
Animal.chicken: (800, Building.coop),
|
|
Animal.cow: (1500, Building.barn),
|
|
Animal.goat: (4000, Building.big_barn),
|
|
Animal.duck: (1200, Building.big_coop),
|
|
Animal.sheep: (8000, Building.deluxe_barn),
|
|
Animal.rabbit: (8000, Building.deluxe_coop),
|
|
Animal.pig: (16000, Building.deluxe_barn)
|
|
}
|
|
|
|
|
|
class AnimalLogicMixin(BaseLogicMixin):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.animal = AnimalLogic(*args, **kwargs)
|
|
|
|
|
|
class AnimalLogic(BaseLogic[Union[HasLogicMixin, MoneyLogicMixin, BuildingLogicMixin]]):
|
|
|
|
def can_buy_animal(self, animal: str) -> StardewRule:
|
|
try:
|
|
price, building = cost_and_building_by_animal[animal]
|
|
except KeyError:
|
|
return true_
|
|
return self.logic.money.can_spend_at(Region.ranch, price) & self.logic.building.has_building(building)
|
|
|
|
def has_animal(self, animal: str) -> StardewRule:
|
|
if animal == Generic.any:
|
|
return self.has_any_animal()
|
|
elif animal == Building.coop:
|
|
return self.has_any_coop_animal()
|
|
elif animal == Building.barn:
|
|
return self.has_any_barn_animal()
|
|
return self.logic.has(animal)
|
|
|
|
def has_happy_animal(self, animal: str) -> StardewRule:
|
|
return self.has_animal(animal) & self.logic.has(Forageable.hay)
|
|
|
|
def has_any_animal(self) -> StardewRule:
|
|
return self.has_any_coop_animal() | self.has_any_barn_animal()
|
|
|
|
def has_any_coop_animal(self) -> StardewRule:
|
|
return self.logic.has_any(*coop_animals)
|
|
|
|
def has_any_barn_animal(self) -> StardewRule:
|
|
return self.logic.has_any(*barn_animals)
|