From 6cdfd4c960daaf9932d06d982e2d36612662dc58 Mon Sep 17 00:00:00 2001 From: Maksim Andrianov Date: Wed, 6 Jan 2021 15:46:10 +0300 Subject: [PATCH] [generator] Fixed download_file() --- tools/python/maps_generator/utils/file.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tools/python/maps_generator/utils/file.py b/tools/python/maps_generator/utils/file.py index 1036d8b9c7..f97bcfae2a 100644 --- a/tools/python/maps_generator/utils/file.py +++ b/tools/python/maps_generator/utils/file.py @@ -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(