Properly handle lack of Content-Size header

This commit is contained in:
Mathieu Comandon 2019-02-01 02:57:27 -08:00
parent 8a02c00bcf
commit 63d9fb3250

View file

@ -111,7 +111,7 @@ class Downloader:
if self.referer:
headers["Referer"] = self.referer
response = requests.get(self.url, headers=headers, stream=True)
self.full_size = int(response.headers.get("Content-Length").strip())
self.full_size = int(response.headers.get("Content-Length", "").strip() or 0)
for chunk in response.iter_content(chunk_size=1024 * 1024):
if not self.file_pointer:
break