qnty 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

qnty/utils/unit_suggestions.py

@@ -0,0 +1,198 @@
+"""
+Unit Suggestions System
+======================
+
+Provides fuzzy string matching for unit validation errors with intelligent recommendations.
+"""
+
+from __future__ import annotations
+
+import json
+from difflib import SequenceMatcher
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..units.registry import Registry
+
+
+class UnitSuggester:
+    """Provides intelligent suggestions for invalid unit strings."""
+    
+    __slots__ = ("_unit_names", "_unit_aliases", "_all_suggestions", "_loaded")
+    
+    def __init__(self):
+        self._unit_names: list[str] = []
+        self._unit_aliases: dict[str, str] = {}  # alias -> canonical_name
+        self._all_suggestions: list[str] = []  # All searchable strings
+        self._loaded = False
+    
+    def _load_units(self) -> None:
+        """Load unit names and aliases from the unit_data.json file."""
+        if self._loaded:
+            return
+            
+        try:
+            # Load from the unit data file used by code generation
+            unit_data_path = Path(__file__).parent.parent.parent.parent / "codegen" / "generators" / "data" / "unit_data.json"
+            
+            if not unit_data_path.exists():
+                # Fallback: try to load from registry if available
+                self._load_from_registry()
+                return
+                
+            with open(unit_data_path, 'r', encoding='utf-8') as f:
+                unit_data = json.load(f)
+            
+            # Extract all unit names and aliases
+            for field_data in unit_data.values():
+                if not isinstance(field_data, dict) or 'units' not in field_data:
+                    continue
+                    
+                for unit in field_data['units']:
+                    # Add normalized name
+                    unit_name = unit.get('normalized_name', '')
+                    if unit_name:
+                        self._unit_names.append(unit_name)
+                        self._all_suggestions.append(unit_name)
+                    
+                    # Add notation as an alias
+                    notation = unit.get('notation', '')
+                    if notation and notation != unit_name:
+                        # Clean up notation (remove LaTeX formatting)
+                        clean_notation = self._clean_notation(notation)
+                        if clean_notation and clean_notation not in self._unit_aliases:
+                            self._unit_aliases[clean_notation] = unit_name
+                            self._all_suggestions.append(clean_notation)
+                    
+                    # Add aliases
+                    aliases = unit.get('aliases', [])
+                    for alias in aliases:
+                        if alias and alias not in self._unit_aliases:
+                            self._unit_aliases[alias] = unit_name
+                            self._all_suggestions.append(alias)
+            
+            self._loaded = True
+            
+        except Exception:
+            # Fallback to registry-based loading
+            self._load_from_registry()
+    
+    def _load_from_registry(self) -> None:
+        """Fallback: load units from registry if available."""
+        try:
+            from ..units.registry import registry
+            
+            if hasattr(registry, 'units'):
+                for unit_name in registry.units.keys():
+                    self._unit_names.append(unit_name)
+                    self._all_suggestions.append(unit_name)
+            
+            self._loaded = True
+        except Exception:
+            # If all else fails, just mark as loaded with empty data
+            self._loaded = True
+    
+    def _clean_notation(self, notation: str) -> str:
+        """Clean LaTeX and special formatting from notation strings."""
+        # Remove common LaTeX patterns
+        notation = notation.replace('\\mathrm{', '').replace('}', '')
+        notation = notation.replace('\\text{', '').replace('$', '')
+        notation = notation.replace('\\', '')
+        notation = notation.replace('{', '').replace('}', '')
+        
+        # Remove extra spaces and common patterns
+        notation = notation.strip()
+        notation = notation.replace(' ', '')
+        
+        # Skip if too long or contains special characters that make it unusable
+        if len(notation) > 20 or any(char in notation for char in ['(', ')', '^', '_', '/']):
+            return ''
+            
+        return notation if notation and len(notation) <= 10 else ''
+    
+    def get_suggestions(self, invalid_unit: str, max_suggestions: int = 3) -> list[str]:
+        """
+        Get fuzzy string matching suggestions for an invalid unit.
+        
+        Args:
+            invalid_unit: The invalid unit string that was entered
+            max_suggestions: Maximum number of suggestions to return
+            
+        Returns:
+            List of suggested unit strings, ordered by similarity
+        """
+        self._load_units()
+        
+        if not self._all_suggestions:
+            return []
+        
+        # Calculate similarity scores for all units
+        similarities = []
+        invalid_lower = invalid_unit.lower().strip()
+        
+        for suggestion in self._all_suggestions:
+            suggestion_lower = suggestion.lower()
+            
+            # Exact match (shouldn't happen, but just in case)
+            if invalid_lower == suggestion_lower:
+                continue
+                
+            # Calculate similarity using SequenceMatcher
+            similarity = SequenceMatcher(None, invalid_lower, suggestion_lower).ratio()
+            
+            # Bonus for starts-with matches
+            if suggestion_lower.startswith(invalid_lower) or invalid_lower.startswith(suggestion_lower):
+                similarity += 0.2
+            
+            # Bonus for contains matches
+            if invalid_lower in suggestion_lower or suggestion_lower in invalid_lower:
+                similarity += 0.1
+            
+            similarities.append((similarity, suggestion))
+        
+        # Sort by similarity score (descending) and return top matches
+        similarities.sort(key=lambda x: x[0], reverse=True)
+        
+        # Filter for meaningful suggestions (similarity > 0.4)
+        meaningful_suggestions = [
+            suggestion for similarity, suggestion in similarities 
+            if similarity > 0.4
+        ]
+        
+        return meaningful_suggestions[:max_suggestions]
+
+
+class UnitValidationError(ValueError):
+    """Error raised when an invalid unit is provided, with suggestions."""
+    
+    def __init__(self, invalid_unit: str, variable_type: str = "", suggestions: list[str] | None = None):
+        self.invalid_unit = invalid_unit
+        self.variable_type = variable_type
+        self.suggestions = suggestions or []
+        
+        # Construct error message
+        if variable_type:
+            msg = f"Unknown unit '{invalid_unit}' for {variable_type}"
+        else:
+            msg = f"Unknown unit '{invalid_unit}'"
+        
+        if self.suggestions:
+            msg += f". Did you mean: {', '.join(repr(s) for s in self.suggestions)}?"
+        
+        super().__init__(msg)
+
+
+# Global suggester instance
+_suggester = UnitSuggester()
+
+
+def get_unit_suggestions(invalid_unit: str, max_suggestions: int = 3) -> list[str]:
+    """Get unit suggestions for an invalid unit string."""
+    return _suggester.get_suggestions(invalid_unit, max_suggestions)
+
+
+def create_unit_validation_error(invalid_unit: str, variable_type: str = "") -> UnitValidationError:
+    """Create a UnitValidationError with intelligent suggestions."""
+    suggestions = get_unit_suggestions(invalid_unit)
+    return UnitValidationError(invalid_unit, variable_type, suggestions)

