stsdb 0.1.2__tar.gz → 0.2.0__tar.gz

{stsdb-0.1.2 → stsdb-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stsdb
-Version: 0.1.2
+Version: 0.2.0
 Summary: Deterministic Slay the Spire card and relic query toolset
 Author: Taardisaa
 License: MIT

{stsdb-0.1.2 → stsdb-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stsdb"
-version = "0.1.2"
+version = "0.2.0"
 description = "Deterministic Slay the Spire card and relic query toolset"
 readme = "README.md"
 requires-python = ">=3.9"

stsdb-0.2.0/stsdb/__init__.py

@@ -0,0 +1,3 @@
+from .core import query_card, query_potion, query_relic
+
+__all__ = ["query_card", "query_potion", "query_relic"]

{stsdb-0.1.2 → stsdb-0.2.0}/stsdb/core.py

@@ -95,7 +95,10 @@ def _apply_searing_blow_upgrades(base_card, upgrade_times: int):
     return upgraded_card
 
 
-def query_card(name: str, upgrade_times: int = 0):
+_CHARACTER_CLASSES = ("Ironclad", "Silent", "Defect", "Watcher")
+
+
+def query_card(name: str, upgrade_times: int = 0, character_class: str = ""):
     if upgrade_times < 0:
         return {"found": False, "error": "INVALID_UPGRADE_TIMES"}
 
@@ -109,6 +112,18 @@ def query_card(name: str, upgrade_times: int = 0):
     canonical_name = entry["name"]
     upgrade_info = _card_upgrade_by_name().get(normalized_name)
 
+    # If the base entry isn't upgradeable, try character-specific variant
+    # e.g. "Strike" -> "Strike (Watcher)" which has upgrade data
+    if not (upgrade_info and upgrade_info["has_upgrade"]):
+        variant_name = _resolve_character_variant(normalized_name, character_class)
+        if variant_name:
+            variant_card = _cards_by_name().get(variant_name)
+            variant_upgrade = _card_upgrade_by_name().get(variant_name)
+            if variant_card and variant_upgrade and variant_upgrade["has_upgrade"]:
+                entry = dict(variant_card)
+                canonical_name = entry["name"]
+                upgrade_info = variant_upgrade
+
     applied_upgrade_times = 0
     max_upgrade_times = 0
 
@@ -130,6 +145,22 @@ def query_card(name: str, upgrade_times: int = 0):
     return {"found": True, "entry": entry}
 
 
+def _resolve_character_variant(normalized_name: str, character_class: str) -> str:
+    """Try to find a character-specific card variant, e.g. 'strike' -> 'strike (watcher)'."""
+    cards = _cards_by_name()
+    # If character_class is provided, try that first
+    if character_class:
+        variant = f"{normalized_name} ({character_class.lower()})"
+        if variant in cards:
+            return variant
+    # Otherwise try all character classes
+    for cls in _CHARACTER_CLASSES:
+        variant = f"{normalized_name} ({cls.lower()})"
+        if variant in cards:
+            return variant
+    return ""
+
+
 def query_relic(name: str):
     relic = _relics_by_name().get(name.lower())
     if relic is None:
@@ -138,3 +169,26 @@ def query_relic(name: str):
     entry = dict(relic)
     entry["available_to"] = _relic_available_to().get(entry["name"], [])
     return {"found": True, "entry": entry}
+
+
+@lru_cache(maxsize=1)
+def _potions_by_name():
+    potions = {}
+    with _data_path("potion.csv").open("r", encoding="utf-8", newline="") as f:
+        reader = csv.reader(f, delimiter=";")
+        for row in reader:
+            name, rarity, playable_by, description = row
+            potions[name.lower()] = {
+                "name": name,
+                "rarity": rarity,
+                "playable_by": playable_by,
+                "description": description,
+            }
+    return potions
+
+
+def query_potion(name: str):
+    potion = _potions_by_name().get(name.lower())
+    if potion is None:
+        return {"found": False, "error": "POTION_NOT_FOUND"}
+    return {"found": True, "entry": dict(potion)}

stsdb-0.2.0/stsdb/data/potion.csv

@@ -0,0 +1,42 @@
+Ambrosia;Rare;Watcher;Enter Divinity Stance.
+Ancient Potion;Uncommon;Any;Gain 1 Artifact.
+Attack Potion;Common;Any;Choose 1 of 3 random Attack cards to add to your hand, it costs 0 this turn.
+Blessing of the Forge;Common;Any;Upgrade all cards in your hand for the rest of combat.
+Block Potion;Common;Any;Gain 12 Block.
+Blood Potion;Common;Ironclad;Heal for 20% of your Max HP.
+Bottled Miracle;Common;Watcher;Add 2 Miracles to your hand.
+Colorless Potion;Common;Any;Choose 1 of 3 random Colorless cards to add to your hand, it costs 0 this turn.
+Cultist Potion;Rare;Any;Gain 1 Ritual.
+Cunning Potion;Uncommon;Silent;Add 3 Shiv+ to your hand.
+Dexterity Potion;Common;Any;Gain 2 Dexterity.
+Distilled Chaos;Uncommon;Any;Play the top 3 cards of your draw pile.
+Duplication Potion;Uncommon;Any;This turn, your next card is played twice.
+Elixir;Uncommon;Ironclad;Exhaust any number of cards in your hand.
+Energy Potion;Common;Any;Gain 2 Energy.
+Entropic Brew;Rare;Any;Fill all your empty potion slots with random potions.
+Essence Of Darkness;Rare;Defect;Channel 1 Dark for each orb slot.
+Essence of Steel;Uncommon;Any;Gain 4 Plated Armor.
+Explosive Potion;Common;Any;Deal 10 damage to ALL enemies.
+Fairy in a Bottle;Rare;Any;When you would die, heal to 30% of your Max HP instead and discard this potion.
+Fear Potion;Common;Any;Apply 3 Vulnerable.
+Fire Potion;Common;Any;Deal 20 damage.
+Flex Potion;Common;Any;Gain 5 Strength. At the end of your turn, lose 5 Strength.
+Focus Potion;Common;Defect;Gain 2 Focus.
+Fruit Juice;Rare;Any;Gain 5 Max HP.
+Gambler's Brew;Uncommon;Any;Discard any number of cards then draw that many.
+Ghost in a Jar;Rare;Any;Gain 1 Intangible.
+Heart of Iron;Rare;Ironclad;Gain 6 Metallicize.
+Liquid Bronze;Uncommon;Any;Gain 3 Thorns.
+Liquid Memories;Uncommon;Any;Choose a card in your discard pile and return it to your hand. It costs 0 this turn.
+Poison Potion;Common;Silent;Apply 6 Poison.
+Potion Of Capacity;Uncommon;Defect;Gain 2 Orb slots.
+Power Potion;Common;Any;Choose 1 of 3 random Power cards to add to your hand, it costs 0 this turn.
+Regen Potion;Uncommon;Any;Gain 5 Regen.
+Skill Potion;Common;Any;Choose 1 of 3 random Skill cards to add to your hand, it costs 0 this turn.
+Smoke Bomb;Rare;Any;Escape from a non-boss combat. Receive no rewards.
+Snecko Oil;Rare;Any;Draw 5 cards. Randomize the cost of cards in your hand.
+Speed Potion;Common;Any;Gain 5 Dexterity. At the end of your turn, lose 5 Dexterity.
+Stance Potion;Uncommon;Watcher;Enter Calm or Wrath.
+Strength Potion;Common;Any;Gain 2 Strength.
+Swift Potion;Common;Any;Draw 3 cards.
+Weak Potion;Common;Any;Apply 3 Weak.

{stsdb-0.1.2 → stsdb-0.2.0}/stsdb.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stsdb
-Version: 0.1.2
+Version: 0.2.0
 Summary: Deterministic Slay the Spire card and relic query toolset
 Author: Taardisaa
 License: MIT

{stsdb-0.1.2 → stsdb-0.2.0}/stsdb.egg-info/SOURCES.txt

@@ -13,6 +13,7 @@ stsdb/data/card.csv
 stsdb/data/card_upgrade.csv
 stsdb/data/hero.csv
 stsdb/data/play.csv
+stsdb/data/potion.csv
 stsdb/data/relic.csv
 stsdb/data/relic_availability.csv
 tests/test_stsdb.py

stsdb-0.1.2/stsdb/__init__.py

@@ -1,3 +0,0 @@
-from .core import query_card, query_relic
-
-__all__ = ["query_card", "query_relic"]