Files
Ultroid-fork/plugins/youtube.py
Devesh Pal 90ad5646ff Ultroid v0.5 Updates
Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com>
Co-authored-by: Aditya <xditya@ultroid.tech>
Co-authored-by: Kaif <88398455+kaif-00z@users.noreply.github.com>
Co-authored-by: Aditya <me@xditya.me>
Co-authored-by: 1Danish-00 <danish@ultroid.tech>
Co-authored-by: smartman_ru <bblkovo@gmail.com>
2022-03-20 00:57:54 +05:30

86 lines
2.6 KiB
Python

# Ultroid - UserBot
# Copyright (C) 2021-2022 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
"""
✘ Commands Available -
• `{i}yta <(youtube/any) link>`
Download audio from the link.
• `{i}ytv <(youtube/any) link>`
Download video from the link.
• `{i}ytsa <(youtube) search query>`
Search and download audio from youtube.
• `{i}ytsv <(youtube) search query>`
Search and download video from youtube.
"""
from pyUltroid.functions.ytdl import download_yt, get_yt_link
from . import get_string, requests, ultroid_cmd
@ultroid_cmd(
pattern="yt(a|v|sa|sv) ?(.*)",
)
async def download_from_youtube_(event):
ytd = {
"prefer_ffmpeg": True,
"addmetadata": True,
"geo-bypass": True,
"nocheckcertificate": True,
}
opt = event.pattern_match.group(1).strip()
xx = await event.eor(get_string("com_1"))
if opt == "a":
ytd["format"] = "bestaudio"
ytd["outtmpl"] = "%(id)s.m4a"
url = event.pattern_match.group(2)
if not url:
return await xx.eor(get_string("youtube_1"))
try:
requests.get(url)
except BaseException:
return await xx.eor(get_string("youtube_2"))
elif opt == "v":
ytd["format"] = "best"
ytd["outtmpl"] = "%(id)s.mp4"
ytd["postprocessors"] = [{"key": "FFmpegMetadata"}]
url = event.pattern_match.group(2)
if not url:
return await xx.eor(get_string("youtube_3"))
try:
requests.get(url)
except BaseException:
return await xx.eor(get_string("youtube_4"))
elif opt == "sa":
ytd["format"] = "bestaudio"
ytd["outtmpl"] = "%(id)s.m4a"
try:
query = event.text.split(" ", 1)[1]
except IndexError:
return await xx.eor(get_string("youtube_5"))
url = get_yt_link(query)
if not url:
return await xx.edit(get_string("unspl_1"))
await xx.eor(get_string("youtube_6"))
elif opt == "sv":
ytd["format"] = "best"
ytd["outtmpl"] = "%(id)s.mp4"
ytd["postprocessors"] = [{"key": "FFmpegMetadata"}]
try:
query = event.text.split(" ", 1)[1]
except IndexError:
return await xx.eor(get_string("youtube_7"))
url = get_yt_link(query)
if not url:
return await xx.edit(get_string("unspl_1"))
await xx.eor(get_string("youtube_8"))
else:
return
await download_yt(xx, url, ytd)