progress -- change get to search for lrclib

2025-01-14 10:04:05 -05:00
parent 9d8d38081a
commit f86d5a4fec
6 changed files with 113 additions and 8 deletions
--- a/lyric_search_new/sources/lrclib.py
+++ b/lyric_search_new/sources/lrclib.py
@@ -0,0 +1,82 @@
+#!/usr/bin/env python3.12
+# pylint: disable=bare-except, broad-exception-caught, wrong-import-position
+
+import sys
+sys.path.insert(1,'..')
+import traceback
+from aiohttp import ClientTimeout, ClientSession
+from lyric_search_new import utils
+from lyric_search_new.constructors import LyricsResult
+from . import common
+
+class InvalidResponseException(Exception):
+    """
+    Invalid Response Exception
+    """
+
+class LRCLib:
+    """LRCLib Search Module"""
+    def __init__(self):
+        self.label = "LRCLib"
+        self.lrclib_url = "https://lrclib.net/api/get"
+        self.headers = common.SCRAPE_HEADERS
+        self.timeout = ClientTimeout(connect=2, sock_read=4)
+        self.datautils = utils.DataUtils()
+        self.matcher = utils.TrackMatcher() 
+
+    async def search(self, artist: str, song: str):
+        """
+        @artist: the artist to search
+        @song: the song to search
+        """
+        try:
+            artist = artist.strip().lower()
+            song = song.strip().lower()
+            returned_lyrics = ''
+            async with ClientSession() as client:
+                async with client.get(self.lrclib_url,
+                                      params = {
+                                          'artist_name': artist,
+                                          'track_name': song,
+                                      },
+                                      timeout=self.timeout,
+                                      headers=self.headers) as request:
+                    request.raise_for_status()
+                    text = await request.text()
+                     
+                    if len(text) < 100:
+                        raise InvalidResponseException("Search response text was invalid (len < 100 chars.)")
+                    
+                    search_data = await request.json()
+                    
+                    if not isinstance(search_data, dict):
+                        raise InvalidResponseException("Invalid JSON.")
+                     
+                    if not isinstance(search_data['artistName'], str):
+                        raise InvalidResponseException(f"Invalid JSON: Cannot find artistName key.\n{search_data}")                     
+                     
+                    if not isinstance(search_data['trackName'], str):
+                        raise InvalidResponseException(f"Invalid JSON: Cannot find trackName key.\n{search_data}")
+                     
+                    returned_artist = search_data['artistName']
+                    returned_song = search_data['trackName']
+                    returned_lyrics = search_data['plainLyrics']
+                    returned_lyrics = self.datautils.scrub_lyrics(returned_lyrics)
+                    input_track = f"{artist} - {song}"
+                    returned_track = f"{artist} - {song}"
+                    (matched, confidence) = self.matcher.find_best_match(input_track=input_track,
+                                                                        candidate_tracks=[(0, returned_track)])
+                    if not confidence:
+                        print(f"I have no confidence! {confidence}, input_track: {input_track}; candidates: {[returned_track]}")
+                        return # No suitable match found
+                    print("Returning!")
+                    return LyricsResult(artist=returned_artist,
+                                        song=returned_song,
+                                        src=self.label,
+                                        lyrics=returned_lyrics,
+                                        confidence=confidence)
+        except:
+            traceback.print_exc()
+            return
+                
+