diff --git a/headphones/albumart.py b/headphones/albumart.py index 6ed73194..1fd68d1d 100644 --- a/headphones/albumart.py +++ b/headphones/albumart.py @@ -22,6 +22,7 @@ import requests as requests import headphones from headphones import db, request, logger + def getAlbumArt(albumid): artwork_path = None @@ -160,7 +161,7 @@ def getImageInfo(data): content_type = None # handle GIFs - if (size >= 10) and data[:6] in ('GIF87a', 'GIF89a'): + if size >= 10 and data[:6] in ('GIF87a', 'GIF89a'): # Check to see if content_type is correct content_type = 'image/gif' w, h = struct.unpack("= 24) and data.startswith('\211PNG\r\n\032\n') - and (data[12:16] == 'IHDR')): + elif size >= 24 and data.startswith('\211PNG\r\n\032\n') and data[12:16] == 'IHDR': content_type = 'image/png' w, h = struct.unpack(">LL", data[16:24]) width = int(w) height = int(h) # Maybe this is for an older PNG version. - elif (size >= 16) and data.startswith('\211PNG\r\n\032\n'): + elif size >= 16 and data.startswith('\211PNG\r\n\032\n'): # Check to see if we have the right content type content_type = 'image/png' w, h = struct.unpack(">LL", data[8:16]) @@ -186,21 +186,23 @@ def getImageInfo(data): height = int(h) # handle JPEGs - elif (size >= 2) and data.startswith('\377\330'): + elif size >= 2 and data.startswith('\377\330'): content_type = 'image/jpeg' jpeg = StringIO.StringIO(data) jpeg.read(2) b = jpeg.read(1) try: - while (b and ord(b) != 0xDA): - while (ord(b) != 0xFF): b = jpeg.read(1) - while (ord(b) == 0xFF): b = jpeg.read(1) - if (ord(b) >= 0xC0 and ord(b) <= 0xC3): + while b and ord(b) != 0xDA: + while ord(b) != 0xFF: + b = jpeg.read(1) + while ord(b) == 0xFF: + b = jpeg.read(1) + if ord(b) >= 0xC0 and ord(b) <= 0xC3: jpeg.read(3) h, w = struct.unpack(">HH", jpeg.read(4)) break else: - jpeg.read(int(struct.unpack(">H", jpeg.read(2))[0])-2) + jpeg.read(int(struct.unpack(">H", jpeg.read(2))[0]) - 2) b = jpeg.read(1) width = int(w) height = int(h)