diff --git a/headphones/webserve.py b/headphones/webserve.py index 517104ab..da747a58 100644 --- a/headphones/webserve.py +++ b/headphones/webserve.py @@ -269,7 +269,11 @@ class WebInterface(object): searcher.searchforalbum(mbid, new=True) if action == 'WantedLossless': searcher.searchforalbum(mbid, lossless=True) - myDB.action('UPDATE artists SET TotalTracks=(SELECT COUNT(*) FROM tracks, artists WHERE tracks.ArtistName = artists.ArtistName AND AlbumTitle IN (SELECT AlbumTitle FROM albums WHERE Status != "Ignored")) WHERE ArtistID=(SELECT ArtistID FROM albums WHERE AlbumID=?)', [mbid]) + if ArtistID: + ArtistIDT = ArtistID + else: + ArtistIDT = myDB.action('SELECT ArtistID FROM albums WHERE AlbumID=?', [mbid]).fetchone()[0] + myDB.action('UPDATE artists SET TotalTracks=(SELECT COUNT(*) FROM tracks WHERE ArtistID = ? AND AlbumTitle IN (SELECT AlbumTitle FROM albums WHERE Status != "Ignored")) WHERE ArtistID=(SELECT ArtistID FROM albums WHERE AlbumID=?)', [ArtistIDT, mbid]) if ArtistID: raise cherrypy.HTTPRedirect("artistPage?ArtistID=%s" % ArtistID) else: