jyapayne/safaribooks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Don't try to write down the image file if download failed

Browse source
This commit is contained in:
spacewander 2020-01-02 15:27:29 +08:00
commit 324751eec5
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
safaribooks.py Normal file → Executable file
View file

@ -850,6 +850,7 @@ class SafariBooks:
response = self.requests_provider(urljoin(SAFARI_BASE_URL, url), stream=True) response = self.requests_provider(urljoin(SAFARI_BASE_URL, url), stream=True)
if response == 0: if response == 0:
self.display.error("Error trying to retrieve this image: %s\n From: %s" % (image_name, url)) self.display.error("Error trying to retrieve this image: %s\n From: %s" % (image_name, url))
return
with open(image_path, 'wb') as img: with open(image_path, 'wb') as img:
for chunk in response.iter_content(1024): for chunk in response.iter_content(1024):