resolves #22, #29

lyric_search_new/sources/lrclib.py

@@ -2,6 +2,7 @@
 # pylint: disable=bare-except, broad-exception-caught, wrong-import-position
 
 import sys
+import time
 sys.path.insert(1,'..')
 import traceback
 import logging
@@ -10,6 +11,7 @@ from aiohttp import ClientTimeout, ClientSession
 from lyric_search_new import utils
 from lyric_search_new.constructors import LyricsResult
 from . import common
+from . import cache
 
 logger = logging.getLogger()
 log_level = logging.getLevelName(logger.level)
@@ -28,6 +30,7 @@ class LRCLib:
         self.timeout = ClientTimeout(connect=2, sock_read=4)
         self.datautils = utils.DataUtils()
         self.matcher = utils.TrackMatcher() 
+        self.cache = cache.Cache()
 
     async def search(self, artist: str, song: str) -> Optional[LyricsResult]:
         """
@@ -37,6 +40,7 @@ class LRCLib:
         try:          
             artist: str = artist.strip().lower()
             song: str = song.strip().lower()
+            time_start: float = time.time()
             logging.info("Searching %s - %s on %s",
                          artist, song, self.label) 
                 
@@ -91,11 +95,16 @@ class LRCLib:
                     if not confidence:
                         return # No suitable match found
                     logging.info("Result found on %s", self.label)
-                    return LyricsResult(artist=returned_artist,
+                    time_end: float = time.time()
+                    time_diff: float = time_end - time_start
+                    matched = LyricsResult(artist=returned_artist,
                                         song=returned_song,
                                         src=self.label,
                                         lyrics=returned_lyrics,
-                                        confidence=confidence)
+                                        confidence=confidence,
+                                        time=time_diff)
+                    await self.cache.store(matched)
+                    return matched
         except:
             if log_level == "DEBUG":
                 traceback.print_exc()