gpesquero/organicmaps
1
Fork 0
forked from organicmaps/organicmaps
Code Pull requests Activity

[generator] Fixed download_file()

Browse source
This commit is contained in:
Maksim Andrianov 2021-01-06 15:46:10 +03:00 • committed by Arsentiy Milchakov
parent 3f66ec64a9
commit 6cdfd4c960
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tools/python/maps_generator/utils/file.py
View file

@ -54,7 +54,14 @@ def download_file(url: AnyStr, name: AnyStr, download_if_exists: bool = True):
session.mount("file://", FileAdapter())
with open(tmp_name, "wb") as handle:
response = session.get(url, stream=True)
file_length = int(response.headers["Content-Length"])
file_length = None
try:
file_length = int(response.headers["Content-Length"])
except KeyError:
logger.warning(
f"There is no attribute Content-Length in headers [{url}]: {response.headers}"
)
current = 0
max_attempts = 32
attempts = max_attempts
@ -63,7 +70,7 @@ def download_file(url: AnyStr, name: AnyStr, download_if_exists: bool = True):
current += len(data)
handle.write(data)
if file_length == current:
if file_length is None or file_length == current:
break
logger.warning(