2020-02-19 15:31:39 +00:00
|
|
|
|
""" Pagermaid plugin base. """
|
|
|
|
|
|
|
|
|
|
|
|
from os import remove
|
|
|
|
|
|
from os.path import exists
|
|
|
|
|
|
from youtube_dl import YoutubeDL
|
|
|
|
|
|
from re import compile as regex_compile
|
|
|
|
|
|
from pagermaid import bot, log
|
|
|
|
|
|
from pagermaid.listener import listener
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@listener(outgoing=True, command="ytdl",
|
|
|
|
|
|
description="YouTube downloader.",
|
|
|
|
|
|
parameters="<url>.")
|
|
|
|
|
|
async def ytdl(context):
|
|
|
|
|
|
url = context.arguments
|
|
|
|
|
|
reply = await context.get_reply_message()
|
|
|
|
|
|
reply_id = None
|
2020-02-20 08:08:50 +00:00
|
|
|
|
await context.edit("获取视频中 . . .")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
if reply:
|
|
|
|
|
|
reply_id = reply.id
|
|
|
|
|
|
if url is None:
|
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
youtube_pattern = regex_compile(r"^(http(s)?://)?((w){3}.)?youtu(be|.be)?(\.com)?/.+")
|
|
|
|
|
|
if youtube_pattern.match(url):
|
|
|
|
|
|
if not await fetch_youtube_video(url, context.chat_id, reply_id):
|
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 视频下载失败。")
|
|
|
|
|
|
await log(f"已拉取UTB视频，地址： {url}.")
|
2020-02-20 08:08:50 +00:00
|
|
|
|
await context.edit("视频获取成功！")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def fetch_youtube_video(url, chat_id, reply_id):
|
|
|
|
|
|
""" Extracts and uploads YouTube video. """
|
|
|
|
|
|
youtube_dl_options = {
|
|
|
|
|
|
'format': 'bestvideo[height=720]+bestaudio/best',
|
|
|
|
|
|
'outtmpl': "video.%(ext)s",
|
|
|
|
|
|
'postprocessors': [{
|
|
|
|
|
|
'key': 'FFmpegVideoConvertor',
|
|
|
|
|
|
'preferedformat': 'mp4'
|
|
|
|
|
|
}]
|
|
|
|
|
|
}
|
|
|
|
|
|
YoutubeDL(youtube_dl_options).download([url])
|
|
|
|
|
|
if not exists("video.mp4"):
|
|
|
|
|
|
return False
|
|
|
|
|
|
await bot.send_file(
|
|
|
|
|
|
chat_id,
|
|
|
|
|
|
"video.mp4",
|
|
|
|
|
|
reply_to=reply_id
|
|
|
|
|
|
)
|
|
|
|
|
|
remove("video.mp4")
|
|
|
|
|
|
return True
|
