music-kraken/music-kraken-core
3
2
Fork 0
Code Issues 10 Pull Requests 1 Packages Projects 2 Releases Wiki Activity

improved progressbar

Browse Source
This commit is contained in:
Hellow 2023-04-04 17:54:05 +02:00
parent 5f735c25de
commit afc3c859b0
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/music_kraken/objects/target.py
View File

@ -70,10 +70,8 @@ class Target(DatabaseObject):
self.create_path()
chunk_size = 1024
total_size = int(r.headers.get('content-length'))
print(total_size)
initial_pos = 0
total_size = int(r.headers.get('content-length'))
with open(self.file_path, 'wb') as f:
try:
@ -81,8 +79,10 @@ class Target(DatabaseObject):
https://en.wikipedia.org/wiki/Kilobyte
> The internationally recommended unit symbol for the kilobyte is kB.
"""
for chunk in tqdm(r.iter_content(chunk_size=chunk_size), unit_scale=True, unit="B", total=total_size):
with tqdm(total=total_size, unit='B', unit_scale=True, unit_divisor=1024) as t:
for chunk in r.iter_content(chunk_size=chunk_size):
size = f.write(chunk)
t.update(size)
except requests.exceptions.Timeout:
shared.DOWNLOAD_LOGGER.error("Stream timed out.")