The Messenger: Make modules and tests PEP8 (#2276)

* The Messenger: PEP8 module and test names * fix dumb * whitespace
2023-10-08 07:33:39 -05:00
parent cc2247bfa0
commit 6f9484f375
13 changed files with 65 additions and 71 deletions
--- a/worlds/messenger/shop.py
+++ b/worlds/messenger/shop.py
@@ -0,0 +1,100 @@
+from typing import Dict, List, NamedTuple, Optional, Set, TYPE_CHECKING, Tuple, Union
+
+if TYPE_CHECKING:
+    from . import MessengerWorld
+else:
+    MessengerWorld = object
+
+PROG_SHOP_ITEMS: List[str] = [
+    "Path of Resilience",
+    "Meditation",
+    "Strike of the Ninja",
+    "Second Wind",
+    "Currents Master",
+    "Aerobatics Warrior",
+]
+
+USEFUL_SHOP_ITEMS: List[str] = [
+    "Karuta Plates",
+    "Serendipitous Bodies",
+    "Kusari Jacket",
+    "Energy Shuriken",
+    "Serendipitous Minds",
+    "Rejuvenate Spirit",
+    "Demon's Bane",
+]
+
+
+class ShopData(NamedTuple):
+    internal_name: str
+    min_price: int
+    max_price: int
+    prerequisite: Optional[Union[str, Set[str]]] = None
+
+
+SHOP_ITEMS: Dict[str, ShopData] = {
+    "Karuta Plates":        ShopData("HP_UPGRADE_1", 20, 200),
+    "Serendipitous Bodies": ShopData("ENEMY_DROP_HP", 20, 300, "The Shop - Karuta Plates"),
+    "Path of Resilience":   ShopData("DAMAGE_REDUCTION", 100, 500, "The Shop - Serendipitous Bodies"),
+    "Kusari Jacket":        ShopData("HP_UPGRADE_2", 100, 500, "The Shop - Serendipitous Bodies"),
+    "Energy Shuriken":      ShopData("SHURIKEN", 20, 200),
+    "Serendipitous Minds":  ShopData("ENEMY_DROP_MANA", 20, 300, "The Shop - Energy Shuriken"),
+    "Prepared Mind":        ShopData("SHURIKEN_UPGRADE_1", 100, 600, "The Shop - Serendipitous Minds"),
+    "Meditation":           ShopData("CHECKPOINT_FULL", 100, 600,
+                                     {"The Shop - Prepared Mind", "The Shop - Kusari Jacket"}),
+    "Rejuvenative Spirit":  ShopData("POTION_FULL_HEAL_AND_HP", 300, 800, "The Shop - Meditation"),
+    "Centered Mind":        ShopData("SHURIKEN_UPGRADE_2", 300, 800, "The Shop - Meditation"),
+    "Strike of the Ninja":  ShopData("ATTACK_PROJECTILE", 20, 200),
+    "Second Wind":          ShopData("AIR_RECOVER", 20, 350, "The Shop - Strike of the Ninja"),
+    "Currents Master":      ShopData("SWIM_DASH", 100, 600, "The Shop - Second Wind"),
+    "Aerobatics Warrior":   ShopData("GLIDE_ATTACK", 300, 800, "The Shop - Currents Master"),
+    "Demon's Bane":         ShopData("CHARGED_ATTACK", 400, 1000,
+                                     {"The Shop - Rejuvenative Spirit", "The Shop - Aerobatics Warrior"}),
+    "Devil's Due":          ShopData("QUARBLE_DISCOUNT_50", 20, 200),
+    "Time Sense":           ShopData("TIME_WARP", 20, 300),
+    "Power Sense":          ShopData("POWER_SEAL", 100, 800, "The Shop - Time Sense"),
+    "Focused Power Sense":  ShopData("POWER_SEAL_WORLD_MAP", 300, 600, "The Shop - Power Sense"),
+}
+
+FIGURINES: Dict[str, ShopData] = {
+    "Green Kappa Figurine":         ShopData("GREEN_KAPPA", 100, 500),
+    "Blue Kappa Figurine":          ShopData("BLUE_KAPPA", 100, 500),
+    "Ountarde Figurine":            ShopData("OUNTARDE", 100, 500),
+    "Red Kappa Figurine":           ShopData("RED_KAPPA", 100, 500),
+    "Demon King Figurine":          ShopData("DEMON_KING", 600, 2000),
+    "Quillshroom Figurine":         ShopData("QUILLSHROOM", 100, 500),
+    "Jumping Quillshroom Figurine": ShopData("JUMPING_QUILLSHROOM", 100, 500),
+    "Scurubu Figurine":             ShopData("SCURUBU", 100, 500),
+    "Jumping Scurubu Figurine":     ShopData("JUMPING_SCURUBU", 100, 500),
+    "Wallaxer Figurine":            ShopData("WALLAXER", 100, 500),
+    "Barmath'azel Figurine":        ShopData("BARMATHAZEL", 600, 2000),
+    "Queen of Quills Figurine":     ShopData("QUEEN_OF_QUILLS", 400, 1000),
+    "Demon Hive Figurine":          ShopData("DEMON_HIVE", 100, 500),
+}
+
+
+def shuffle_shop_prices(world: MessengerWorld) -> Tuple[Dict[str, int], Dict[str, int]]:
+    shop_price_mod = world.multiworld.shop_price[world.player].value
+    shop_price_planned = world.multiworld.shop_price_plan[world.player]
+
+    shop_prices: Dict[str, int] = {}
+    figurine_prices: Dict[str, int] = {}
+    for item, price in shop_price_planned.value.items():
+        if not isinstance(price, int):
+            price = world.random.choices(list(price.keys()), weights=list(price.values()))[0]
+        if "Figurine" in item:
+            figurine_prices[item] = price
+        else:
+            shop_prices[item] = price
+
+    remaining_slots = [item for item in [*SHOP_ITEMS, *FIGURINES] if item not in shop_price_planned.value]
+    for shop_item in remaining_slots:
+        shop_data = SHOP_ITEMS.get(shop_item, FIGURINES.get(shop_item))
+        price = world.random.randint(shop_data.min_price, shop_data.max_price)
+        adjusted_price = min(int(price * shop_price_mod / 100), 5000)
+        if "Figurine" in shop_item:
+            figurine_prices[shop_item] = adjusted_price
+        else:
+            shop_prices[shop_item] = adjusted_price
+
+    return shop_prices, figurine_prices