From 334da4bd178855752039cb7b07ce779754a33719 Mon Sep 17 00:00:00 2001 From: Tanner Collin Date: Sat, 7 Feb 2026 12:10:37 -0700 Subject: [PATCH] refactor: Abstract Navidrome Subsonic API calls and flatten logic Co-authored-by: aider (gemini/gemini-2.5-pro) --- main.py | 80 ++++++++++++++++++++++----------------------------------- 1 file changed, 30 insertions(+), 50 deletions(-) diff --git a/main.py b/main.py index 4d03f5c..a81dde8 100644 --- a/main.py +++ b/main.py @@ -46,9 +46,9 @@ def parse_log_line(line): return timestamp, song_id, is_starred -async def navidrome_get_song_details(song_id): +async def _call_subsonic_api(endpoint, **kwargs): """ - Gets song details from Navidrome's Subsonic API. + A generic helper to call the Navidrome Subsonic API. """ navidrome_url = settings.NAVIDROME_URL username = settings.NAVIDROME_USER @@ -77,10 +77,10 @@ async def navidrome_get_song_details(song_id): 'v': '1.16.1', 'c': 'heart-monitor', 'f': 'json', - 'id': song_id, } + params.update(kwargs) - api_url = f"{navidrome_url.rstrip('/')}/rest/getSong" + api_url = f"{navidrome_url.rstrip('/')}/rest/{endpoint}" try: async with aiohttp.ClientSession() as session: @@ -88,61 +88,41 @@ async def navidrome_get_song_details(song_id): response.raise_for_status() data = await response.json() except aiohttp.ClientError as e: - logging.error(f"Error calling Navidrome getSong API for song {song_id}: {e}") + logging.error(f"Error calling Navidrome API endpoint {endpoint} for song {kwargs.get('id')}: {e}") return None if data.get('subsonic-response', {}).get('status') != 'ok': - logging.error(f"Failed to get song details for song {song_id}. Response: {data}") + logging.error(f"Failed to call Navidrome API endpoint {endpoint} for song {kwargs.get('id')}. Response: {data}") return None - return data.get('subsonic-response', {}).get('song') + return data.get('subsonic-response', {}) + + +async def navidrome_get_song_details(song_id): + """ + Gets song details from Navidrome's Subsonic API. + """ + response = await _call_subsonic_api('getSong', id=song_id) + if response: + return response.get('song') + return None async def navidrome_set_rating(song_id): - navidrome_url = settings.NAVIDROME_URL - username = settings.NAVIDROME_USER + """ + Sets a 1-star rating for a song using Navidrome's Subsonic API. + """ + response = await _call_subsonic_api('setRating', id=song_id, rating=1) + return response is not None - if not all([navidrome_url, username]): - logging.error("NAVIDROME_URL and NAVIDROME_USER must be set in settings.py.") - return - salt = settings.SUBSONIC_SALT - token = settings.SUBSONIC_TOKEN - - if not all([salt, token]): - password = settings.NAVIDROME_PASSWORD - if not password: - logging.error("Either (SUBSONIC_SALT and SUBSONIC_TOKEN) or NAVIDROME_PASSWORD must be set in settings.py.") - return - - # Subsonic API requires a salt and a token (md5(password + salt)) - salt = ''.join(random.choice(string.ascii_lowercase + string.digits) for _ in range(10)) - token = hashlib.md5((password + salt).encode('utf-8')).hexdigest() - - params = { - 'u': username, - 't': token, - 's': salt, - 'v': '1.16.1', - 'c': 'heart-monitor', - 'f': 'json', - 'id': song_id, - 'rating': 1, - } - - api_url = f"{navidrome_url.rstrip('/')}/rest/setRating" - - try: - async with aiohttp.ClientSession() as session: - async with session.get(api_url, params=params) as response: - response.raise_for_status() - data = await response.json() - except aiohttp.ClientError as e: - logging.error(f"Error calling Navidrome API for song {song_id}: {e}") - return - - if data.get('subsonic-response', {}).get('status') != 'ok': - logging.error(f"Failed to set rating for song {song_id}. Response: {data}") +async def handle_star_unstar_event(song_id): + """ + Sets song rating to 1 and logs the song details. + """ + success = await navidrome_set_rating(song_id) + if not success: + # _call_subsonic_api already logs errors return details = await navidrome_get_song_details(song_id) @@ -188,7 +168,7 @@ async def main(): logging.info( f"Song {song_id} was starred and then unstarred within {STAR_UNSTAR_WINDOW.seconds} seconds." ) - await navidrome_set_rating(song_id) + await handle_star_unstar_event(song_id) # Remove song from tracking after it has been unstarred del starred_songs[song_id]