35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
from .base_logic import BaseLogic
|
||
|
|
from ..stardew_rule import StardewRule, And, Or, Has, Count
|
||
|
|
|
||
|
|
|
||
|
|
class HasLogicMixin(BaseLogic[None]):
|
||
|
|
# Should be cached
|
||
|
|
def has(self, item: str) -> StardewRule:
|
||
|
|
return Has(item, self.registry.item_rules)
|
||
|
|
|
||
|
|
def has_all(self, *items: str):
|
||
|
|
assert items, "Can't have all of no items."
|
||
|
|
|
||
|
|
return And(*(self.has(item) for item in items))
|
||
|
|
|
||
|
|
def has_any(self, *items: str):
|
||
|
|
assert items, "Can't have any of no items."
|
||
|
|
|
||
|
|
return Or(*(self.has(item) for item in items))
|
||
|
|
|
||
|
|
def has_n(self, *items: str, count: int):
|
||
|
|
return self.count(count, *(self.has(item) for item in items))
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def count(count: int, *rules: StardewRule) -> StardewRule:
|
||
|
|
assert rules, "Can't create a Count conditions without rules"
|
||
|
|
assert len(rules) >= count, "Count need at least as many rules as the count"
|
||
|
|
|
||
|
|
if count == 1:
|
||
|
|
return Or(*rules)
|
||
|
|
|
||
|
|
if count == len(rules):
|
||
|
|
return And(*rules)
|
||
|
|
|
||
|
|
return Count(list(rules), count)
|