danvics/soulsync
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Make watchlist Spotify matching strict

Browse source 
Resolve Spotify artist matching through the exact Spotify client only, so watchlist ID backfill cannot drift to fallback-provider results. Remove the remaining preemptive provider availability check from the backfill loop.
This commit is contained in:
Antti Kettunen 2026-04-16 09:59:49 +03:00
parent 992be1a056
commit e657a1d432
2 changed files with 64 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
core/watchlist_scanner.py
View file

@ -355,7 +355,6 @@ class ScanResult:
class WatchlistDiscographyResult:
"""Resolved watchlist artist discography for a specific metadata source."""
source: str
client: Any
artist_id: str
albums: List[Any]
image_url: Optional[str] = None
@ -560,8 +559,9 @@ class WatchlistScanner:
return self._get_artist_image_from_data(artist_data)
def _get_album_data_for_source(self, source: str, client: Any, album_id: str, album_name: str = '') -> Optional[Dict[str, Any]]:
def _get_album_data_for_source(self, source: str, album_id: str, album_name: str = '') -> Optional[Dict[str, Any]]:
"""Fetch album data for a specific source and normalize track payloads when needed."""
client = get_client_for_source(source)
if not client or not album_id or not hasattr(client, 'get_album'):
return None
@ -642,7 +642,6 @@ class WatchlistScanner:
image_url = self._get_artist_image_for_source(watchlist_artist, source, client, artist_id)
return WatchlistDiscographyResult(
source=source,
client=client,
artist_id=artist_id,
albums=albums,
image_url=image_url,
@ -913,14 +912,11 @@ class WatchlistScanner:
_emit('scan_started', profile_id=profile_id, total_artists=len(watchlist_artists))
# Rate-limit-aware backfill of artist IDs for the providers we can safely resolve.
providers_to_backfill = []
for source in self._watchlist_source_priority():
if source == 'spotify':
if self._spotify_available_for_run():
providers_to_backfill.append(source)
elif source in {'itunes', 'deezer', 'discogs'} and get_client_for_source(source):
providers_to_backfill.append(source)
# Keep this as a plain source list; resolve the client right before each use.
providers_to_backfill = [
source for source in self._watchlist_source_priority()
if source in {'spotify', 'itunes', 'deezer', 'discogs'}
]
for provider in providers_to_backfill:
try:
@ -1005,11 +1001,10 @@ class WatchlistScanner:
album_fetcher = lambda album_id, album_name='': self.metadata_service.get_album(album_id)
else:
source = discography_result.source
client = discography_result.client
albums = discography_result.albums
source_artist_id = discography_result.artist_id
artist_image_url = discography_result.image_url or self.get_artist_image_url(artist) or ''
album_fetcher = lambda album_id, album_name='': self._get_album_data_for_source(source, client, album_id, album_name)
album_fetcher = lambda album_id, album_name='': self._get_album_data_for_source(source, album_id, album_name)
absolute_index = artist_index_offset + i + 1
if scan_state is not None:
@ -1251,8 +1246,11 @@ class WatchlistScanner:
logger.debug(f"Fetching discography for artist {spotify_artist_id}" +
(" (full)" if needs_full_discog else " (recent only, max 1 page)"))
albums = self.spotify_client.get_artist_albums(
spotify_artist_id, album_type='album,single', limit=50,
skip_cache=True, max_pages=0 if needs_full_discog else 1
spotify_artist_id,
album_type='album,single',
limit=50,
skip_cache=True,
max_pages=0 if needs_full_discog else 1,
)
if not albums:
@ -1413,8 +1411,8 @@ class WatchlistScanner:
update_fn = {
'spotify': self.database.update_watchlist_spotify_id,
'itunes': self.database.update_watchlist_itunes_id,
'deezer': getattr(self.database, 'update_watchlist_deezer_id', None),
'discogs': getattr(self.database, 'update_watchlist_discogs_id', None),
'deezer': self.database.update_watchlist_deezer_id,
'discogs': self.database.update_watchlist_discogs_id,
}.get(provider)
if not match_fn or not update_fn:
@ -1518,10 +1516,11 @@ class WatchlistScanner:
def _match_to_spotify(self, artist_name: str) -> Optional[str]:
"""Match artist name to Spotify ID using fuzzy name comparison."""
try:
if hasattr(self, '_metadata_service') and self._metadata_service:
results = self._metadata_service.spotify.search_artists(artist_name, limit=5)
else:
results = self.spotify_client.search_artists(artist_name, limit=5)
client = get_client_for_source('spotify')
if not client:
return None
results = client.search_artists(artist_name, limit=5, allow_fallback=False)
return self._best_artist_match(results, artist_name)
except Exception as e:
@ -2101,7 +2100,7 @@ class WatchlistScanner:
try:
# Try Spotify search (only when Spotify is the configured primary source)
if self._spotify_is_primary_source():
searched_results = self.spotify_client.search_artists(artist_name, limit=1)
searched_results = self.spotify_client.search_artists(artist_name, limit=1, allow_fallback=False)
if searched_results and len(searched_results) > 0:
searched_spotify_id = searched_results[0].id
except Exception as e:
@ -2139,7 +2138,11 @@ class WatchlistScanner:
# Try to match on Spotify (only when Spotify is the configured primary source)
if self._spotify_is_primary_source():
try:
spotify_results = self.spotify_client.search_artists(artist_name_to_match, limit=1)
spotify_results = self.spotify_client.search_artists(
artist_name_to_match,
limit=1,
allow_fallback=False,
)
if spotify_results and len(spotify_results) > 0:
spotify_artist = spotify_results[0]
# Skip if this is the searched artist
@ -2443,7 +2446,7 @@ class WatchlistScanner:
artist_id,
album_type='album,single,ep',
limit=50,
skip_cache=True
skip_cache=True,
)
else: # itunes or deezer fallback
all_albums = itunes_client.get_artist_albums(
@ -2668,7 +2671,11 @@ class WatchlistScanner:
# Try Spotify first if available
if spotify_available:
try:
search_results = self.spotify_client.search_albums(f"album:{album_row['title']} artist:{album_row['artist_name']}", limit=1)
search_results = self.spotify_client.search_albums(
f"album:{album_row['title']} artist:{album_row['artist_name']}",
limit=1,
allow_fallback=False,
)
if search_results and len(search_results) > 0:
spotify_album = search_results[0]
album_data = self.spotify_client.get_album(spotify_album.id)
@ -2854,7 +2861,7 @@ class WatchlistScanner:
artist.spotify_artist_id,
album_type='album,single,ep',
limit=5,
skip_cache=True
skip_cache=True,
)
if not recent_releases:
@ -3095,7 +3102,7 @@ class WatchlistScanner:
artist.spotify_artist_id,
album_type='album,single,ep',
limit=20,
skip_cache=True
skip_cache=True,
)
for album in albums or []:
process_album(album, artist.artist_name, artist.spotify_artist_id, fallback_id if fallback_source == 'itunes' else None, 'spotify')
@ -3152,7 +3159,7 @@ class WatchlistScanner:
artist.similar_artist_spotify_id,
album_type='album,single,ep',
limit=20,
skip_cache=True
skip_cache=True,
)
for album in albums or []:
process_album(album, artist.similar_artist_name, artist.similar_artist_spotify_id, fallback_id if fallback_source == 'itunes' else None, 'spotify')

