Pythonic implementation of LRU Cache

other/lru_cache_pythonic.py

@@ -0,0 +1,86 @@
+from typing import Any, Hashable
+
+'''
+The following implementation of LRU Cache is one of the most elegant pythonic implementations. 
+It only uses the in-built python dictionary (https://docs.python.org/3/library/stdtypes.html#typesmapping). This works because 
+the Python dictionary maintains the order of insertion of keys and ensures O(1) operations on insert, delete and access.
+'''
+class LRUCache(dict):
+    '''
+    Initialize an LRU Cache with given capacity.
+    capacity : int -> the capacity of the LRU Cache
+    '''
+    def __init__(self, capacity : int):
+        self.remaining:int = capacity
+
+    '''
+    This method gets the value associated with the key.
+    key : Hashable -> a hashable object that is mapped to a value inside of the LRU cache.
+    returns -> value : Any -> any object that is stored as a value inside of the LRU cache.
+
+    >>> cache = LRUCache(2)
+    >>> cache.put(1,1)
+    >>> cache.get(1)
+    1
+    >>> cache.get(2)
+    KeyError: '2 not found.'
+    '''
+    def get(self, key:Hashable)->Any:
+        if key not in self:
+            raise KeyError(f"{key} not found.")
+        val = self.pop(key) # Pop the key-value and re-insert to maintain the order
+        self[key] = val
+        return val
+
+    '''
+    This method puts the value associated with the key provided inside of the LRU cache.
+    key : Hashable -> a hashable object that is mapped to a value inside of the LRU cache.
+    value: Any -> any object that is to be associated with the key inside of the LRU cache.
+    '''
+    def put(self, key:Hashable, value:Any):
+        # To pop the last value inside of the LRU cache
+        if key in self:
+            self.pop(key)
+            self[key] = value
+            return
+
+        if self.remaining > 0: self.remaining -= 1
+        # To pop the least recently used item from the dictionary 
+        else: self.pop(next(iter(self)))
+        self[key] = value
+
+def main():
+    '''Example test case with LRU_Cache of size 2'''
+    cache = LRUCache(2) # Creates an LRU cache with size 2
+    cache.put(1,1) # cache = {1:1}
+    cache.put(2,2) # cache = {1:1, 2:2}
+    try:
+        print(cache.get(1)) # Prints 1
+    except KeyError:
+        print("Key not found in cache")
+    cache.put(3,3) # cache = {1:1, 3:3} key=2 is evicted because it wasn't used recently
+    try:
+        print(cache.get(2)) 
+    except KeyError:
+        print("Key=2 not found in cache") # Prints key not found
+    cache.put(4,4) # cache = {4:4, 3:3} key=1 is evicted because it wasn't used recently
+    try:
+        print(cache.get(1)) 
+    except KeyError:
+        print("Key=1 not found in cache") # Prints key not found
+    try:
+        print(cache.get(3)) # Prints value 3
+    except KeyError:
+        print("Key not found in cache")
+
+    try:
+        print(cache.get(4)) # Prints value 4
+    except KeyError:
+        print("Key not found in cache")
+
+    import doctest
+    doctest.testmod()
+
+if __name__ == '__main__':
+    main()
+