fix: resolve undefined reference error

The error occurred when downloading a file and the 'content-length' field is missing from the
HTTP response header.
This commit is contained in:
ulinja
2022-04-21 01:14:18 +02:00
parent 21ff40f75b
commit 5c2eabf758

View File

@@ -48,8 +48,8 @@ def download_file(path_to_output_file, url_to_file, show_progress = False):
file_response = requests.get(url_to_file, stream=True) file_response = requests.get(url_to_file, stream=True)
total_length = file_response.headers.get('content-length') total_length = file_response.headers.get('content-length')
if total_length is None: # no content length header if total_length is None: # no content length header
output_file.write(response.content) output_file.write(file_response.content)
else: else:
if (show_progress): if (show_progress):
total_length = int(total_length) total_length = int(total_length)