41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
from typing import Tuple, Optional
|
||
|
|
|
||
|
|
from .game_item import ItemSource, kw_only, Requirement
|
||
|
|
from ..strings.season_names import Season
|
||
|
|
|
||
|
|
ItemPrice = Tuple[int, str]
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True, **kw_only)
|
||
|
|
class ShopSource(ItemSource):
|
||
|
|
shop_region: str
|
||
|
|
money_price: Optional[int] = None
|
||
|
|
items_price: Optional[Tuple[ItemPrice, ...]] = None
|
||
|
|
seasons: Tuple[str, ...] = Season.all
|
||
|
|
other_requirements: Tuple[Requirement, ...] = ()
|
||
|
|
|
||
|
|
def __post_init__(self):
|
||
|
|
assert self.money_price or self.items_price, "At least money price or items price need to be defined."
|
||
|
|
assert self.items_price is None or all(type(p) == tuple for p in self.items_price), "Items price should be a tuple."
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True, **kw_only)
|
||
|
|
class MysteryBoxSource(ItemSource):
|
||
|
|
amount: int
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True, **kw_only)
|
||
|
|
class ArtifactTroveSource(ItemSource):
|
||
|
|
amount: int
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True, **kw_only)
|
||
|
|
class PrizeMachineSource(ItemSource):
|
||
|
|
amount: int
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True, **kw_only)
|
||
|
|
class FishingTreasureChestSource(ItemSource):
|
||
|
|
amount: int
|