localdex 0.2.37__py3-none-any.whl → 0.2.39__py3-none-any.whl

localdex/core.py

@@ -18,6 +18,7 @@ from .exceptions import (
     ItemNotFoundError, DataLoadError, SearchError
 )
 from .data_loader import DataLoader
+from .name_normalizer import PokemonNameNormalizer
 
 
 class LocalDex:
@@ -169,63 +170,10 @@ class LocalDex:
         # Load from data
         pokemon_data = self.data_loader.load_pokemon_by_name(name)
         if not pokemon_data:
-
-            if ' ' in name:
-                name = name.replace(' ', '-')
-
-            if '-f' in name or '-m' in name:
-                name = name.replace('-f', '-female').replace('-m', '-male')
-
-            if 'florges' in name.lower():
-                name = 'florges'
-            
-            if 'maushold' in name.lower():
-                name = 'maushold'
-            
-            if 'pikachu' in name.lower():
-                name = 'pikachu'
-
-            if 'ogerpon' in name.lower():
-                if 'cornerstone' in name.lower():
-                    name = 'ogerpon-cornerstone-mask'
-                elif 'wellspring' in name.lower():
-                    name = 'ogerpon-wellspring-mask'
-                elif 'hearthflame' in name.lower():
-                    name = 'ogerpon-hearthflame-mask'
-                else:
-                    name = 'ogerpon'
+            # Try with normalized name
+            normalized_name = PokemonNameNormalizer.normalize_name(name)
+            pokemon_data = self.data_loader.load_pokemon_by_name(normalized_name)
             
-            if 'squawkabilly' in name.lower():
-                if 'blue' in name.lower():
-                    name = 'squawkabilly-blue-plumage'
-                elif 'red' in name.lower():
-                    name = 'squawkabilly-red-plumage'
-                elif 'white' in name.lower():
-                    name = 'squawkabilly-white-plumage'
-                else:
-                    name = 'squawkabilly-white-plumage'
-            if 'tauros' in name.lower() and 'paldea' in name.lower():
-                if 'blaze' in name.lower():
-                    name = 'tauros-paldea-blaze-breed'
-                elif 'aqua' in name.lower():
-                    name = 'tauros-paldea-aqua-breed'
-                elif 'combat' in name.lower():
-                    name = 'tauros-paldea-combat-breed'
-                else:
-                    name = 'tauros-paldea-blaze-breed'
-            if 'arceus' in name.lower():
-                name = 'arceus'
-
-            if 'indeedee' in name.lower():
-                if 'f' in name.lower():
-                    name = 'indeedee-female'
-                elif 'm' in name.lower():
-                    name = 'indeedee-male'
-                else:
-                    name = 'indeedee-female'
-
-            pokemon_data = self.data_loader.load_pokemon_by_name(name)
-
             if not pokemon_data:
                 raise PokemonNotFoundError(name)
         

localdex/name_normalizer.py

@@ -0,0 +1,142 @@
+"""
+Pokemon name normalization and fuzzing utilities.
+
+This module handles the normalization of Pokemon names to match the expected
+data format, including handling special cases, forms, and variations.
+"""
+
+from typing import Optional
+
+
+class PokemonNameNormalizer:
+    """
+    Handles normalization of Pokemon names for data lookup.
+    
+    This class contains logic to handle various Pokemon name variations,
+    special forms, and edge cases to ensure proper data retrieval.
+    """
+    
+    @staticmethod
+    def normalize_name(name: str) -> str:
+        """
+        Normalize a Pokemon name for data lookup.
+        
+        Args:
+            name: The input Pokemon name
+            
+        Returns:
+            Normalized name that should match the data format
+        """
+        # Handle spaces and hyphens
+        if ' ' in name:
+            name = name.replace(' ', '-')
+
+        
+
+        # Handle special cases
+        name = PokemonNameNormalizer._handle_special_cases(name)
+        
+        # Convert to lowercase at the end
+        return name.lower()
+    
+    @staticmethod
+    def _handle_special_cases(name: str) -> str:
+        """
+        Handle special Pokemon name cases and forms.
+        
+        Args:
+            name: The name to process
+            
+        Returns:
+            Processed name
+        """
+        name_lower = name.lower()
+        if 'urshifu' in name_lower:
+            if 'single' in name_lower:
+                return 'urshifu-single-strike'
+            elif 'rapid' in name_lower:
+                return 'urshifu-rapid-strike'
+            else:
+                return 'urshifu-single-strike'
+        
+        if 'zacian' in name_lower:
+            if 'crowned' in name_lower:
+                return 'zacian-crowned'
+            else:
+                return 'zacian'
+        
+        if 'zamzenta' in name_lower:
+            if 'crowned' in name_lower:
+                return 'zamzenta-crowned'
+            else:
+                return 'zamzenta'
+        
+        if 'necrozma' in name_lower:
+            if 'dawn' in name_lower:
+                return 'necrozma-dawn'
+            elif 'dusk' in name_lower:
+                return 'necrozma-dusk'
+            elif 'ultra' in name_lower:
+                return 'necrozma-ultra'
+            else:
+                return 'necrozma'
+
+        # Florges forms
+        if 'florges' in name_lower:
+            return 'florges'
+        
+        # Maushold forms
+        if 'maushold' in name_lower:
+            return 'maushold'
+        
+        # Pikachu forms
+        if 'pikachu' in name_lower:
+            return 'pikachu'
+        
+        # Ogerpon forms
+        if 'ogerpon' in name_lower:
+            if 'cornerstone' in name_lower:
+                return 'ogerpon-cornerstone-mask'
+            elif 'wellspring' in name_lower:
+                return 'ogerpon-wellspring-mask'
+            elif 'hearthflame' in name_lower:
+                return 'ogerpon-hearthflame-mask'
+            else:
+                return 'ogerpon'
+        
+        # Squawkabilly forms
+        if 'squawkabilly' in name_lower:
+            if 'blue' in name_lower:
+                return 'squawkabilly-blue-plumage'
+            elif 'red' in name_lower:
+                return 'squawkabilly-red-plumage'
+            elif 'white' in name_lower:
+                return 'squawkabilly-white-plumage'
+            else:
+                return 'squawkabilly-white-plumage'
+        
+        # Tauros Paldea forms
+        if 'tauros' in name_lower and 'paldea' in name_lower:
+            if 'blaze' in name_lower:
+                return 'tauros-paldea-blaze-breed'
+            elif 'aqua' in name_lower:
+                return 'tauros-paldea-aqua-breed'
+            elif 'combat' in name_lower:
+                return 'tauros-paldea-combat-breed'
+            else:
+                return 'tauros-paldea-blaze-breed'
+        
+        # Arceus forms
+        if 'arceus' in name_lower:
+            return 'arceus'
+        
+        # Indeedee forms
+        if 'indeedee' in name_lower:
+            if 'f' in name_lower:
+                return 'indeedee-female'
+            elif 'm' in name_lower:
+                return 'indeedee-male'
+            else:
+                return 'indeedee-female'
+        
+        return name 