{qnty-0.1.3.dist-info → qnty-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qnty
-Version: 0.1.3
+Version: 0.1.4
 Summary: High-performance unit system library for Python with dimensional safety and fast unit conversions
 License: Apache-2.0
 Keywords: units,dimensional analysis,engineering,physics,quantities,measurements

{qnty-0.1.3.dist-info → qnty-0.1.4.dist-info}/RECORD

@@ -9,12 +9,12 @@ qnty/dimensions/field_dims.py,sha256=bcPu1xxPhjoNjc7TxyP_B4xKDLHKGdtNne-sCB9hz-8
 qnty/dimensions/field_dims.pyi,sha256=lk3YrH3Ovs3CJCZe5MfX334kdpmsfEql4D3fLKjuYDs,4575
 qnty/dimensions/signature.py,sha256=yk7QGejAV-TEPTqWE1Q5yV2sZA-RWGiK_rHiMT0Q2yU,4173
 qnty/equations/__init__.py,sha256=Ou5H6tTFXgVw16JYan_a4653NxroBxcnTY6YWt380Qo,108
-qnty/equations/equation.py,sha256=6Ot3-XhSFyxdv3hUwJvdTvB2BGZlAoXX46uok_6q-14,9041
+qnty/equations/equation.py,sha256=EPw1mKTMriI94uKCqSWTPvQfcZTBLPhX-D4nDZI9vJg,10580
 qnty/equations/system.py,sha256=vMoD1iTUrAHnVFVvCUKeyNfSBfMiqpwQbfDx46kN9N8,5155
 qnty/expressions/__init__.py,sha256=DA2s7DBhVCmdUgsYSTJWObsp2DbbpFn492yr1nUTg2g,930
 qnty/expressions/formatter.py,sha256=yLGLwLYjhBvVi2Q6rfkg8pbyH0-a1Ko0AYLsqJTJf50,7806
 qnty/expressions/functions.py,sha256=ek43udfUDpThKo38rVPBYPvKfZNc9Bbs8RuL-CvQc_A,2729
-qnty/expressions/nodes.py,sha256=oqMFfMeeWhccv4KBHNI5qYBz2pSe6jD-OwE4P2HyG1A,28644
+qnty/expressions/nodes.py,sha256=Q6DgmhP7MVUkdsUpzzp7_yjV8hegd0AwETvd4bo8nWc,29362
 qnty/expressions/types.py,sha256=eoM-IqY-k-IypRHAlRwjEtMmB6DiwX7YGot8t_vGw3o,1729
 qnty/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qnty/extensions/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,18 +22,18 @@ qnty/extensions/plotting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 qnty/extensions/reporting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qnty/problems/__init__.py,sha256=g7zuml2IecoSwgzX6r1zZ5SlmBKFc8qqTR_cao037pc,4808
 qnty/problems/composition.py,sha256=HI5ffsF14IXRuqjM0z_O0kkpme9fPnJVdz6m4rDCJsM,44916
-qnty/problems/problem.py,sha256=8dg_AwX3lgN2rpa33RhhmA-AXbPg8lmSBJzeB3iXBMk,31565
+qnty/problems/problem.py,sha256=xnkHsI9OcY3HJ0UxtMnMQrS6n7accl4d-5fSJQ6ITPo,31721
 qnty/problems/rules.py,sha256=NwIStAa8bocVtvzAsnPmRdC_0ENTJWyXLOoYBnkvpPA,5176
 qnty/problems/solving.py,sha256=LTI8F9ujDiSqXE9Aiz8sOgaGJNX9p7oaR5CQIZHpCY8,44315
 qnty/problems/validation.py,sha256=SmFEsgHx5XwRNlR2suOhxO-WNsOwPZhCP8wyVKYo1EE,4826
 qnty/quantities/__init__.py,sha256=K_h5v6X6-OyITSXOhbIZTDAJe6-y_7iMMDEIQ4O3luc,809
-qnty/quantities/base_qnty.py,sha256=QasOR4-a7gwPBvc6cLJ3ooQHmOcWbYexgtNQ9I6bXI8,32516
-qnty/quantities/field_converters.py,sha256=rDWttIE0lwF1doGlLG5RJTcTikYEYruMrRBMlr8fvBI,1008701
-qnty/quantities/field_qnty.py,sha256=ZUZ8N102WqVW0r3JZtSCbSSrP1co5r_swOBNTJJCNTk,43053
+qnty/quantities/base_qnty.py,sha256=u-1kzSAup47SUMjYVvnFyfVGk6hW8eUCcOtgUeFaMtM,33776
+qnty/quantities/field_converters.py,sha256=GNG1fgd4odagfAEtmhSHwG9DvX8DL4cq8DHlRYwaw3U,1008869
+qnty/quantities/field_qnty.py,sha256=YZ0Gd2r5tz55E18a1_ItHke7P7VJZ15_fhI6DLoFxAI,43938
 qnty/quantities/field_setter.py,sha256=JCvRom4qvCYvgRNgFLZEuwb1PCmz0qrKzxDv0-h4RMo,449348
 qnty/quantities/field_setter.pyi,sha256=NDsOvngsfEaXczZ56CcxlCOsLFGpXbEhxaBq1qW80Us,140812
-qnty/quantities/field_vars.py,sha256=-wZ91NHxgRg8jJnSbggaWphRICjuqKPr6B7PVfIjk4Y,401188
-qnty/quantities/field_vars.pyi,sha256=nTrhUI1U_CFeVzAuYlHgp2supUvkkb0EAM-74kiJ7L4,145467
+qnty/quantities/field_vars.py,sha256=Zhc7PoCqtqfmaAw-MPMiwBSPRxTcLruzo-9Krusdbsw,517162
+qnty/quantities/field_vars.pyi,sha256=_O5QaeclK5sgVYUlTLFyX4whCtcNIEBk5WVd0DbAxss,146420
 qnty/solving/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qnty/solving/manager.py,sha256=LQBMhWD3ajRYMBXkwRpkVzdo7qVEDviBAoHpjAzS-0U,3945
 qnty/solving/order.py,sha256=q1G3fMWamhmBK6vN0L2BuTqWl0aa94ZJPCWusrntcXo,15488
@@ -56,6 +56,7 @@ qnty/utils/error_handling/handlers.py,sha256=_q12co-jr4YSktRoCPpGBbh6WXEDw9MbmWx
 qnty/utils/logging.py,sha256=2H6_gSOQjxdK5024XTY3E1jGIQPE8WdalVhVBFw51OA,1143
 qnty/utils/protocols.py,sha256=c_Ya_epCm7qenAADRMZiwiQ0PdD-Z4T85b1z1YQNXAk,5247
 qnty/utils/scope_discovery.py,sha256=mQc-FHJ5-VNBzqQwiFofV-hqeF3GpLRaLlTjYDRnOqs,15184
-qnty-0.1.3.dist-info/METADATA,sha256=bHbY7G0DtCTsP3buOzzTusq8SEGEMwuwii6_7VAQhU4,6761
-qnty-0.1.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-qnty-0.1.3.dist-info/RECORD,,
+qnty/utils/unit_suggestions.py,sha256=V_eNGYIXayyHY7bi4tA_bDuiNwKlkbV8R2OgMMGLk_w,7774
+qnty-0.1.4.dist-info/METADATA,sha256=55oIRTrDystUWnodxecG8Gbf0zx42u5wPuR00JSET_Q,6761
+qnty-0.1.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+qnty-0.1.4.dist-info/RECORD,,