tgbot-collection/ytdlbot
1
1
Fork 0
mirror of https://github.com/tgbot-collection/ytdlbot.git synced 2025-02-24 15:24:03 +08:00
Code Issues Projects Releases Packages Wiki Activity

fix download hook

Browse source
This commit is contained in:
Benny 2024-08-04 10:35:55 +02:00
parent 03e3010ad3
commit c39b0207b0
No known key found for this signature in database
GPG key ID: 6CD0DBDA5235D481
1 changed files with 1 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
ytdlbot/downloader.py
View file

@ -114,12 +114,6 @@ def remove_bash_color(text):
def download_hook(d: dict, bot_msg):
# since we're using celery, server location may be located in different region.
# Therefore, we can't trigger the hook very often.
# the key is user_id + download_link
original_url = d["info_dict"]["original_url"]
key = f"{bot_msg.chat.id}-{original_url}"
if d["status"] == "downloading":
downloaded = d.get("downloaded_bytes", 0)
total = d.get("total_bytes") or d.get("total_bytes_estimate", 0)
@ -136,6 +130,7 @@ def download_hook(d: dict, bot_msg):
speed = remove_bash_color(d.get("_speed_str", "N/A"))
eta = remove_bash_color(d.get("_eta_str", d.get("eta")))
text = tqdm_progress("Downloading...", total, downloaded, speed, eta)
# debounce in here
edit_text(bot_msg, text)