localdex 0.2.2__py3-none-any.whl → 0.2.23__py3-none-any.whl

localdex/__init__.py

@@ -2,7 +2,6 @@
 from .core import LocalDex
 from .models import Pokemon, Move, Ability, Item, BaseStats
 from .exceptions import PokemonNotFoundError, MoveNotFoundError, AbilityNotFoundError, ItemNotFoundError
-from .random_battles import RandomBattleSets
 
 __all__ = [
     "LocalDex",
@@ -15,6 +14,4 @@ __all__ = [
     "MoveNotFoundError", 
     "AbilityNotFoundError",
     "ItemNotFoundError",
-    "RandomBattleSets",
-    "SpriteDownloader",
 ] 

localdex/core.py

@@ -18,7 +18,6 @@ from .exceptions import (
     ItemNotFoundError, DataLoadError, SearchError
 )
 from .data_loader import DataLoader
-from .random_battles import RandomBattleSets
 
 
 class LocalDex:
@@ -30,7 +29,7 @@ class LocalDex:
     performance and comprehensive search capabilities.
     """
     
-    def __init__(self, data_path: Optional[str] = None, data_dir: Optional[str] = None, enable_caching: bool = True, enable_random_battles: bool = True):
+    def __init__(self, data_path: Optional[str] = None, data_dir: Optional[str] = None, enable_caching: bool = True):
         """
         Initialize the LocalDex.
         
@@ -38,18 +37,12 @@ class LocalDex:
             data_path: Optional path to data directory. If None, uses package data.
             data_dir: Alias for data_path for backward compatibility.
             enable_caching: Whether to enable caching for better performance.
-            enable_random_battles: Whether to enable random battle sets functionality.
-
         """
         # Use data_dir if provided, otherwise use data_path
         final_data_path = data_dir if data_dir is not None else data_path
         self.data_loader = DataLoader(final_data_path)
         self.data_dir = final_data_path  # Store for backward compatibility
         self.enable_caching = enable_caching
-
-        
-        # Initialize random battle sets
-        self.random_battles = RandomBattleSets() if enable_random_battles else None
         
         # Initialize caches
         self._pokemon_cache: Dict[str, Pokemon] = {}
@@ -433,54 +426,38 @@ class LocalDex:
         base_stats_data = data.get("baseStats", {})
         base_stats = BaseStats(
             hp=base_stats_data.get("hp", 0),
-            attack=base_stats_data.get("attack", base_stats_data.get("atk", 0)),
-            defense=base_stats_data.get("defense", base_stats_data.get("def", 0)),
-            special_attack=base_stats_data.get("special_attack", base_stats_data.get("spa", 0)),
-            special_defense=base_stats_data.get("special_defense", base_stats_data.get("spd", 0)),
-            speed=base_stats_data.get("speed", base_stats_data.get("spe", 0))
+            attack=base_stats_data.get("attack", 0),
+            defense=base_stats_data.get("defense", 0),
+            special_attack=base_stats_data.get("special-attack", 0),
+            special_defense=base_stats_data.get("special-defense", 0),
+            speed=base_stats_data.get("speed", 0),
         )
-        
-        # Create Pokemon object
-        pokemon = Pokemon(
-            id=data.get("id", 0),
-            name=data.get("name", ""),
+        return Pokemon(
+            id=data["id"],
+            name=data["name"],
             types=data.get("types", []),
             base_stats=base_stats,
             height=data.get("height"),
             weight=data.get("weight"),
             color=data.get("color"),
-            evolutions=data.get("evolutions"),
+            abilities=data.get("abilities", {}),
+            moves=data.get("moves", []),
+            learnset=data.get("learnset", {}),
+            evolutions=data.get("evolutions", []),
             prevo=data.get("prevo"),
             evo_level=data.get("evoLevel"),
             evo_type=data.get("evoType"),
             evo_condition=data.get("evoCondition"),
             evo_item=data.get("evoItem"),
-            egg_groups=data.get("eggGroups"),
-            gender_ratio=data.get("genderRatio"),
+            egg_groups=data.get("eggGroups", []),
+            gender_ratio=data.get("genderRatio", {}),
             generation=data.get("generation"),
             description=data.get("description"),
             is_legendary=data.get("isLegendary", False),
             is_mythical=data.get("isMythical", False),
             is_ultra_beast=data.get("isUltraBeast", False),
-            random_battle_sets=data.get("randomBattleSets"),
-            metadata=data
+            metadata=data.get("metadata", {}),
         )
-        
-        # Add abilities
-        abilities_data = data.get("abilities", {})
-        for slot, ability_data in abilities_data.items():
-            if isinstance(ability_data, dict) and "data" in ability_data:
-                ability = self._create_ability_from_data(ability_data["data"])
-                pokemon.abilities[slot] = ability
-        
-        # Add moves
-        moves_data = data.get("moves", [])
-        for move_data in moves_data:
-            if isinstance(move_data, dict):
-                move = self._create_move_from_data(move_data)
-                pokemon.moves.append(move)
-        
-        return pokemon
     
     def _create_move_from_data(self, data: Dict[str, Any]) -> Move:
         """Create a Move object from raw data."""
@@ -565,138 +542,4 @@ class LocalDex:
             "items": len(self._item_cache)
         }
     
-    # Random Battle Sets Methods
-    
-    def get_random_battle_sets(self, pokemon_name: str, generation: int = 9, force_refresh: bool = False) -> Optional[Dict[str, Any]]:
-        """
-        Get random battle sets for a specific Pokemon.
-        
-        Args:
-            pokemon_name: Name of the Pokemon (case-insensitive)
-            generation: Generation to get sets from (8 or 9)
-            force_refresh: If True, re-download data
-            
-        Returns:
-            Pokemon's random battle sets or None if not found
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        if generation == 9:
-            return self.random_battles.get_pokemon_gen9_sets(pokemon_name, force_refresh)
-        elif generation == 8:
-            return self.random_battles.get_pokemon_gen8_data(pokemon_name, force_refresh)
-        else:
-            raise ValueError(f"Generation {generation} is not supported. Use 8 or 9.")
-    
-    def get_all_random_battle_sets(self, generation: int = 9, force_refresh: bool = False) -> Dict[str, Any]:
-        """
-        Get all random battle sets for a generation.
-        
-        Args:
-            generation: Generation to get sets from (8 or 9)
-            force_refresh: If True, re-download data
-            
-        Returns:
-            Dictionary containing all random battle sets for the generation
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        if generation == 9:
-            return self.random_battles.get_gen9_sets(force_refresh)
-        elif generation == 8:
-            return self.random_battles.get_gen8_data(force_refresh)
-        else:
-            raise ValueError(f"Generation {generation} is not supported. Use 8 or 9.")
-    
-    def search_random_battle_pokemon_by_move(self, move_name: str, generation: int = 9, force_refresh: bool = False) -> List[str]:
-        """
-        Search for Pokemon that have a specific move in their random battle sets.
-        
-        Args:
-            move_name: Name of the move to search for (case-insensitive)
-            generation: Generation to search in (8 or 9)
-            force_refresh: If True, re-download data
-            
-        Returns:
-            List of Pokemon names that have the move
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        return self.random_battles.search_pokemon_by_move(move_name, generation, force_refresh)
-    
-    def search_random_battle_pokemon_by_ability(self, ability_name: str, generation: int = 9, force_refresh: bool = False) -> List[str]:
-        """
-        Search for Pokemon that have a specific ability in their random battle sets.
-        
-        Args:
-            ability_name: Name of the ability to search for (case-insensitive)
-            generation: Generation to search in (8 or 9)
-            force_refresh: If True, re-download data
-            
-        Returns:
-            List of Pokemon names that have the ability
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        return self.random_battles.search_pokemon_by_ability(ability_name, generation, force_refresh)
-    
-    def get_available_random_battle_generations(self) -> List[int]:
-        """
-        Get list of available generations for random battles.
-        
-        Returns:
-            List of generation numbers
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        return self.random_battles.get_available_generations()
-    
-    def get_random_battle_formats(self, generation: int) -> List[str]:
-        """
-        Get available formats for a specific generation.
-        
-        Args:
-            generation: Generation number
-            
-        Returns:
-            List of available format names
-            
-        Raises:
-            ValueError: If random battles are not enabled
-        """
-        if self.random_battles is None:
-            raise ValueError("Random battle sets are not enabled. Set enable_random_battles=True when initializing LocalDex.")
-        
-        return self.random_battles.get_generation_formats(generation)
-    
-    def clear_random_battle_cache(self) -> None:
-        """Clear random battle cache."""
-        if self.random_battles is not None:
-            self.random_battles.clear_cache()
-    
-    def cleanup_random_battle_downloads(self) -> None:
-        """Remove downloaded Pokemon Showdown repository."""
-        if self.random_battles is not None:
-            self.random_battles.cleanup_downloads()
-    
     

localdex/data/pokemon/aegislash.json

@@ -0,0 +1,88 @@
+{
+  "id": 681,
+  "name": "aegislash",
+  "types": [
+    "steel",
+    "ghost"
+  ],
+  "baseStats": {
+    "hp": 60,
+    "attack": 50,
+    "defense": 140,
+    "special_attack": 50,
+    "special_defense": 140,
+    "speed": 60
+  },
+  "height": 1.7,
+  "weight": 53.0,
+  "abilities": {
+    "0": {
+      "name": "stance-change"
+    }
+  },
+  "moves": [
+    "swords-dance",
+    "cut",
+    "tackle",
+    "hyper-beam",
+    "toxic",
+    "screech",
+    "double-team",
+    "reflect",
+    "rest",
+    "rock-slide",
+    "slash",
+    "substitute",
+    "snore",
+    "reversal",
+    "spite",
+    "protect",
+    "endure",
+    "false-swipe",
+    "swagger",
+    "fury-cutter",
+    "attract",
+    "sleep-talk",
+    "return",
+    "frustration",
+    "pursuit",
+    "hidden-power",
+    "rain-dance",
+    "sunny-day",
+    "shadow-ball",
+    "rock-smash",
+    "facade",
+    "brick-break",
+    "secret-power",
+    "metal-sound",
+    "aerial-ace",
+    "iron-defense",
+    "block",
+    "shock-wave",
+    "gyro-ball",
+    "close-combat",
+    "power-trick",
+    "magnet-rise",
+    "night-slash",
+    "air-slash",
+    "giga-impact",
+    "shadow-claw",
+    "shadow-sneak",
+    "psycho-cut",
+    "flash-cannon",
+    "iron-head",
+    "head-smash",
+    "autotomize",
+    "after-you",
+    "round",
+    "retaliate",
+    "sacred-sword",
+    "kings-shield",
+    "confide",
+    "solar-blade",
+    "laser-focus",
+    "brutal-swing",
+    "steel-beam"
+  ],
+  "generation": 6
+}

localdex/data/pokemon/basculegion.json

@@ -0,0 +1,86 @@
+{
+  "id": 902,
+  "name": "basculegion",
+  "types": [
+    "water",
+    "ghost"
+  ],
+  "baseStats": {
+    "hp": 120,
+    "attack": 112,
+    "defense": 65,
+    "special_attack": 80,
+    "special_defense": 75,
+    "speed": 78
+  },
+  "height": 3.0,
+  "weight": 110.0,
+  "abilities": {
+    "0": {
+      "name": "swift-swim"
+    },
+    "1": {
+      "name": "adaptability"
+    },
+    "H": {
+      "name": "mold-breaker"
+    }
+  },
+  "moves": [
+    "headbutt",
+    "tackle",
+    "take-down",
+    "thrash",
+    "double-edge",
+    "tail-whip",
+    "bite",
+    "water-gun",
+    "hydro-pump",
+    "surf",
+    "ice-beam",
+    "blizzard",
+    "hyper-beam",
+    "agility",
+    "night-shade",
+    "confuse-ray",
+    "waterfall",
+    "rest",
+    "substitute",
+    "flail",
+    "spite",
+    "protect",
+    "scary-face",
+    "icy-wind",
+    "outrage",
+    "sleep-talk",
+    "pain-split",
+    "rain-dance",
+    "crunch",
+    "shadow-ball",
+    "whirlpool",
+    "uproar",
+    "facade",
+    "endeavor",
+    "muddy-water",
+    "mud-shot",
+    "water-pulse",
+    "giga-impact",
+    "ice-fang",
+    "zen-headbutt",
+    "aqua-jet",
+    "head-smash",
+    "soak",
+    "hex",
+    "phantom-force",
+    "psychic-fangs",
+    "liquidation",
+    "scale-shot",
+    "flip-turn",
+    "wave-crash",
+    "tera-blast",
+    "last-respects",
+    "snowscape",
+    "chilling-water"
+  ],
+  "generation": 9
+}

localdex/data/pokemon/basculin.json

@@ -0,0 +1,105 @@
+{
+  "id": 550,
+  "name": "basculin",
+  "types": [
+    "water"
+  ],
+  "baseStats": {
+    "hp": 70,
+    "attack": 92,
+    "defense": 65,
+    "special_attack": 80,
+    "special_defense": 55,
+    "speed": 98
+  },
+  "height": 1.0,
+  "weight": 18.0,
+  "abilities": {
+    "0": {
+      "name": "reckless"
+    },
+    "1": {
+      "name": "adaptability"
+    },
+    "H": {
+      "name": "mold-breaker"
+    }
+  },
+  "moves": [
+    "cut",
+    "headbutt",
+    "tackle",
+    "take-down",
+    "thrash",
+    "double-edge",
+    "tail-whip",
+    "bite",
+    "water-gun",
+    "hydro-pump",
+    "surf",
+    "ice-beam",
+    "blizzard",
+    "bubble-beam",
+    "hyper-beam",
+    "toxic",
+    "agility",
+    "rage",
+    "double-team",
+    "waterfall",
+    "swift",
+    "rest",
+    "substitute",
+    "snore",
+    "flail",
+    "reversal",
+    "protect",
+    "scary-face",
+    "icy-wind",
+    "endure",
+    "swagger",
+    "attract",
+    "sleep-talk",
+    "return",
+    "frustration",
+    "hidden-power",
+    "rain-dance",
+    "crunch",
+    "whirlpool",
+    "uproar",
+    "hail",
+    "facade",
+    "taunt",
+    "superpower",
+    "revenge",
+    "endeavor",
+    "secret-power",
+    "dive",
+    "muddy-water",
+    "bounce",
+    "mud-shot",
+    "water-pulse",
+    "brine",
+    "assurance",
+    "aqua-tail",
+    "giga-impact",
+    "ice-fang",
+    "zen-headbutt",
+    "aqua-jet",
+    "head-smash",
+    "soak",
+    "round",
+    "chip-away",
+    "scald",
+    "final-gambit",
+    "confide",
+    "psychic-fangs",
+    "liquidation",
+    "scale-shot",
+    "flip-turn",
+    "wave-crash",
+    "tera-blast",
+    "snowscape",
+    "chilling-water"
+  ],
+  "generation": 5
+}

localdex/data/pokemon/darmanitan.json

@@ -0,0 +1,117 @@
+{
+  "id": 555,
+  "name": "darmanitan",
+  "types": [
+    "fire"
+  ],
+  "baseStats": {
+    "hp": 105,
+    "attack": 140,
+    "defense": 55,
+    "special_attack": 30,
+    "special_defense": 55,
+    "speed": 95
+  },
+  "height": 1.3,
+  "weight": 92.9,
+  "abilities": {
+    "0": {
+      "name": "sheer-force"
+    },
+    "H": {
+      "name": "zen-mode"
+    }
+  },
+  "moves": [
+    "mega-punch",
+    "fire-punch",
+    "mega-kick",
+    "headbutt",
+    "tackle",
+    "body-slam",
+    "thrash",
+    "bite",
+    "roar",
+    "ember",
+    "flamethrower",
+    "hyper-beam",
+    "strength",
+    "solar-beam",
+    "fire-spin",
+    "earthquake",
+    "dig",
+    "toxic",
+    "psychic",
+    "rage",
+    "double-team",
+    "focus-energy",
+    "fire-blast",
+    "rest",
+    "rock-slide",
+    "substitute",
+    "thief",
+    "snore",
+    "reversal",
+    "protect",
+    "belly-drum",
+    "endure",
+    "rollout",
+    "swagger",
+    "attract",
+    "sleep-talk",
+    "return",
+    "frustration",
+    "encore",
+    "hidden-power",
+    "sunny-day",
+    "future-sight",
+    "rock-smash",
+    "uproar",
+    "heat-wave",
+    "torment",
+    "will-o-wisp",
+    "facade",
+    "focus-punch",
+    "taunt",
+    "trick",
+    "superpower",
+    "brick-break",
+    "endeavor",
+    "snatch",
+    "secret-power",
+    "overheat",
+    "rock-tomb",
+    "iron-defense",
+    "bulk-up",
+    "hammer-arm",
+    "gyro-ball",
+    "u-turn",
+    "payback",
+    "fling",
+    "power-swap",
+    "guard-swap",
+    "flare-blitz",
+    "focus-blast",
+    "giga-impact",
+    "fire-fang",
+    "zen-headbutt",
+    "iron-head",
+    "stone-edge",
+    "grass-knot",
+    "smack-down",
+    "flame-charge",
+    "round",
+    "incinerate",
+    "bulldoze",
+    "work-up",
+    "confide",
+    "mystical-fire",
+    "power-up-punch",
+    "laser-focus",
+    "body-press",
+    "expanding-force",
+    "burning-jealousy",
+    "lash-out"
+  ],
+  "generation": 5
+}