localdex 0.1.20__py3-none-any.whl → 0.1.22__py3-none-any.whl

localdex/__init__.py

@@ -2,8 +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
-from .sprite_downloader import SpriteDownloader
 
 __all__ = [
     "LocalDex",
@@ -16,6 +14,4 @@ __all__ = [
     "MoveNotFoundError", 
     "AbilityNotFoundError",
     "ItemNotFoundError",
-    "RandomBattleSets",
-    "SpriteDownloader",
 ] 

localdex/core.py

@@ -18,8 +18,6 @@ from .exceptions import (
     ItemNotFoundError, DataLoadError, SearchError
 )
 from .data_loader import DataLoader
-from .random_battles import RandomBattleSets
-from .sprite_downloader import SpriteDownloader
 
 
 class LocalDex:
@@ -31,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, enable_sprite_downloader: bool = True, sprite_max_workers: int = 8):
+    def __init__(self, data_path: Optional[str] = None, data_dir: Optional[str] = None, enable_caching: bool = True):
         """
         Initialize the LocalDex.
         
@@ -39,9 +37,6 @@ 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.
-            enable_sprite_downloader: Whether to enable sprite downloader functionality.
-            sprite_max_workers: Maximum number of worker threads for parallel sprite processing.
         """
         # Use data_dir if provided, otherwise use data_path
         final_data_path = data_dir if data_dir is not None else data_path
@@ -49,12 +44,6 @@ class LocalDex:
         self.data_dir = final_data_path  # Store for backward compatibility
         self.enable_caching = enable_caching
         
-        # Sprite downloader integration
-        self.sprite_downloader = SpriteDownloader(final_data_path, max_workers=sprite_max_workers) if enable_sprite_downloader else None
-        
-        # Initialize random battle sets
-        self.random_battles = RandomBattleSets() if enable_random_battles else None
-        
         # Initialize caches
         self._pokemon_cache: Dict[str, Pokemon] = {}
         self._pokemon_id_cache: Dict[int, Pokemon] = {}
@@ -569,176 +558,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()
-    
-    def download_all_sprites(self):
-        """
-        Download and extract all sprites (Pokemon and items).
-        """
-        if not self.sprite_downloader:
-            raise RuntimeError("SpriteDownloader is not enabled.")
-        return self.sprite_downloader.download_all_sprites()
-
-    def get_pokemon_sprite_path(self, pokemon_name: str):
-        """
-        Get the path to a Pokemon sprite image.
-        """
-        if not self.sprite_downloader:
-            raise RuntimeError("SpriteDownloader is not enabled.")
-        return self.sprite_downloader.get_pokemon_sprite_path(pokemon_name)
-
-    def get_item_sprite_path(self, item_name: str):
-        """
-        Get the path to an item sprite image.
-        """
-        if not self.sprite_downloader:
-            raise RuntimeError("SpriteDownloader is not enabled.")
-        return self.sprite_downloader.get_item_sprite_path(item_name)
-
-    def get_sprite_metadata(self, sprite_type: str = "pokemon"):
-        """
-        Get sprite metadata for either 'pokemon' or 'items'.
-        """
-        if not self.sprite_downloader:
-            raise RuntimeError("SpriteDownloader is not enabled.")
-        return self.sprite_downloader.get_sprite_metadata(sprite_type)
-
-    def list_available_sprites(self, sprite_type: str = "pokemon"):
-        """
-        List all available sprites of a given type ('pokemon' or 'items').
-        """
-        if not self.sprite_downloader:
-            raise RuntimeError("SpriteDownloader is not enabled.")
-        return self.sprite_downloader.list_available_sprites(sprite_type) 
+    

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
+}