{localdex-0.2.37.dist-info → localdex-0.2.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: localdex
-Version: 0.2.37
+Version: 0.2.39
 Summary: A local Pokemon data repository/Pokedex with fast offline access
 Author-email: LocalDex Team <localdex@example.com>
 License: MIT

{localdex-0.2.37.dist-info → localdex-0.2.39.dist-info}/RECORD

@@ -1,9 +1,10 @@
 localdex/__init__.py,sha256=TTeh9Eys8nLtcw7Dg2equDqnouthIUyfsqvABtzR2vA,403
 localdex/cli.py,sha256=WqBCyA0akAFJNrYa8jCA3zoZgYEptDYDMGfAGkvKyZc,19317
-localdex/core.py,sha256=OTM3UmswgE9-jYRJWSD-mtKLGW5XuqZt-n1CJyvw4L0,22400
+localdex/core.py,sha256=0kjuL388OniX7hRG6c5fT72nCC76LolpAB2UFrNXPns,20480
 localdex/data_loader.py,sha256=hi9aSTto5Ti-OBGOgrQ-XwD5hmivsUwS1uC4rulhwQI,11366
 localdex/download_data.py,sha256=T4A3OliGe2RsrRV1glyCqRcFhQIRt29KcY1K1-IqTuU,21597
 localdex/exceptions.py,sha256=Z02-8Kci6jFDk2nnGdVSHZJMDDWE9vuwuASs4VM3To8,2777
+localdex/name_normalizer.py,sha256=36PK1txN3P8CGUUIr_tGkKD1yLLMbCmfGmryCdwYblI,4285
 localdex/data/abilities/adaptability.json,sha256=FGEEZmL80YVcIXHV-h287Wu-UJycM1QEJxiOHK2I4mY,289
 localdex/data/abilities/aerilate.json,sha256=KRZOVH2KGdEQ_3UktFoXoagUOaC8jIPZ6Ti-YGzW44s,301
 localdex/data/abilities/aftermath.json,sha256=Awv_8jquu3pM4CemD5ylS1v83ABrqGtrgMEkL-1ldBk,285
@@ -4825,8 +4826,8 @@ localdex/models/ability.py,sha256=AQzv3XUHHl4sustMJjPDDjJOjXu2GMLTfcM3-tqQ_1w,30
 localdex/models/item.py,sha256=zXao8F-jBPUGq_YLeGeYeK_dZVI7aZMXtWOPwR3qusY,4677
 localdex/models/move.py,sha256=hfgcWI4ziz5MMvc9ddmkotxzYYdrSUqZZQ72IU5tucs,7629
 localdex/models/pokemon.py,sha256=glnOiEhkFc25U54hJHwUkT-okUgtL2prg3269YXI3w0,5422
-localdex-0.2.37.dist-info/METADATA,sha256=_XTETQDl9yYeVkxFrHxykvnIu-sTdXyeWMwEDDq83z8,8016
-localdex-0.2.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-localdex-0.2.37.dist-info/entry_points.txt,sha256=n5GxSeQo-MRuvrT2wVk7hOzEFFsWf6tkBjkzmGIYJe4,47
-localdex-0.2.37.dist-info/top_level.txt,sha256=vtupDMH-IaxVCoEZrmE0QzdTwhaKzngVJbTA1NkR_MY,9
-localdex-0.2.37.dist-info/RECORD,,
+localdex-0.2.39.dist-info/METADATA,sha256=uQCeAUZtWDcb_vVwiH4zeer0h9_TrXpVVzIo9RCR7rE,8016
+localdex-0.2.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+localdex-0.2.39.dist-info/entry_points.txt,sha256=n5GxSeQo-MRuvrT2wVk7hOzEFFsWf6tkBjkzmGIYJe4,47
+localdex-0.2.39.dist-info/top_level.txt,sha256=vtupDMH-IaxVCoEZrmE0QzdTwhaKzngVJbTA1NkR_MY,9
+localdex-0.2.39.dist-info/RECORD,,