This commit is contained in:
Bastian Kleineidam 2012-09-27 21:54:56 +02:00
parent f3365f6a5e
commit 3ab526f7d8

View file

@ -34,15 +34,15 @@ class Comic(object):
def connect(self): def connect(self):
"""Connect to host and get meta information.""" """Connect to host and get meta information."""
out.write('Getting headers for %s...' % (url,), 2) out.write('Getting headers for %s...' % (self.url,), 2)
try: try:
self.urlobj = urlopen(url, referrer=referrer) self.urlobj = urlopen(self.url, referrer=self.referrer)
except urllib2.HTTPError, he: except urllib2.HTTPError, he:
raise FetchComicError, ('Unable to retrieve URL.', url, he.code) raise FetchComicError, ('Unable to retrieve URL.', self.url, he.code)
if self.urlobj.info().getmaintype() != 'image' and \ if self.urlobj.info().getmaintype() != 'image' and \
self.urlobj.info().gettype() not in ('application/octet-stream', 'application/x-shockwave-flash'): self.urlobj.info().gettype() not in ('application/octet-stream', 'application/x-shockwave-flash'):
raise FetchComicError, ('No suitable image found to retrieve.', url) raise FetchComicError, ('No suitable image found to retrieve.', self.url)
# Always use mime type for file extension if it is sane. # Always use mime type for file extension if it is sane.
if self.urlobj.info().getmaintype() == 'image': if self.urlobj.info().getmaintype() == 'image':