codey/api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

various

Browse Source
This commit is contained in:
codey 2025-01-18 14:17:39 -05:00
parent dbaacf42a9
commit d7fef03f1b
2 changed files with 39 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
lyric_search_new/sources/cache.py
View File

@ -42,9 +42,10 @@ class Cache:
"""Get Matched Result""" """Get Matched Result"""
matched_id: int = matched_candidate[0] matched_id: int = matched_candidate[0]
if redis_results: if redis_results:
logging.info("Matched id: %s", matched_id) for res in redis_results:
for row in redis_results: (key, row) = res
if row['id'] == matched_id: if key == matched_id:
logging.info("Matched row: %s", row)
return LyricsResult( return LyricsResult(
artist=row['artist'], artist=row['artist'],
song=row['song'], song=row['song'],
@ -137,11 +138,17 @@ class Cache:
artist: str = artist.strip().lower() artist: str = artist.strip().lower()
song: str = song.strip().lower() song: str = song.strip().lower()
input_track: str = f"{artist} - {song}" input_track: str = f"{artist} - {song}"
search_query = None
search_params: Optional[tuple] = None search_params: Optional[tuple] = None
random_search: bool = False random_search: bool = False
time_start: float = time.time() time_start: float = time.time()
matcher = utils.TrackMatcher() matcher = utils.TrackMatcher()
if artist == "!" and song == "!":
random_search = True
search_query: str = 'SELECT id, artist, song, lyrics, src, confidence\
FROM lyrics ORDER BY RANDOM() LIMIT 1'
logging.info("Searching %s - %s on %s", logging.info("Searching %s - %s on %s",
artist, song, self.label) artist, song, self.label)
@ -154,11 +161,19 @@ class Cache:
if redis_result: if redis_result:
result_tracks: list = [] result_tracks: list = []
for track in redis_result: for returned in redis_result:
result_tracks.append((track['id'], f"{track['artist']} - {track['song']}")) (key, track) = returned
result_tracks.append((key, f"{track['artist']} - {track['song']}"))
if not random_search:
best_match: tuple|None = matcher.find_best_match(input_track=input_track, best_match: tuple|None = matcher.find_best_match(input_track=input_track,
candidate_tracks=result_tracks) candidate_tracks=result_tracks)
else:
best_match = (result_tracks[0], 100)
if not best_match:
return None
(candidate, confidence) = best_match (candidate, confidence) = best_match
matched = self.get_matched(redis_results=redis_result, matched_candidate=candidate, matched = self.get_matched(redis_results=redis_result, matched_candidate=candidate,
confidence=confidence) confidence=confidence)
@ -179,16 +194,13 @@ class Cache:
for ext in self.sqlite_exts: for ext in self.sqlite_exts:
await db_conn.load_extension(ext) await db_conn.load_extension(ext)
async with await db_conn.executescript(self.cache_pre_query) as _db_cursor: async with await db_conn.executescript(self.cache_pre_query) as _db_cursor:
if not random_search:
search_query: str = 'SELECT id, artist, song, lyrics, src, confidence FROM lyrics\ search_query: str = 'SELECT id, artist, song, lyrics, src, confidence FROM lyrics\
WHERE editdist3((lower(artist) || " " || lower(song)), (? || " " || ?))\ WHERE editdist3((lower(artist) || " " || lower(song)), (? || " " || ?))\
<= 410 ORDER BY editdist3((lower(artist) || " " || lower(song)), ?) ASC LIMIT 10' <= 410 ORDER BY editdist3((lower(artist) || " " || lower(song)), ?) ASC LIMIT 10'
search_params: tuple = (artist.strip(), song.strip(), search_params: tuple = (artist.strip(), song.strip(),
f"{artist.strip()} {song.strip()}") f"{artist.strip()} {song.strip()}")
if artist == "!" and song == "!":
random_search: bool = True
search_query: str = 'SELECT id, artist, song, lyrics, src, confidence\
FROM lyrics ORDER BY RANDOM() LIMIT 1'
search_params = None
async with await _db_cursor.execute(search_query, search_params) as db_cursor: async with await _db_cursor.execute(search_query, search_params) as db_cursor:
results: list = await db_cursor.fetchall() results: list = await db_cursor.fetchall()
result_tracks: list = [] result_tracks: list = []

12
lyric_search_new/sources/redis_cache.py
View File

@ -52,19 +52,25 @@ class RedisCache:
""" """
try: try:
artist = kwargs.get('artist') artist = kwargs.get('artist', '')
song = kwargs.get('song') song = kwargs.get('song', '')
lyrics = kwargs.get('lyrics') lyrics = kwargs.get('lyrics')
is_random_search = artist == "!" and song == "!"
if lyrics: if lyrics:
# to code later # to code later
raise RedisException("Lyric search not yet implemented") raise RedisException("Lyric search not yet implemented")
if not is_random_search:
search_res = await self.redis_client.ft().search( search_res = await self.redis_client.ft().search(
Query(f"@artist:{artist} @song:{song}" Query(f"@artist:{artist} @song:{song}"
)) ))
search_res_out = [dict(json.loads(result['json'])) search_res_out = [(result['id'], dict(json.loads(result['json'])))
for result in search_res.docs] for result in search_res.docs]
else:
random_redis_key = await self.redis_client.randomkey()
search_res = await self.redis_client.json().get(random_redis_key)
search_res_out = [(random_redis_key, search_res)]
return search_res_out return search_res_out
except: except: