6
6
7
7
from . import data
8
8
from .bundles .bundle_room import BundleRoom
9
- from .data .fish_data import legendary_fish , special_fish , get_fish_for_mods
9
+ from .data .fish_data import special_fish , get_fish_for_mods
10
10
from .data .museum_data import all_museum_items
11
11
from .data .villagers_data import get_villagers_for_mods
12
12
from .mods .mod_data import ModNames
@@ -206,7 +206,8 @@ def extend_fishsanity_locations(randomized_locations: List[LocationData], option
206
206
if fishsanity == Fishsanity .option_none :
207
207
return
208
208
elif fishsanity == Fishsanity .option_legendaries :
209
- randomized_locations .extend (location_table [f"{ prefix } { legendary .name } " ] for legendary in legendary_fish )
209
+ fish_locations = [location_table [f"{ prefix } { fish .name } " ] for fish in active_fish if fish .legendary ]
210
+ randomized_locations .extend (filter_disabled_locations (options , fish_locations ))
210
211
elif fishsanity == Fishsanity .option_special :
211
212
randomized_locations .extend (location_table [f"{ prefix } { special .name } " ] for special in special_fish )
212
213
elif fishsanity == Fishsanity .option_randomized :
@@ -216,7 +217,7 @@ def extend_fishsanity_locations(randomized_locations: List[LocationData], option
216
217
fish_locations = [location_table [f"{ prefix } { fish .name } " ] for fish in active_fish ]
217
218
randomized_locations .extend (filter_disabled_locations (options , fish_locations ))
218
219
elif fishsanity == Fishsanity .option_exclude_legendaries :
219
- fish_locations = [location_table [f"{ prefix } { fish .name } " ] for fish in active_fish if fish not in legendary_fish ]
220
+ fish_locations = [location_table [f"{ prefix } { fish .name } " ] for fish in active_fish if not fish . legendary ]
220
221
randomized_locations .extend (filter_disabled_locations (options , fish_locations ))
221
222
elif fishsanity == Fishsanity .option_exclude_hard_fish :
222
223
fish_locations = [location_table [f"{ prefix } { fish .name } " ] for fish in active_fish if fish .difficulty < 80 ]
0 commit comments