From 4a12307cb8d7b991786dd5de647fb072c9131c7e Mon Sep 17 00:00:00 2001
From: rembo10 <rembo10@headphones>
Date: Fri, 22 Jun 2012 11:04:39 +0530
Subject: [PATCH] Modifieded the getArtwork and getInfo functions so you can
 now just pass an ArtistID or AlbumID without having to specify the type as
 well. Also - changed the error catching when opening the url & parsing the
 data

---
 headphones/cache.py    | 73 ++++++++++++++++++++++++++++--------------
 headphones/webserve.py | 15 +++------
 2 files changed, 53 insertions(+), 35 deletions(-)

diff --git a/headphones/cache.py b/headphones/cache.py
index e8d36c05..6bbabfd1 100644
--- a/headphones/cache.py
+++ b/headphones/cache.py
@@ -99,28 +99,37 @@ class Cache(object):
         else:
             return False
         
-    def get_artwork_from_cache(self, id, id_type):
+    def get_artwork_from_cache(self, ArtistID=None, AlbumID=None):
         '''
-        Pass a musicbrainz id to this function, and a type (either artist or album)
+        Pass a musicbrainz id to this function (either ArtistID or AlbumID)
         '''
-        
-        self.id = id
-        self.id_type = id_type
+        if ArtistID:
+            self.id = ArtistID
+            self.id_type = 'artist'
+        else:
+            self.id = AlbumID
+            self.id_type = 'album'
         
         if self._exists('artwork') and self._is_current(self.artwork_files[0]):
             return self.artwork_files[0]
         else:
             self._update_cache()
-            
-            if self.artwork_errors:
+            # If we failed to get artwork, either return the url or the older file
+            if self.artwork_errors and self.artwork_url:
                 return self.artwork_url
-            if self._exists('artwork'):
+            elif self._exists('artwork'):
                 return self.artwork_files[0]
+            else:
+                return None
                 
-    def get_info_from_cache(self, id, id_type):
+    def get_info_from_cache(self, ArtistID=None, AlbumID=None):
         
-        self.id = id
-        self.id_type = id_type
+        if ArtistID:    
+            self.id = ArtistID
+            self.id_type = 'artist'
+        else:
+            self.id = AlbumID
+            self.id_type = 'album'
         
         if self._exists('info') and self._is_current(self.info_files[0]):
             f = open(self.info_files[0], 'r').read()
@@ -128,12 +137,15 @@ class Cache(object):
         else:
             self._update_cache()
             
-            if self.info_errors:
+            if self.info_errors and self.info:
                 return self.info
             
-            if self._exists('info'):
+            elif self._exists('info'):
                 f = open(self.info_files[0],'r').read()
                 return f.decode('utf-8')
+                
+            else:
+                return None
 
     def _update_cache(self):
         '''
@@ -147,12 +159,19 @@ class Cache(object):
             
             try:
                 result = urllib2.urlopen(url).read()
-                data = simplejson.JSONDecoder().decode(result)
-                info = data['artist']['bio']['content']
-                image_url = data['artist']['image'][-1]['#text']
             except:
                 logger.warn('Could not open url: ' + url)
-                return        
+                return
+            
+            if result:    
+            
+                try:
+                    data = simplejson.JSONDecoder().decode(result)
+                    info = data['artist']['bio']['content']
+                    image_url = data['artist']['image'][-1]['#text']
+                except:
+                    logger.warn('Could not parse data from url: ' + url)
+                    return
         
         else:
             myDB = db.DBConnection()
@@ -162,12 +181,18 @@ class Cache(object):
             logger.debug('Retrieving artist information from: ' + url)
             try:
                 result = urllib2.urlopen(url).read()
-                data = simplejson.JSONDecoder().decode(result)
-                info = data['album']['wiki']['content']
-                image_url = data['album']['image'][-1]['#text']
             except:
                 logger.warn('Could not open url: ' + url)
                 return
+                
+            if result:
+                try:    
+                    data = simplejson.JSONDecoder().decode(result)
+                    info = data['album']['wiki']['content']
+                    image_url = data['album']['image'][-1]['#text']
+                except:
+                    logger.warn('Could not parse data from url: ' + url)
+                    return
 
         if info:
             
@@ -241,18 +266,18 @@ class Cache(object):
                     self.artwork_errors = True
                     self.artwork_url = image_url
                     
-def getArtwork(id, id_type):
+def getArtwork(ArtistID=None, AlbumID=None):
     
     c = Cache()
-    artwork_path = c.get_artwork_from_cache(id, id_type)
+    artwork_path = c.get_artwork_from_cache(ArtistID, AlbumID)
     
     if artwork_path.startswith('http://'):
         return artwork_path
     else:
         return "file:///" + artwork_path
     
-def getInfo(id, id_type):
+def getInfo(ArtistID=None, AlbumID=None):
     
     c = Cache()
-    info = c.get_info_from_cache(id, id_type)
+    info = c.get_info_from_cache(ArtistID, AlbumID)
     return info
diff --git a/headphones/webserve.py b/headphones/webserve.py
index da6bf2e1..45d5b3ea 100644
--- a/headphones/webserve.py
+++ b/headphones/webserve.py
@@ -592,23 +592,16 @@ class WebInterface(object):
         
     api.exposed = True
     
-    def getInfo(self, id, id_type):
+    def getInfo(self, ArtistID=None, AlbumID=None):
         
         from headphones import cache
-        return cache.getInfo(id, id_type)
+        return cache.getInfo(ArtistID, AlbumID)
         
     getInfo.exposed = True
     
-    def getArtwork(self, id, id_type):
+    def getArtwork(self, ArtistID=None, AlbumID=None):
         
         from headphones import cache
-        return cache.getArtwork(id, id_type)
+        return cache.getArtwork(ArtistID, AlbumID)
         
     getArtwork.exposed = True
-    
-    def getArtworkURL(self,id, id_type):
-        
-        from headphones import cache
-        return cache.getArtworkURL(id, id_type)
-        
-    getArtworkURL.exposed = True