feat: add look_up dict and use caching

PabloSanchi · PabloSanchi · commit 5ff45710c947 · 2024-01-04T15:20:34.000+01:00
diff --git a/core/range_mapper.py b/core/range_mapper.py
@@ -7,22 +7,23 @@
 class RangeMapper(RangeBase):
     src_range: List[str]
     to_range: List[str]
-    
-    
+    _lookup_dict: dict = field(default_factory=dict, init=False)
+    _cache: dict = field(default_factory=dict, init=False)
+
+
+    def __post_init__(self):
+        self._lookup_dict = {char: i for i, char in enumerate(self.src_range)}
+        self._cache = {}
+
     def execute(self, src_string: str) -> str:
-        result = ''
-        for src_char in src_string:
-            index = self._find_index(self.src_range, src_char)
-            result += self._get_char_from_index(self.to_range, index) if index > -1 else src_char
-            
-        return result
+        return ''.join(self.map_char(char) for char in  src_string)
+        
+    def map_char(self, src_char):
+        if src_char in self._cache:
+            return self._cache[src_char]
+        
+        index = self._lookup_dict.get(src_char, -1)
+        return self._get_char_from_index(self.to_range, index) if index > -1 else src_char
             
-    def _find_index(self, range_to_find: List[str], char: str) -> int:
-        try:
-            reverse_index = range_to_find[::-1].index(char)
-            return len(range_to_find) - 1 - reverse_index
-        except ValueError:
-            return -1
-    
     def _get_char_from_index(self, range_to_get: List[str], index: int) -> str:
         return range_to_get[index] if index < len(range_to_get) else range_to_get[-1]
