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

use quota then generate caption

Browse source
This commit is contained in:
BennyThink 2022-02-20 16:16:39 +08:00
parent f99addd94f
commit 2ce135d549
No known key found for this signature in database
GPG key ID: 6CD0DBDA5235D481
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ytdlbot/tasks.py
View file

@ -110,14 +110,14 @@ def forward_video(url, client, bot_msg):
if not res_msg:
raise ValueError("Failed to forward message")
obj = res_msg.document or res_msg.video or res_msg.audio
caption, _ = gen_cap(chat_id, url, obj)
res_msg.edit_text(caption, reply_markup=gen_video_markup())
if ENABLE_VIP:
file_size = getattr(obj, "file_size", None) \
or getattr(obj, "file_size", None) \
or getattr(obj, "file_size", 10)
# TODO: forward file size may exceed the limit
vip.use_quota(chat_id, file_size)
caption, _ = gen_cap(chat_id, url, obj)
res_msg.edit_text(caption, reply_markup=gen_video_markup())
red.update_metrics("cache_hit")
return True