27 lines
955 B
Python
27 lines
955 B
Python
from typing import Union
|
|
|
|
from .base_logic import BaseLogic, BaseLogicMixin
|
|
from .received_logic import ReceivedLogicMixin
|
|
from ..stardew_rule import True_
|
|
from ..strings.calendar_names import Weekday
|
|
|
|
|
|
class TravelingMerchantLogicMixin(BaseLogicMixin):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.traveling_merchant = TravelingMerchantLogic(*args, **kwargs)
|
|
|
|
|
|
class TravelingMerchantLogic(BaseLogic[Union[TravelingMerchantLogicMixin, ReceivedLogicMixin]]):
|
|
|
|
def has_days(self, number_days: int = 1):
|
|
if number_days <= 0:
|
|
return True_()
|
|
|
|
traveling_merchant_days = tuple(f"Traveling Merchant: {day}" for day in Weekday.all_days)
|
|
if number_days == 1:
|
|
return self.logic.received_any(*traveling_merchant_days)
|
|
|
|
tier = min(7, max(1, number_days))
|
|
return self.logic.received_n(*traveling_merchant_days, count=tier)
|