NussNougate/nhentai
4
0
Fork 0
mirror of https://github.com/RicterZ/nhentai.git synced 2025-04-19 18:31:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix #31

Browse Source
This commit is contained in:
Ricter Z 2018-10-24 21:56:21 +08:00
parent db6d45efe0
commit 8cbb334d36
2 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nhentai/__init__.py
View File

@ -1,3 +1,3 @@
__version__ = '0.2.16'
__version__ = '0.2.17'
__author__ = 'RicterZ'
__email__ = 'ricterzheng@gmail.com'

22
nhentai/downloader.py
View File

@ -41,20 +41,27 @@ class Downloader(Singleton):
extension)))
return 1, url
response = None
with open(os.path.join(folder, base_filename.zfill(3) + extension), "wb") as f:
i=0
while i<10:
i = 0
while i < 10:
try:
response = request('get', url, stream=True, timeout=self.timeout)
if response.status_code != 200:
raise NhentaiImageNotExistException
except NhentaiImageNotExistException as e:
raise e
except Exception as e:
i+=1
if not i<10:
i += 1
if not i < 10:
logger.critical(str(e))
return 0, None
continue
break
if response.status_code != 200:
raise NhentaiImageNotExistException
length = response.headers.get('content-length')
if length is None:
f.write(response.content)
@ -96,6 +103,9 @@ class Downloader(Singleton):
if self.path:
folder = os.path.join(self.path, folder)
if len(folder) > 100:
folder = folder[:100]
if not os.path.exists(folder):
logger.warn('Path \'{0}\' does not exist, creating.'.format(folder))
try: