88
99import requests
1010
11- from .exceptions import (AssetChecksumMismatch , AssetChecksumNotPresent ,
12- DownloadException )
11+ from .exceptions import (
12+ AssetChecksumMismatch ,
13+ AssetChecksumNotPresent ,
14+ DownloadException ,
15+ )
1316from .logger import SDKLogger
1417from .utils import FormatTypes , Utils
1518
1619logger = SDKLogger ("downloads" )
1720
1821from .bandwidth import DiskBandwidth , NetworkBandwidth
19- from .exceptions import (AssetNotFullyUploaded , DownloadException ,
20- WatermarkIDDownloadException )
22+ from .exceptions import (
23+ AssetNotFullyUploaded ,
24+ DownloadException ,
25+ WatermarkIDDownloadException ,
26+ )
2127from .transport import HTTPClient
2228
2329
@@ -71,7 +77,7 @@ def get_path(self):
7177 def _evaluate_asset (self ):
7278 if self .asset .get ("_type" ) != "file" :
7379 raise DownloadException (
74- message = "Unsupport Asset type: {}" . format ( self .asset .get (" _type" ))
80+ message = f "Unsupport Asset type: { self .asset .get (' _type' ) } "
7581 )
7682
7783 # This logic may block uploads that were started before this field was introduced
@@ -196,7 +202,7 @@ def __init__(self, downloader: FrameioDownloader, concurrency=None, progress=Tru
196202 self .bytes_completed = 0
197203 self .downloader = downloader
198204 self .futures = []
199- self .original = self .downloader .asset [' original' ]
205+ self .original = self .downloader .asset [" original" ]
200206
201207 # Ensure this is a valid number before assigning
202208 if concurrency is not None and type (concurrency ) == int and concurrency > 0 :
@@ -299,10 +305,7 @@ def _download_whole(self, url: str):
299305 math .ceil (self .downloader .filesize / (download_time ))
300306 )
301307 print (
302- "Downloaded {} at {}" .format (
303- Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE ),
304- download_speed ,
305- )
308+ f"Downloaded { Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE )} at { download_speed } "
306309 )
307310
308311 return self .destination , download_speed
@@ -378,10 +381,7 @@ def multi_thread_download(self):
378381 in_byte = 0 # Set initially here, but then override
379382
380383 print (
381- "Multi-part download -- {} -- {}" .format (
382- self .downloader .asset ["name" ],
383- Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE ),
384- )
384+ f"Multi-part download -- { self .downloader .asset ['name' ]} -- { Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE )} "
385385 )
386386
387387 with concurrent .futures .ThreadPoolExecutor (
@@ -392,7 +392,7 @@ def multi_thread_download(self):
392392 out_byte = offset * (i + 1 )
393393
394394 # Create task tuple
395- task = (self .downloader .asset [' original' ], in_byte , out_byte , i )
395+ task = (self .downloader .asset [" original" ], in_byte , out_byte , i )
396396
397397 # Stagger start for each chunk by 0.1 seconds
398398 if i < self .concurrency :
@@ -423,22 +423,22 @@ def multi_thread_download(self):
423423 raise AssetChecksumNotPresent
424424
425425 # Calculate the file hash
426- if Utils .calculate_hash (self .destination ) != self .downloader .original_checksum :
426+ if (
427+ Utils .calculate_hash (self .destination )
428+ != self .downloader .original_checksum
429+ ):
427430 raise AssetChecksumMismatch
428431
429432 # Log completion event
430433 SDKLogger ("downloads" ).info (
431- "Downloaded {} at {}" .format (
432- Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE ),
433- download_speed ,
434- )
434+ f"Downloaded { Utils .format_value (self .downloader .filesize , type = FormatTypes .SIZE )} at { download_speed } "
435435 )
436436
437437 # Submit telemetry
438438 transfer_stats = {
439439 "speed" : download_speed ,
440440 "time" : download_time ,
441- "cdn" : AWSClient .check_cdn (self .original )
441+ "cdn" : AWSClient .check_cdn (self .original ),
442442 }
443443
444444 # Event(self.user_id, 'python-sdk-download-stats', transfer_stats)
0 commit comments