tgbot-collection/ytdlbot
1
1
Fork 0
mirror of https://github.com/tgbot-collection/ytdlbot.git synced 2024-09-20 23:15:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

disable channel/user download

Browse source
This commit is contained in:
BennyThink 2022-02-04 16:52:49 +08:00
parent 1b12d1e56b
commit 6b6bb9db0a
No known key found for this signature in database
GPG key ID: 6CD0DBDA5235D481
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
ytdlbot/ytdl_bot.py
View file

@ -226,19 +226,25 @@ def vip_handler(client: "Client", message: "types.Message"):
@private_use @private_use
def download_handler(client: "Client", message: "types.Message"): def download_handler(client: "Client", message: "types.Message"):
# check remaining quota # check remaining quota
red = Redis()
chat_id = message.from_user.id chat_id = message.from_user.id
client.send_chat_action(chat_id, 'typing') client.send_chat_action(chat_id, 'typing')
Redis().user_count(chat_id) red.user_count(chat_id)
url = re.sub(r'/ytdl\s*', '', message.text) url = re.sub(r'/ytdl\s*', '', message.text)
logging.info("start %s", url) logging.info("start %s", url)
if not re.findall(r"^https?://", url.lower()): if not re.findall(r"^https?://", url.lower()):
Redis().update_metrics("bad_request") red.update_metrics("bad_request")
message.reply_text("I think you should send me a link.", quote=True) message.reply_text("I think you should send me a link.", quote=True)
return return
Redis().update_metrics("video_request") if re.findall(r"^https://www.youtube.com/channel/", VIP.extract_canonical_link(url)):
message.reply_text("Channel download is disabled now. Please send me individual video link.", quote=True)
red.update_metrics("reject_channel")
return
red.update_metrics("video_request")
text = bot_text.get_receive_link_text() text = bot_text.get_receive_link_text()
bot_msg: typing.Union["types.Message", "typing.Any"] = message.reply_text(text, quote=True) bot_msg: typing.Union["types.Message", "typing.Any"] = message.reply_text(text, quote=True)
client.send_chat_action(chat_id, 'upload_video') client.send_chat_action(chat_id, 'upload_video')