add a hint about users

This commit is contained in:
BennyThink 2022-03-04 20:28:41 +08:00
parent e35fc53807
commit fb0348e0a1
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481
3 changed files with 19 additions and 7 deletions

View File

@ -1,7 +1,7 @@
pyrogram==1.4.7
pyrogram==1.4.8
tgcrypto==1.2.3
yt-dlp==2022.2.4
APScheduler==3.8.1
APScheduler==3.9.1
beautifultable==1.0.1
ffmpeg-python==0.2.0
PyMySQL==1.0.2
@ -16,6 +16,6 @@ supervisor==4.2.4
tgbot-ping==1.0.4
redis==4.1.4
requests==2.27.1
tqdm==4.62.3
tqdm==4.63.0
requests-toolbelt==0.9.1
ffpb==0.4.1

View File

@ -116,7 +116,7 @@ def forward_video(url, client, bot_msg):
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)
caption, _ = gen_cap(bot_msg, url, obj)
res_msg.edit_text(caption, reply_markup=gen_video_markup())
red.update_metrics("cache_hit")
return True
@ -288,7 +288,7 @@ def upload_processor(client, bot_msg, url, vp_or_fid: "typing.Any[str, pathlib.P
chat_id = bot_msg.chat.id
red = Redis()
markup = gen_video_markup()
cap, meta = gen_cap(chat_id, url, vp_or_fid)
cap, meta = gen_cap(bot_msg, url, vp_or_fid)
settings = get_user_settings(str(chat_id))
if ARCHIVE_ID and isinstance(vp_or_fid, pathlib.Path):
chat_id = ARCHIVE_ID
@ -324,7 +324,18 @@ def upload_processor(client, bot_msg, url, vp_or_fid: "typing.Any[str, pathlib.P
return res_msg
def gen_cap(chat_id, url, video_path):
def gen_cap(bm, url, video_path):
chat_id = bm.chat.id
user = bm.chat
if user is None:
user_info = ""
else:
user_info = "@{}({})-{}".format(
user.username or "N/A",
user.first_name or "" + user.last_name or "",
user.id
)
if isinstance(video_path, pathlib.Path):
meta = get_metadata(video_path)
file_name = video_path.name
@ -340,7 +351,7 @@ def gen_cap(chat_id, url, video_path):
)
remain = bot_text.remaining_quota_caption(chat_id)
worker = get_dl_source()
cap = f"`{file_name}`\n\n{url}\n\nInfo: {meta['width']}x{meta['height']} {file_size}\t" \
cap = f"{user_info}\n`{file_name}`\n\n{url}\n\nInfo: {meta['width']}x{meta['height']} {file_size}\t" \
f"{meta['duration']}s\n{remain}\n{worker}\n{bot_text.custom_text}"
return cap, meta

View File

@ -282,6 +282,7 @@ def download_handler(client: "Client", message: "types.Message"):
time.sleep(e.x)
client.send_chat_action(chat_id, 'upload_video')
bot_msg.chat = message.chat
ytdl_download_entrance(bot_msg, client, url)