mirror of
https://github.com/omg-xtao/ytdlbot.git
synced 2024-11-16 11:52:14 +00:00
220 lines
7.8 KiB
Python
220 lines
7.8 KiB
Python
#!/usr/local/bin/python3
|
|
# coding: utf-8
|
|
|
|
# ytdlbot - new.py
|
|
# 8/14/21 14:37
|
|
#
|
|
|
|
__author__ = "Benny <benny.think@gmail.com>"
|
|
|
|
import logging
|
|
import os
|
|
import pathlib
|
|
import re
|
|
import tempfile
|
|
import typing
|
|
|
|
import ffmpeg
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from pyrogram import Client, filters, types
|
|
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
|
from tgbot_ping import get_runtime
|
|
|
|
from config import APP_HASH, APP_ID, ENABLE_VIP, OWNER, TOKEN, WORKERS
|
|
from constant import BotText
|
|
from downloader import convert_flac, sizeof_fmt, upload_hook, ytdl_download
|
|
from limit import Redis, verify_payment
|
|
|
|
|
|
def customize_logger(logger: "list"):
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(filename)s [%(levelname)s]: %(message)s')
|
|
for log in logger:
|
|
logging.getLogger(log).setLevel(level=logging.WARNING)
|
|
|
|
|
|
def create_app(session="ytdl", workers=WORKERS):
|
|
_app = Client(session, APP_ID, APP_HASH,
|
|
bot_token=TOKEN, workers=workers)
|
|
|
|
return _app
|
|
|
|
|
|
customize_logger(["pyrogram.client", "pyrogram.session.session", "pyrogram.client", "pyrogram.connection.connection"])
|
|
app = create_app()
|
|
bot_text = BotText()
|
|
|
|
|
|
def get_metadata(video_path):
|
|
width, height, duration = 1280, 720, 0
|
|
try:
|
|
video_streams = ffmpeg.probe(video_path, select_streams="v")
|
|
for item in video_streams.get("streams", []):
|
|
height = item["height"]
|
|
width = item["width"]
|
|
duration = int(float(video_streams["format"]["duration"]))
|
|
except Exception as e:
|
|
logging.error(e)
|
|
return dict(height=height, width=width, duration=duration)
|
|
|
|
|
|
@app.on_message(filters.command(["start"]))
|
|
def start_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
logging.info("Welcome to youtube-dl bot!")
|
|
client.send_chat_action(chat_id, "typing")
|
|
greeting = bot_text.get_vip_greeting(chat_id)
|
|
client.send_message(message.chat.id, greeting + bot_text.start + "\n\n" + bot_text.remaining_quota_caption(chat_id))
|
|
|
|
|
|
@app.on_message(filters.command(["help"]))
|
|
def help_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
client.send_chat_action(chat_id, "typing")
|
|
client.send_message(chat_id, bot_text.help, disable_web_page_preview=True)
|
|
|
|
|
|
@app.on_message(filters.command(["ping"]))
|
|
def ping_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
client.send_chat_action(chat_id, "typing")
|
|
if os.uname().sysname == "Darwin" or ".heroku" in os.getenv("PYTHONHOME", ""):
|
|
bot_info = "ping unavailable."
|
|
else:
|
|
bot_info = get_runtime("botsrunner_ytdl_1", "YouTube-dl")
|
|
if message.chat.username == OWNER:
|
|
client.send_document(chat_id, Redis().generate_file(), caption=bot_info)
|
|
else:
|
|
client.send_message(chat_id, f"{bot_info}")
|
|
|
|
|
|
@app.on_message(filters.command(["about"]))
|
|
def help_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
client.send_chat_action(chat_id, "typing")
|
|
client.send_message(chat_id, bot_text.about)
|
|
|
|
|
|
@app.on_message(filters.command(["terms"]))
|
|
def terms_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
client.send_chat_action(chat_id, "typing")
|
|
client.send_message(chat_id, bot_text.terms)
|
|
|
|
|
|
@app.on_message(filters.command(["vip"]))
|
|
def vip_handler(client: "Client", message: "types.Message"):
|
|
chat_id = message.chat.id
|
|
text = message.text.strip()
|
|
client.send_chat_action(chat_id, "typing")
|
|
if text == "/vip":
|
|
client.send_message(chat_id, bot_text.vip, disable_web_page_preview=True)
|
|
else:
|
|
bm: typing.Union["types.Message", "typing.Any"] = message.reply_text(bot_text.vip_pay, quote=True)
|
|
unique = text.replace("/vip", "").strip()
|
|
msg = verify_payment(chat_id, unique)
|
|
bm.edit_text(msg)
|
|
|
|
|
|
@app.on_message(filters.incoming)
|
|
def download_handler(client: "Client", message: "types.Message"):
|
|
# check remaining quota
|
|
chat_id = message.chat.id
|
|
Redis().user_count(chat_id)
|
|
|
|
if message.chat.type != "private" and not message.text.lower().startswith("/ytdl"):
|
|
logging.warning("%s, it's annoying me...🙄️ ", message.text)
|
|
return
|
|
|
|
url = re.sub(r'/ytdl\s*', '', message.text)
|
|
logging.info("start %s", url)
|
|
|
|
if not re.findall(r"^https?://", url.lower()):
|
|
Redis().update_metrics("bad_request")
|
|
message.reply_text("I think you should send me a link.", quote=True)
|
|
return
|
|
|
|
Redis().update_metrics("video_request")
|
|
bot_msg: typing.Union["types.Message", "typing.Any"] = message.reply_text("Processing", quote=True)
|
|
client.send_chat_action(chat_id, 'upload_video')
|
|
temp_dir = tempfile.TemporaryDirectory()
|
|
|
|
result = ytdl_download(url, temp_dir.name, bot_msg)
|
|
logging.info("Download complete.")
|
|
|
|
markup = InlineKeyboardMarkup(
|
|
[
|
|
[ # First row
|
|
InlineKeyboardButton( # Generates a callback query when pressed
|
|
"audio",
|
|
callback_data="audio"
|
|
)
|
|
]
|
|
]
|
|
)
|
|
|
|
if result["status"]:
|
|
client.send_chat_action(chat_id, 'upload_document')
|
|
video_paths = result["filepath"]
|
|
bot_msg.edit_text('Download complete. Sending now...')
|
|
for video_path in video_paths:
|
|
filename = pathlib.Path(video_path).name
|
|
remain = bot_text.remaining_quota_caption(chat_id)
|
|
size = sizeof_fmt(os.stat(video_path).st_size)
|
|
meta = get_metadata(video_path)
|
|
client.send_video(chat_id, video_path,
|
|
supports_streaming=True,
|
|
caption=f"`{filename}`\n\n{url}\n\nsize: {size}\n\n{remain}",
|
|
progress=upload_hook, progress_args=(bot_msg,),
|
|
reply_markup=markup,
|
|
**meta
|
|
)
|
|
Redis().update_metrics("video_success")
|
|
bot_msg.edit_text('Download success!✅')
|
|
else:
|
|
client.send_chat_action(chat_id, 'typing')
|
|
tb = result["error"][0:4000]
|
|
bot_msg.edit_text(f"Download failed!❌\n\n```{tb}```", disable_web_page_preview=True)
|
|
|
|
temp_dir.cleanup()
|
|
|
|
|
|
@app.on_callback_query()
|
|
def answer(client: "Client", callback_query: types.CallbackQuery):
|
|
callback_query.answer(f"Converting to audio...please wait patiently")
|
|
Redis().update_metrics("audio_request")
|
|
|
|
msg = callback_query.message
|
|
|
|
chat_id = msg.chat.id
|
|
mp4_name = msg.video.file_name # 'youtube-dl_test_video_a.mp4'
|
|
flac_name = mp4_name.replace("mp4", "m4a")
|
|
|
|
with tempfile.NamedTemporaryFile() as tmp:
|
|
logging.info("downloading to %s", tmp.name)
|
|
client.send_chat_action(chat_id, 'record_video_note')
|
|
client.download_media(msg, tmp.name)
|
|
logging.info("downloading complete %s", tmp.name)
|
|
# execute ffmpeg
|
|
client.send_chat_action(chat_id, 'record_audio')
|
|
flac_tmp = convert_flac(flac_name, tmp)
|
|
client.send_chat_action(chat_id, 'upload_audio')
|
|
client.send_audio(chat_id, flac_tmp)
|
|
Redis().update_metrics("audio_success")
|
|
os.unlink(flac_tmp)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
scheduler = BackgroundScheduler()
|
|
scheduler.add_job(Redis().reset_today, 'cron', hour=0, minute=0)
|
|
scheduler.start()
|
|
banner = f"""
|
|
▌ ▌ ▀▛▘ ▌ ▛▀▖ ▜ ▌
|
|
▝▞ ▞▀▖ ▌ ▌ ▌ ▌ ▌ ▛▀▖ ▞▀▖ ▌ ▌ ▞▀▖ ▌ ▌ ▛▀▖ ▐ ▞▀▖ ▝▀▖ ▞▀▌
|
|
▌ ▌ ▌ ▌ ▌ ▌ ▌ ▌ ▌ ▌ ▛▀ ▌ ▌ ▌ ▌ ▐▐▐ ▌ ▌ ▐ ▌ ▌ ▞▀▌ ▌ ▌
|
|
▘ ▝▀ ▝▀▘ ▘ ▝▀▘ ▀▀ ▝▀▘ ▀▀ ▝▀ ▘▘ ▘ ▘ ▘ ▝▀ ▝▀▘ ▝▀▘
|
|
|
|
By @BennyThink, VIP mode: {ENABLE_VIP}
|
|
"""
|
|
print(banner)
|
|
app.run()
|