add html doujinshi viewer

[nhentai.git] / nhentai / downloader.py

diff --git a/nhentai/downloader.py b/nhentai/downloader.py
index 276e1cb73e6963bfdb99e7c39d378cfd0ce786cd..b954e27654498c1d41081cad63f9f7248761ea1a 100644 (file)
--- a/nhentai/downloader.py
+++ b/nhentai/downloader.py
@@ -14,6 +14,9 @@ from nhentai.parser import request
 from nhentai.utils import Singleton
 
 
+requests.packages.urllib3.disable_warnings()
+
+
 class NhentaiImageNotExistException(Exception):
     pass
 
@@ -44,7 +47,7 @@ class Downloader(Singleton):
                     for chunk in response.iter_content(2048):
                         f.write(chunk)
 
-        except requests.HTTPError as e:
+        except (requests.HTTPError, requests.Timeout) as e:
             if retried < 3:
                 logger.warning('Warning: {0}, retrying({1}) ...'.format(str(e), retried))
                 return 0, self._download(url=url, folder=folder, filename=filename, retried=retried+1)