33
tests/test_watchlist_scanner_scan.py
View file

@ -53,13 +53,21 @@ from core.watchlist_scanner import WatchlistScanner
class _FakeSpotifyClient:
def __init__(self, search_results=None):
self.search_results = list(search_results or [])
self.search_calls = []
def is_spotify_authenticated(self):
return False
def search_artists(self, query, limit=1, allow_fallback=True):
self.search_calls.append((query, limit, allow_fallback))
return list(self.search_results) if allow_fallback else []
class _FakeMetadataService:
def __init__(self, album_data):
self.spotify = _FakeSpotifyClient()
def __init__(self, album_data, spotify_client=None):
self.spotify = spotify_client or _FakeSpotifyClient()
self.itunes = types.SimpleNamespace()
self._album_data = album_data
@ -76,8 +84,8 @@ class _FakeSourceClient:
self.album_calls = []
self.artist_calls = []
def search_artists(self, query, limit=1):
self.search_calls.append((query, limit))
def search_artists(self, query, limit=1, **kwargs):
self.search_calls.append((query, limit, kwargs))
return [types.SimpleNamespace(id=self.artist_id, name=query)]
def get_artist_albums(self, artist_id, album_type='album,single', limit=50, **kwargs):
@ -374,3 +382,20 @@ def test_get_artist_discography_for_watchlist_falls_back_when_primary_empty(monk
assert result.albums and result.albums[0].id == "sp-album"
assert deezer_client.album_calls
assert spotify_client.album_calls
def test_match_to_spotify_uses_strict_lookup():
spotify_client = _FakeSpotifyClient(
search_results=[types.SimpleNamespace(id="fallback-id", name="Artist One")]
)
scanner = WatchlistScanner(metadata_service=_FakeMetadataService(None, spotify_client=spotify_client))
original_get_client_for_source = watchlist_scanner_module.get_client_for_source
watchlist_scanner_module.get_client_for_source = lambda source: spotify_client if source == "spotify" else None
try:
result = scanner._match_to_spotify("Artist One")
finally:
watchlist_scanner_module.get_client_for_source = original_get_client_for_source
assert result is None
assert spotify_client.search_calls == [("Artist One", 5, False)]