File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -1368,6 +1368,7 @@ def _download_file_to_stream(
13681368 get_func = self ._session_functions_mapping [RequestTypes .get ]
13691369
13701370 retries = self .get_default_max_retries ()
1371+ content_size_set = False
13711372 for attempt in range (retries ):
13721373 # Continue in download
13731374 offset = progress .get_transferred_size ()
@@ -1377,9 +1378,12 @@ def _download_file_to_stream(
13771378 try :
13781379 with get_func (url , ** kwargs ) as response :
13791380 response .raise_for_status ()
1380- progress .set_content_size (
1381- response .headers ["Content-length" ]
1382- )
1381+ if not content_size_set :
1382+ content_size_set = True
1383+ progress .set_content_size (
1384+ response .headers ["Content-length" ]
1385+ )
1386+
13831387 for chunk in response .iter_content (chunk_size = chunk_size ):
13841388 stream .write (chunk )
13851389 progress .add_transferred_chunk (len (chunk ))
You can’t perform that action at this time.
0 commit comments