From e3418be75d3eb7e8a3de30caa4f0ca004ef243d6 Mon Sep 17 00:00:00 2001
From: Johannes Baiter <johannes.baiter@gmail.com>
Date: Tue, 28 May 2013 12:58:05 +0200
Subject: [PATCH] Add support for manually entered release ID in Discogs

---
 beetsplug/discogs.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/beetsplug/discogs.py b/beetsplug/discogs.py
index 951c5d593..0d30bb39d 100644
--- a/beetsplug/discogs.py
+++ b/beetsplug/discogs.py
@@ -60,6 +60,26 @@ class DiscogsPlugin(BeetsPlugin):
             log.debug('Discogs API Error: %s (query: %s' % (e, query))
             return []
 
+    def album_for_id(self, album_id):
+        """Fetches an album by its Discogs ID and returns an AlbumInfo object
+        or None if the album is not found.
+        """
+        log.debug('Searching for release with id %s' % str(album_id))
+        # discogs-client can handle both int and str, so we just strip
+        # the leading 'r' that might be accidentally pasted into the form
+        if not isinstance(album_id, int) and album_id.startswith('r'):
+            album_id = album_id[1:]
+        result = Release(album_id)
+        # Try to obtain title to verify that we indeed have a valid Release
+        try:
+            getattr(result, 'title')
+        except DiscogsAPIError as e:
+            if e.message != '404 Not Found':
+                log.debug('Discogs API Error: %s (query: %s)'
+                          % (e, result._uri))
+            return None
+        return self.get_album_info(result)
+
     def get_albums(self, query):
         """Returns a list of AlbumInfo objects for a discogs search query.
         """