FetchArt plugin: Support grabbing album art from MB release group

Author: fortes

beetsplug/fetchart.py

@@ -19,6 +19,7 @@
 import logging
 import os
 import tempfile
+import requests
 
 from beets.plugins import BeetsPlugin
 from beets.util.artresizer import ArtResizer
@@ -65,12 +66,20 @@ def _fetch_image(url):
 # Cover Art Archive.
 
 CAA_URL = 'http://coverartarchive.org/release/{mbid}/front-500.jpg'
+CAA_GROUP_URL = 'http://coverartarchive.org/release-group/{mbid}/'
 
 def caa_art(release_id):
     """Return the Cover Art Archive URL given a MusicBrainz release ID.
     """
     return CAA_URL.format(mbid=release_id)
 
+def caa_group(release_group_id):
+    """Return the Cover Art Archive release group URL given a MusicBrainz
+    release group ID.
+    """
+    return CAA_GROUP_URL.format(mbid=release_group_id)
+
+RELEASE_ID_PAT = r'^http://musicbrainz.org/release/([^/]+)'
 
 # Art from Amazon.
 
@@ -153,6 +162,19 @@ def _source_urls(album):
         if url:
             yield url
 
+    if album.mb_releasegroupid:
+        group_url = caa_group(album.mb_releasegroupid)
+        response = requests.get(group_url)
+        if response.status_code == 200:
+            json = response.json()
+            if json.has_key('release'):
+                # URL in the form of http://musicbrainz.org/release/{mbid}
+                m = re.search(RELEASE_ID_PAT, json['release'])
+                if m:
+                    url = caa_art(m.group(1))
+                    if url:
+                        yield url
+
     # Amazon and AlbumArt.org.
     if album.asin:
         for url in art_for_asin(album.asin):