Implement cache with fuzzy matching nicks

.gitignore

@@ -1 +1,3 @@
 python
+__pycache__
+.idea

cache.py

@@ -0,0 +1,39 @@
+import datetime
+
+import logging
+from fuzzywuzzy import fuzz
+
+
+class Cache:
+    def __init__(self, key, data, case_insensitive=True):
+        self.data_store = {}
+        self.case_insensitive = case_insensitive
+        for entity in data:
+            key_val = entity[key]
+            if not case_insensitive:
+                key_val = key_val.lower()
+            self.data_store[key_val] = entity
+        self.timestamp = datetime.datetime.now()
+
+    def get(self, key):
+        if self.case_insensitive:
+            return self.data_store[key.lower()]
+        return self.data_store[key]
+
+    def fuzzy_get(self, key):
+        try:
+            return self.get(key)
+        except KeyError:
+            pass
+        if self.case_insensitive:
+            key = key.lower()
+        fuzzes = [(data_key, fuzz.partial_ratio(data_key, key)) for data_key in self.data_store.keys()]
+        fuzzes = sorted(fuzzes, key=lambda key_ratio_pair: key_ratio_pair[1])
+        return self.data_store[fuzzes[-1][0]]
+
+    def is_valid(self):
+        if (datetime.datetime.now() - self.timestamp).total_seconds() > 30:
+            logging.debug("Cache is too old")
+            return False
+        logging.debug("Cache ok")
+        return True

main.py

@@ -15,6 +15,7 @@ def game(bot, update):
 
 
 def rank(bot, update):
+    logging.info("Rank query")
     logging.debug(update.message.text)
     try:
         nick = update.message.text.split("/rank ", 1)[1]

pip_requirements

@@ -1,2 +1,4 @@
 python-telegram-bot
 requests
+fuzzywuzzu
+python-Levenshtein

player.py

@@ -1,10 +1,14 @@
+import logging
+
 from api import API
+from cache import Cache
 
 
 class Player:
     NICK = "name"
     RANK = "rank"
     SCORE = "score"
+    _cache = None
 
     def __init__(self, nick, rank, score):
         self.nick = nick
@@ -13,12 +17,11 @@ def __init__(self, nick, rank, score):
 
     @classmethod
     def by_nick(cls, nick):
-        json_data = API.get_players()
-        player_data = None
-        for p in json_data:
-            if p[Player.NICK] == nick:
-                player_data = p
-                break
+        if Player._cache is None or not Player._cache.is_valid():
+            logging.info("Getting newer data from server")
+            json_data = API.get_players()
+            Player._cache = Cache(Player.NICK, json_data)
+        player_data = Player._cache.fuzzy_get(nick)
         return cls(player_data[Player.NICK], player_data[Player.RANK], player_data[Player.SCORE])
 
     def __str__(self):

test.py

@@ -1,4 +1,4 @@
 from player import Player
 
 
-print(Player.by_nick("T3mu"))
+print(Player.by_nick("T4,mu"))