localdex 0.2.55__py3-none-any.whl → 0.3.1__py3-none-any.whl

localdex/__init__.py

@@ -2,6 +2,8 @@
 from .core import LocalDex
 from .models import Pokemon, Move, Ability, Item, BaseStats
 from .exceptions import PokemonNotFoundError, MoveNotFoundError, AbilityNotFoundError, ItemNotFoundError
+from .stat_guesser import BattleStatGuesser, PokemonSet, battle_stat_optimizer
+from .stat_calculator import StatCalculator
 
 __all__ = [
     "LocalDex",
@@ -14,4 +16,8 @@ __all__ = [
     "MoveNotFoundError", 
     "AbilityNotFoundError",
     "ItemNotFoundError",
+    "BattleStatGuesser",
+    "PokemonSet",
+    "battle_stat_optimizer",
+    "StatCalculator",
 ] 

localdex/core.py

@@ -20,6 +20,7 @@ from .exceptions import (
 )
 from .data_loader import DataLoader
 from .name_normalizer import PokemonNameNormalizer
+from .stat_guesser import BattleStatGuesser, PokemonSet, battle_stat_optimizer
 
 
 class LocalDex:
@@ -72,9 +73,10 @@ class LocalDex:
             for pokemon in all_pokemon:
                 # Index by type
                 for pokemon_type in pokemon.types:
-                    if pokemon_type not in self._pokemon_by_type:
-                        self._pokemon_by_type[pokemon_type] = set()
-                    self._pokemon_by_type[pokemon_type].add(pokemon.name.lower())
+                    type_key = pokemon_type.lower()
+                    if type_key not in self._pokemon_by_type:
+                        self._pokemon_by_type[type_key] = set()
+                    self._pokemon_by_type[type_key].add(pokemon.name.lower())
                 
                 # Index by generation
                 if pokemon.generation:
@@ -513,7 +515,7 @@ class LocalDex:
             name=data.get("name", ""),
             description=data.get("description", data.get("desc", "")),
             short_description=data.get("shortDesc"),
-            generation=data.get("gen"),
+            generation=data.get("generation", data.get("gen")),
             num=data.get("num"),
             spritenum=data.get("spritenum"),
             fling=data.get("fling"),
@@ -555,3 +557,42 @@ class LocalDex:
         """Access to stat calculation functionality."""
         from .stat_calculator import StatCalculator
         return StatCalculator(self)
+    
+    def create_stat_guesser(self, format_id: str) -> BattleStatGuesser:
+        """
+        Create a BattleStatGuesser instance for the specified format.
+        
+        Args:
+            format_id: The format ID (e.g., "gen9ou", "gen8ou", etc.)
+            
+        Returns:
+            BattleStatGuesser instance configured for the format
+        """
+        return BattleStatGuesser(format_id, self)
+    
+    def guess_pokemon_stats(self, pokemon_set: PokemonSet, format_id: str = "gen9ou") -> Dict[str, Any]:
+        """
+        Guess the role and EV spread for a Pokemon set.
+        
+        Args:
+            pokemon_set: The Pokemon set to analyze
+            format_id: The format ID (default: "gen9ou")
+            
+        Returns:
+            Dictionary containing role, EVs, plus/minus stats, and move information
+        """
+        guesser = self.create_stat_guesser(format_id)
+        return guesser.guess(pokemon_set)
+    
+    def optimize_pokemon_stats(self, pokemon_set: PokemonSet, format_id: str = "gen9ou") -> Optional[Dict[str, Any]]:
+        """
+        Optimize a Pokemon's EV spread and nature.
+        
+        Args:
+            pokemon_set: The Pokemon set to optimize
+            format_id: The format ID (default: "gen9ou")
+            
+        Returns:
+            Optimized spread or None if no optimization is possible
+        """
+        return battle_stat_optimizer(pokemon_set, format_id, self)

localdex/data/items/crystal-cluster.json

@@ -0,0 +1,3 @@
+{
+  "name": "crystal-cluster"
+}