Artist check within local, mb and lastfm databases

lib/tasks/artist_correct.rake

@@ -0,0 +1,47 @@
+require 'cgi'
+
+namespace :artist do
+  desc 'Find and fix uncorrect mbid'
+  task :correct => :environment do
+    Artist.where( :approved => nil ).each do |artist|
+      puts "Artist: "+artist.name
+      bh_albums = artist.albums.map{ |a| a.name.scan(/[a-zA-Z0-9]/).join() }
+      tmp = LastFM::Artist.get_top_albums( :artist => artist.name )['topalbums']
+      if tmp['album'].nil?
+        tmp['album'] = []
+      elsif tmp['album'].is_a? Hash
+        tmp['album'] = [tmp['album']]
+      end
+      lastfm_albums = tmp['album'].map do |a|
+        a['name'].scan(/[a-zA-Z0-9]/).join()
+      end
+      if (bh_albums & lastfm_albums).empty?
+        puts "Result: No intersection"
+        best_candidate = nil
+        best_intersection = 0
+        MusicBrainz::Artist.search(artist.name.gsub(/\!/, '')).each do |candidate|
+          candidate = MusicBrainz::Artist.find(candidate[:mbid])
+          candidate_albums = candidate.release_groups.map{ |a| a.title.scan(/[a-zA-Z0-9]/).join() }
+          if (candidate_albums & lastfm_albums).length > best_intersection
+            best_intersection = (candidate_albums & lastfm_albums).length
+            best_candidate = candidate.id
+          end
+        end
+        if best_candidate.nil?
+          puts "No better candidate found (now: "+artist.mbid.to_s+")\n\n"
+          artist.approved = 0
+          artist.save
+        else
+          puts "Better candidate found: "+best_candidate+" (now: "+artist.mbid.to_s+")\n\n"
+          artist.approved = 2
+          artist.save
+        end
+      else
+        artist.approved = 1
+        artist.save
+        puts "Result: OK\n\n"
+      end
+      sleep 1
+    end
+  end
+end

lib/tasks/db_artist.rake

@@ -6,14 +6,15 @@ namespace :db do
         artist.albums.each do |album|
           album.destroy
         end
-        puts artist.name+(artist.mbid.nil? or artist.mbid.empty? ? '' : ' ('+artist.mbid+')')
+        puts 'Artist:'+artist.name
+        puts 'MBID:  '+artist.mbid unless artist.mbid.nil?
         res = ImportController.importArtist(artist.name)
         if res == 1
-          puts 'OK'
+          puts 'Status: OK'
         elsif res == 2
-          puts 'FAIL'
+          puts 'Status: FAILED'
         elsif res == 3
-          puts 'SKIP'
+          puts 'Status: SKIP'
         end
         puts ''
       end