Replaced the custom weighting calculation by making use of MusicBrainz scoring scheme.
This commit is contained in:
parent
88c4128d82
commit
0b68c0c570
|
@ -46,19 +46,6 @@ module MusicBrainz
|
|||
:query => CGI.escape(name).gsub(/\!/, '\!') + '~',
|
||||
:limit => 50
|
||||
})
|
||||
artists.each { |artist|
|
||||
if artist[:name].downcase == name.downcase
|
||||
artist[:weight] += 80
|
||||
elsif artist[:name].downcase.gsub(/\s/, "") == name.downcase.gsub(/\s/, "")
|
||||
artist[:weight] += 25
|
||||
elsif artist[:aliases].include? name
|
||||
artist[:weight] += 20
|
||||
elsif artist[:aliases].map { |item| item.downcase }.include?(name.downcase)
|
||||
artist[:weight] += 10
|
||||
elsif artist[:aliases].map { |item| item.downcase.gsub(/\s/, "") }.include?(name.downcase.gsub(/\s/, ""))
|
||||
artist[:weight] += 5
|
||||
end
|
||||
}
|
||||
artists.sort{ |a, b| b[:weight] <=> a[:weight] }.take(10)
|
||||
end
|
||||
|
||||
|
|
|
@ -26,7 +26,7 @@ module MusicBrainz
|
|||
artists << {
|
||||
:name => a.first_element_child.text.gsub(/[`’]/, "'"),
|
||||
:sort_name => safe_get_value(a, "sort-name").gsub(/[`’]/, "'"),
|
||||
:weight => 0,
|
||||
:score => (safe_get_attr(a, nil, "score") .to_i rescue 0),
|
||||
:desc => safe_get_value(a, "disambiguation"),
|
||||
:type => safe_get_attr(a, nil, "type"),
|
||||
:mbid => safe_get_attr(a, nil, "id"),
|
||||
|
|
|
@ -20,6 +20,19 @@ describe MusicBrainz::Artist do
|
|||
matches.first[:name].should == "Kasabian"
|
||||
end
|
||||
|
||||
it "should return search results in the right order and pass back the correct score" do
|
||||
matches = MusicBrainz::Artist.search('Chris Martin')
|
||||
|
||||
matches.first[:score].should == 100
|
||||
matches.first[:mbid].should == "98d1ec5a-dd97-4c0b-9c83-7928aac89bca"
|
||||
matches[1][:score].should == 100
|
||||
matches[1][:mbid].should == "af2ab893-3212-4226-9e73-73a1660b6952"
|
||||
matches[2][:score].should == 95
|
||||
matches[2][:mbid].should == "444d1b63-534b-4ea6-89f0-0af6ab2e20c3"
|
||||
matches[3][:score].should == 95
|
||||
matches[3][:mbid].should == "90fff570-a4ef-4cd4-ba21-e00c7261b05a"
|
||||
end
|
||||
|
||||
it "finds name first than alias" do
|
||||
matches = MusicBrainz::Artist.search('Chris Martin')
|
||||
matches.length.should be > 0
|
||||
|
|
Loading…
Reference in New Issue