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

added a description and removed redundant output of progressbar

Browse Source
This commit is contained in:
Hellow
2023-04-04 17:59:08 +02:00
parent afc3c859b0
commit f970950aea
4 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -63,14 +63,12 @@ class Target(DatabaseObject):
with open(copy_to.file_path, "wb") as write_to:
write_to.write(read_from.read())
def stream_into(self, r: requests.Response) -> bool:
def stream_into(self, r: requests.Response, desc: str = None) -> bool:
if r is None:
return False
self.create_path()
chunk_size = 1024
total_size = int(r.headers.get('content-length'))
with open(self.file_path, 'wb') as f:
@@ -79,8 +77,9 @@ class Target(DatabaseObject):
https://en.wikipedia.org/wiki/Kilobyte
> The internationally recommended unit symbol for the kilobyte is kB.
"""
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):
with tqdm(total=total_size, unit='B', unit_scale=True, unit_divisor=1024, desc=desc) as t:
for chunk in r.iter_content(chunk_size=shared.CHUNK_SIZE):
size = f.write(chunk)
t.update(size)