192 lines
12 KiB
Python
192 lines
12 KiB
Python
from dataclasses import dataclass
|
|
from typing import List, Tuple, Union, Optional, Set
|
|
|
|
from . import season_data as season
|
|
from ..strings.fish_names import Fish, SVEFish, DistantLandsFish
|
|
from ..strings.region_names import Region, SVERegion
|
|
from ..mods.mod_data import ModNames
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class FishItem:
|
|
name: str
|
|
locations: Tuple[str]
|
|
seasons: Tuple[str]
|
|
difficulty: int
|
|
legendary: bool
|
|
extended_family: bool
|
|
mod_name: Optional[str] = None
|
|
|
|
def __repr__(self):
|
|
return f"{self.name} (Locations: {self.locations} |" \
|
|
f" Seasons: {self.seasons} |" \
|
|
f" Difficulty: {self.difficulty}) |" \
|
|
f"Mod: {self.mod_name}"
|
|
|
|
|
|
fresh_water = (Region.farm, Region.forest, Region.town, Region.mountain)
|
|
ocean = (Region.beach,)
|
|
town_river = (Region.town,)
|
|
mountain_lake = (Region.mountain,)
|
|
forest_pond = (Region.forest,)
|
|
forest_river = (Region.forest,)
|
|
secret_woods = (Region.secret_woods,)
|
|
mines_floor_20 = (Region.mines_floor_20,)
|
|
mines_floor_60 = (Region.mines_floor_60,)
|
|
mines_floor_100 = (Region.mines_floor_100,)
|
|
sewers = (Region.sewer,)
|
|
desert = (Region.desert,)
|
|
mutant_bug_lair = (Region.mutant_bug_lair,)
|
|
witch_swamp = (Region.witch_swamp,)
|
|
night_market = (Region.beach,)
|
|
ginger_island_ocean = (Region.island_south, Region.island_west)
|
|
ginger_island_river = (Region.island_west,)
|
|
pirate_cove = (Region.pirate_cove,)
|
|
|
|
crimson_badlands = (SVERegion.crimson_badlands,)
|
|
shearwater = (SVERegion.shearwater,)
|
|
highlands = (SVERegion.highlands_outside,)
|
|
sprite_spring = (SVERegion.sprite_spring,)
|
|
fable_reef = (SVERegion.fable_reef,)
|
|
vineyard = (SVERegion.blue_moon_vineyard,)
|
|
|
|
all_fish: List[FishItem] = []
|
|
|
|
|
|
def create_fish(name: str, locations: Tuple[str, ...], seasons: Union[str, Tuple[str, ...]],
|
|
difficulty: int, legendary: bool = False, extended_family: bool = False, mod_name: Optional[str] = None) -> FishItem:
|
|
if isinstance(seasons, str):
|
|
seasons = (seasons,)
|
|
|
|
fish_item = FishItem(name, locations, seasons, difficulty, legendary, extended_family, mod_name)
|
|
all_fish.append(fish_item)
|
|
return fish_item
|
|
|
|
|
|
albacore = create_fish("Albacore", ocean, (season.fall, season.winter), 60)
|
|
anchovy = create_fish("Anchovy", ocean, (season.spring, season.fall), 30)
|
|
blue_discus = create_fish("Blue Discus", ginger_island_river, season.all_seasons, 60)
|
|
bream = create_fish("Bream", town_river + forest_river, season.all_seasons, 35)
|
|
bullhead = create_fish("Bullhead", mountain_lake, season.all_seasons, 46)
|
|
carp = create_fish(Fish.carp, mountain_lake + secret_woods + sewers + mutant_bug_lair, season.not_winter, 15)
|
|
catfish = create_fish("Catfish", town_river + forest_river + secret_woods, (season.spring, season.fall), 75)
|
|
chub = create_fish("Chub", forest_river + mountain_lake, season.all_seasons, 35)
|
|
dorado = create_fish("Dorado", forest_river, season.summer, 78)
|
|
eel = create_fish("Eel", ocean, (season.spring, season.fall), 70)
|
|
flounder = create_fish("Flounder", ocean, (season.spring, season.summer), 50)
|
|
ghostfish = create_fish("Ghostfish", mines_floor_20 + mines_floor_60, season.all_seasons, 50)
|
|
halibut = create_fish("Halibut", ocean, season.not_fall, 50)
|
|
herring = create_fish("Herring", ocean, (season.spring, season.winter), 25)
|
|
ice_pip = create_fish("Ice Pip", mines_floor_60, season.all_seasons, 85)
|
|
largemouth_bass = create_fish("Largemouth Bass", mountain_lake, season.all_seasons, 50)
|
|
lava_eel = create_fish("Lava Eel", mines_floor_100, season.all_seasons, 90)
|
|
lingcod = create_fish("Lingcod", town_river + forest_river + mountain_lake, season.winter, 85)
|
|
lionfish = create_fish("Lionfish", ginger_island_ocean, season.all_seasons, 50)
|
|
midnight_carp = create_fish("Midnight Carp", mountain_lake + forest_pond + ginger_island_river,
|
|
(season.fall, season.winter), 55)
|
|
octopus = create_fish("Octopus", ocean, season.summer, 95)
|
|
perch = create_fish("Perch", town_river + forest_river + forest_pond + mountain_lake, season.winter, 35)
|
|
pike = create_fish("Pike", town_river + forest_river + forest_pond, (season.summer, season.winter), 60)
|
|
pufferfish = create_fish("Pufferfish", ocean + ginger_island_ocean, season.summer, 80)
|
|
rainbow_trout = create_fish("Rainbow Trout", town_river + forest_river + mountain_lake, season.summer, 45)
|
|
red_mullet = create_fish("Red Mullet", ocean, (season.summer, season.winter), 55)
|
|
red_snapper = create_fish("Red Snapper", ocean, (season.summer, season.fall), 40)
|
|
salmon = create_fish("Salmon", town_river + forest_river, season.fall, 50)
|
|
sandfish = create_fish("Sandfish", desert, season.all_seasons, 65)
|
|
sardine = create_fish("Sardine", ocean, (season.spring, season.fall, season.winter), 30)
|
|
scorpion_carp = create_fish("Scorpion Carp", desert, season.all_seasons, 90)
|
|
sea_cucumber = create_fish("Sea Cucumber", ocean, (season.fall, season.winter), 40)
|
|
shad = create_fish("Shad", town_river + forest_river, season.not_winter, 45)
|
|
slimejack = create_fish("Slimejack", mutant_bug_lair, season.all_seasons, 55)
|
|
smallmouth_bass = create_fish("Smallmouth Bass", town_river + forest_river, (season.spring, season.fall), 28)
|
|
squid = create_fish("Squid", ocean, season.winter, 75)
|
|
stingray = create_fish("Stingray", pirate_cove, season.all_seasons, 80)
|
|
stonefish = create_fish("Stonefish", mines_floor_20, season.all_seasons, 65)
|
|
sturgeon = create_fish("Sturgeon", mountain_lake, (season.summer, season.winter), 78)
|
|
sunfish = create_fish("Sunfish", town_river + forest_river, (season.spring, season.summer), 30)
|
|
super_cucumber = create_fish("Super Cucumber", ocean + ginger_island_ocean, (season.summer, season.fall), 80)
|
|
tiger_trout = create_fish("Tiger Trout", town_river + forest_river, (season.fall, season.winter), 60)
|
|
tilapia = create_fish("Tilapia", ocean + ginger_island_ocean, (season.summer, season.fall), 50)
|
|
# Tuna has different seasons on ginger island. Should be changed when the whole fish thing is refactored
|
|
tuna = create_fish("Tuna", ocean + ginger_island_ocean, (season.summer, season.winter), 70)
|
|
void_salmon = create_fish("Void Salmon", witch_swamp, season.all_seasons, 80)
|
|
walleye = create_fish("Walleye", town_river + forest_river + forest_pond + mountain_lake, season.fall, 45)
|
|
woodskip = create_fish("Woodskip", secret_woods, season.all_seasons, 50)
|
|
|
|
blob_fish = create_fish("Blobfish", night_market, season.winter, 75)
|
|
midnight_squid = create_fish("Midnight Squid", night_market, season.winter, 55)
|
|
spook_fish = create_fish("Spook Fish", night_market, season.winter, 60)
|
|
|
|
angler = create_fish(Fish.angler, town_river, season.fall, 85, True, False)
|
|
crimsonfish = create_fish(Fish.crimsonfish, ocean, season.summer, 95, True, False)
|
|
glacierfish = create_fish(Fish.glacierfish, forest_river, season.winter, 100, True, False)
|
|
legend = create_fish(Fish.legend, mountain_lake, season.spring, 110, True, False)
|
|
mutant_carp = create_fish(Fish.mutant_carp, sewers, season.all_seasons, 80, True, False)
|
|
|
|
ms_angler = create_fish(Fish.ms_angler, town_river, season.fall, 85, True, True)
|
|
son_of_crimsonfish = create_fish(Fish.son_of_crimsonfish, ocean, season.summer, 95, True, True)
|
|
glacierfish_jr = create_fish(Fish.glacierfish_jr, forest_river, season.winter, 100, True, True)
|
|
legend_ii = create_fish(Fish.legend_ii, mountain_lake, season.spring, 110, True, True)
|
|
radioactive_carp = create_fish(Fish.radioactive_carp, sewers, season.all_seasons, 80, True, True)
|
|
|
|
baby_lunaloo = create_fish(SVEFish.baby_lunaloo, ginger_island_ocean, season.all_seasons, 15, mod_name=ModNames.sve)
|
|
bonefish = create_fish(SVEFish.bonefish, crimson_badlands, season.all_seasons, 70, mod_name=ModNames.sve)
|
|
bull_trout = create_fish(SVEFish.bull_trout, forest_river, season.not_spring, 45, mod_name=ModNames.sve)
|
|
butterfish = create_fish(SVEFish.butterfish, shearwater, season.not_winter, 75, mod_name=ModNames.sve)
|
|
clownfish = create_fish(SVEFish.clownfish, ginger_island_ocean, season.all_seasons, 45, mod_name=ModNames.sve)
|
|
daggerfish = create_fish(SVEFish.daggerfish, highlands, season.all_seasons, 50, mod_name=ModNames.sve)
|
|
frog = create_fish(SVEFish.frog, mountain_lake, (season.spring, season.summer), 70, mod_name=ModNames.sve)
|
|
gemfish = create_fish(SVEFish.gemfish, highlands, season.all_seasons, 100, mod_name=ModNames.sve)
|
|
goldenfish = create_fish(SVEFish.goldenfish, sprite_spring, season.all_seasons, 60, mod_name=ModNames.sve)
|
|
grass_carp = create_fish(SVEFish.grass_carp, secret_woods, (season.spring, season.summer), 85, mod_name=ModNames.sve)
|
|
king_salmon = create_fish(SVEFish.king_salmon, forest_river, (season.spring, season.summer), 80, mod_name=ModNames.sve)
|
|
kittyfish = create_fish(SVEFish.kittyfish, shearwater, (season.fall, season.winter), 85, mod_name=ModNames.sve)
|
|
lunaloo = create_fish(SVEFish.lunaloo, ginger_island_ocean, season.all_seasons, 70, mod_name=ModNames.sve)
|
|
meteor_carp = create_fish(SVEFish.meteor_carp, sprite_spring, season.all_seasons, 80, mod_name=ModNames.sve)
|
|
minnow = create_fish(SVEFish.minnow, town_river, season.all_seasons, 1, mod_name=ModNames.sve)
|
|
puppyfish = create_fish(SVEFish.puppyfish, shearwater, season.not_winter, 85, mod_name=ModNames.sve)
|
|
radioactive_bass = create_fish(SVEFish.radioactive_bass, sewers, season.all_seasons, 90, mod_name=ModNames.sve)
|
|
seahorse = create_fish(SVEFish.seahorse, ginger_island_ocean, season.all_seasons, 25, mod_name=ModNames.sve)
|
|
shiny_lunaloo = create_fish(SVEFish.shiny_lunaloo, ginger_island_ocean, season.all_seasons, 110, mod_name=ModNames.sve)
|
|
snatcher_worm = create_fish(SVEFish.snatcher_worm, mutant_bug_lair, season.all_seasons, 75, mod_name=ModNames.sve)
|
|
starfish = create_fish(SVEFish.starfish, ginger_island_ocean, season.all_seasons, 75, mod_name=ModNames.sve)
|
|
torpedo_trout = create_fish(SVEFish.torpedo_trout, fable_reef, season.all_seasons, 70, mod_name=ModNames.sve)
|
|
undeadfish = create_fish(SVEFish.undeadfish, crimson_badlands, season.all_seasons, 80, mod_name=ModNames.sve)
|
|
void_eel = create_fish(SVEFish.void_eel, witch_swamp, season.all_seasons, 100, mod_name=ModNames.sve)
|
|
water_grub = create_fish(SVEFish.water_grub, mutant_bug_lair, season.all_seasons, 60, mod_name=ModNames.sve)
|
|
sea_sponge = create_fish(SVEFish.sea_sponge, ginger_island_ocean, season.all_seasons, 40, mod_name=ModNames.sve)
|
|
dulse_seaweed = create_fish(SVEFish.dulse_seaweed, vineyard, season.all_seasons, 50, mod_name=ModNames.sve)
|
|
|
|
void_minnow = create_fish(DistantLandsFish.void_minnow, witch_swamp, season.all_seasons, 15, mod_name=ModNames.distant_lands)
|
|
purple_algae = create_fish(DistantLandsFish.purple_algae, witch_swamp, season.all_seasons, 15, mod_name=ModNames.distant_lands)
|
|
swamp_leech = create_fish(DistantLandsFish.swamp_leech, witch_swamp, season.all_seasons, 15, mod_name=ModNames.distant_lands)
|
|
giant_horsehoe_crab = create_fish(DistantLandsFish.giant_horsehoe_crab, witch_swamp, season.all_seasons, 90, mod_name=ModNames.distant_lands)
|
|
|
|
|
|
clam = create_fish("Clam", ocean, season.all_seasons, -1)
|
|
cockle = create_fish("Cockle", ocean, season.all_seasons, -1)
|
|
crab = create_fish("Crab", ocean, season.all_seasons, -1)
|
|
crayfish = create_fish("Crayfish", fresh_water, season.all_seasons, -1)
|
|
lobster = create_fish("Lobster", ocean, season.all_seasons, -1)
|
|
mussel = create_fish("Mussel", ocean, season.all_seasons, -1)
|
|
oyster = create_fish("Oyster", ocean, season.all_seasons, -1)
|
|
periwinkle = create_fish("Periwinkle", fresh_water, season.all_seasons, -1)
|
|
shrimp = create_fish("Shrimp", ocean, season.all_seasons, -1)
|
|
snail = create_fish("Snail", fresh_water, season.all_seasons, -1)
|
|
|
|
legendary_fish = [angler, crimsonfish, glacierfish, legend, mutant_carp]
|
|
extended_family = [ms_angler, son_of_crimsonfish, glacierfish_jr, legend_ii, radioactive_carp]
|
|
special_fish = [*legendary_fish, blob_fish, lava_eel, octopus, scorpion_carp, ice_pip, super_cucumber, dorado]
|
|
island_fish = [lionfish, blue_discus, stingray, *extended_family]
|
|
|
|
all_fish_by_name = {fish.name: fish for fish in all_fish}
|
|
|
|
|
|
def get_fish_for_mods(mods: Set[str]) -> List[FishItem]:
|
|
fish_for_mods = []
|
|
for fish in all_fish:
|
|
if fish.mod_name and fish.mod_name not in mods:
|
|
continue
|
|
fish_for_mods.append(fish)
|
|
return fish_for_mods
|