Skip to content

Commit 1befe2e

Browse files
authored
added progress_desc optional parameter
1 parent deb58a1 commit 1befe2e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

firerequests/main.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ async def download_chunk(
4646

4747
async def download_file(
4848
self, url: str, filename: str, max_files: int, chunk_size: int, headers: Optional[Dict[str, str]] = None,
49-
parallel_failures: int = 3, max_retries: int = 5, callback: Optional[Any] = None, show_progress: bool = True
49+
parallel_failures: int = 3, max_retries: int = 5, callback: Optional[Any] = None, show_progress: bool = True, progress_desc: Optional[str] = None
5050
):
5151
headers = headers or {"User-Agent": "Wget/1.21.2", "Accept": "*/*", "Accept-Encoding": "identity", "Connection": "Keep-Alive"}
5252
try:
@@ -81,7 +81,7 @@ async def download_file(
8181
))
8282

8383
if show_progress:
84-
progress_bar = tqdm(total=file_size, unit="B", unit_scale=True, desc="Downloading on 🔥")
84+
progress_bar = tqdm(total=file_size, unit="B", unit_scale=True, desc=progress_desc)
8585

8686
for chunk_result in asyncio.as_completed(tasks):
8787
downloaded = await chunk_result
@@ -114,7 +114,7 @@ async def download_chunk_with_retries(
114114

115115
async def upload_file(
116116
self, file_path: str, parts_urls: List[str], chunk_size: int, max_files: int,
117-
parallel_failures: int = 3, max_retries: int = 5, callback: Optional[Any] = None, show_progress: bool = True
117+
parallel_failures: int = 3, max_retries: int = 5, callback: Optional[Any] = None, show_progress: bool = True, progress_desc: Optional[str] = None
118118
):
119119
file_size = os.path.getsize(file_path)
120120
part_size = file_size // len(parts_urls)
@@ -137,7 +137,7 @@ async def upload_file(
137137
))
138138

139139
if show_progress:
140-
progress_bar = tqdm(total=file_size, unit="B", unit_scale=True, desc="Uploading on 🔥")
140+
progress_bar = tqdm(total=file_size, unit="B", unit_scale=True, desc=progress_desc)
141141

142142
for chunk_result in asyncio.as_completed(tasks):
143143
uploaded = await chunk_result

0 commit comments

Comments
 (0)