49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from dataclasses import dataclass
|
|
from typing import List
|
|
|
|
from .. import data
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class SeedItem:
|
|
name: str
|
|
seasons: List[str]
|
|
regions: List[str]
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CropItem:
|
|
name: str
|
|
farm_growth_seasons: List[str]
|
|
seed: SeedItem
|
|
|
|
|
|
def load_crop_csv():
|
|
import csv
|
|
try:
|
|
from importlib.resources import files
|
|
except ImportError:
|
|
from importlib_resources import files # noqa
|
|
|
|
with files(data).joinpath("crops.csv").open() as file:
|
|
reader = csv.DictReader(file)
|
|
crops = []
|
|
seeds = []
|
|
|
|
for item in reader:
|
|
seeds.append(SeedItem(item["seed"],
|
|
[season for season in item["seed_seasons"].split(",")]
|
|
if item["seed_seasons"] else [],
|
|
[region for region in item["seed_regions"].split(",")]
|
|
if item["seed_regions"] else []))
|
|
crops.append(CropItem(item["crop"],
|
|
[season for season in item["farm_growth_seasons"].split(",")]
|
|
if item["farm_growth_seasons"] else [],
|
|
seeds[-1]))
|
|
return crops, seeds
|
|
|
|
|
|
# TODO Those two should probably be split to we can include rest of seeds
|
|
all_crops, all_purchasable_seeds = load_crop_csv()
|
|
crops_by_name = {crop.name: crop for crop in all_crops}
|