From: Fredrik Tolf Date: Sun, 12 May 2013 04:01:36 +0000 (+0200) Subject: Handle mangafox download errors slightly nicer. X-Git-Url: http://dolda2000.com/gitweb/?p=automanga.git;a=commitdiff_plain;h=30053c2cfe18b77da52df38daebbc91b3a2b5240 Handle mangafox download errors slightly nicer. --- diff --git a/manga/mangafox.py b/manga/mangafox.py index ed40efb..b165d21 100644 --- a/manga/mangafox.py +++ b/manga/mangafox.py @@ -6,8 +6,16 @@ soup = BeautifulSoup.BeautifulSoup class imgstream(lib.imgstream): def __init__(self, url): self.bk = urllib.urlopen(url) - self.ctype = self.bk.info()["Content-Type"] - self.clen = int(self.bk.info()["Content-Length"]) + ok = False + try: + if self.bk.getcode() != 200: + raise IOError("Server error: " + str(self.bk.getcode())) + self.ctype = self.bk.info()["Content-Type"] + self.clen = int(self.bk.info()["Content-Length"]) + ok = True + finally: + if not ok: + self.bk.close() def fileno(self): return self.bk.fileno()