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

added progressbar

Browse Source
This commit is contained in:
Hellow2 2023-04-04 17:20:27 +02:00
parent 2bde1f683e
commit da28616011
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,7 +2,7 @@ from typing import Optional, List, Tuple
from pathlib import Path from pathlib import Path
from collections import defaultdict from collections import defaultdict
import requests import requests
# from tqdm import tqdm from tqdm import tqdm
from ..utils import shared from ..utils import shared
from .parents import DatabaseObject from .parents import DatabaseObject
@ -71,13 +71,19 @@ class Target(DatabaseObject):
chunk_size = 1024 chunk_size = 1024
total_size = int(r.headers.get('content-length')) total_size = int(r.headers.get('content-length'))
print(total_size)
initial_pos = 0 initial_pos = 0
with open(self.file_path,'wb') as f: with open(self.file_path,'wb') as f:
try: try:
for chunk in r.iter_content(chunk_size=chunk_size): """
size = f.write(chunk) 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):
size = f.write(chunk)
except requests.exceptions.Timeout: except requests.exceptions.Timeout:
shared.DOWNLOAD_LOGGER.error("Stream timed out.") shared.DOWNLOAD_LOGGER.error("Stream timed out.")
return False return False