async def yt_dl(event):
"""
Download videos from YouTube with their url in multiple formats.
**{prefix}ytdl link1 link2 link3 [kwargs]**
Stream and progress are set to True, while update is at 10% by default.
**Arguments:**
`format` (The format to convert/download the video in),
`delete` (Whether to delete the local files or not),
`upload` (Whether to upload the downloaded files or not),
`update` (The percentage to update the progress at),
`stream` (Whether to upload the files as streamable or not),
`progress` (Whether to update the progress by edits or not)
"""
match = event.matches[0].group(1)
force_document = True
if not match:
await event.answer(
"`.ytdl <url>` or `.ytdl <url1> .. <urln> format=<fmt>`")
return
args, kwargs = await client.parse_arguments(match)
fmt = kwargs.get('format', kwargs.get('fmt', False))
auto_delete = kwargs.get('delete', False)
upload = kwargs.get('upload', True)
round_message = kwargs.get('round_message', kwargs.get('round', False))
update = kwargs.get('update', 10)
supports_streaming = kwargs.get('supports_streaming',
kwargs.get('stream', True))
progress = kwargs.get('progress', True)
if not upload and auto_delete:
await event.answer(
"`The void doesn't make sense! Either don't upload or delete.`")
return
ffmpeg = await is_ffmpeg_there()
params = copy.deepcopy(ydl_opts)
warnings = []
if fmt:
fmt = fmt.strip().lower()
if fmt == 'listformats':
fmts = []
for url in args:
info = await extract_info(
client.loop, concurrent.futures.ThreadPoolExecutor(),
params, url)
if isinstance(info, dict):
fmts.append(await list_formats(info))
elif isinstance(info, str):
warnings.append(info)
else:
warnings.append(
f'```{await client.get_traceback(info)}```')
if fmts:
text = "**Formats:**\n"
text += "\n\n".join(fmts)
await event.answer(text)
if warnings:
text = "**Warnings:**\n"
text += "\n\n".join(f"```{w}```" for w in warnings)
reply = True if fmts else False
await event.answer(text, reply=reply)
return
elif fmt in audioFormats and ffmpeg:
params.update(format='bestaudio')
params['postprocessors'].append({
'key': 'FFmpegExtractAudio',
'preferredcodec': fmt,
'preferredquality': '320',
})
elif fmt in videoFormats and ffmpeg:
params['postprocessors'].append({
'key': 'FFmpegVideoConvertor',
'preferedformat': fmt
})
else:
params.update(format=fmt)
if ffmpeg:
params.update(key='FFmpegMetadata')
if fmt in ['mp3', 'mp4', 'm4a']:
params.update(writethumbnail=True)
params['postprocessors'].append({'key': 'EmbedThumbnail'})
if progress:
event.media = None
progress = ProgressHook(event, update)
params['progress_hooks'].append(progress.hook)
progress_cb = ProgressCallback(event, update=update)
for url in args:
await event.answer(f"`Processing {url}...`")
output = await extract_info(
loop=client.loop,
ydl_opts=params,
url=url,
download=True,
executor=concurrent.futures.ThreadPoolExecutor())
if isinstance(output, str):
result = warning + output if not ffmpeg else output
warnings.append(result)
elif isinstance(output, BaseException):
warnings.append(f'```{await client.get_traceback(output)}```')
elif output is None:
await event.answer('`Oh oh, sum ting went wong`')
return
else:
if upload:
path, thumb, info = output
title = info.get('title', info.get('id', 'Unknown title'))
url = info.get('webpage_url', None)
href = f"[{title}]({url})"
text = success.format(href)
result = warning + text if not ffmpeg else text
dl = io.open(path, 'rb')
if progress:
progress_cb.filen = title
uploaded = await client.fast_upload_file(
dl, progress_cb.up_progress if progress else None)
dl.close()
attributes, mime_type = await fix_attributes(
path, info, round_message, supports_streaming)
media = types.InputMediaUploadedDocument(
file=uploaded,
mime_type=mime_type,
attributes=attributes,
thumb=await client.upload_file(thumb) if thumb else None)
if supports_streaming and path.suffix == '.mp4':
force_document = False
if round_message:
force_document = False
await client.send_file(event.chat_id,
media,
caption=href,
force_document=force_document)
if thumb:
os.remove(thumb)
if auto_delete:
os.remove(path)
else:
if thumb:
os.remove(thumb)
if warnings:
text = "**Warnings:**\n"
text += ",\n\n".join(warnings)
await event.answer(text)
else:
await event.delete()
async def yt_dl(event):
"""
Download videos from YouTube with their url in multiple formats.
**{prefix}ytdl link1 link2 link3 [kwargs]**
Stream and progress are set to True, while update is at 10% by default.
**Arguments:**
`format` (The format to convert/download the video in),
`delete` (Whether to delete the local files or not),
`upload` (Whether to upload the downloaded files or not),
`update` (The percentage to update the progress at),
`stream` (Whether to upload the files as streamable or not),
`progress` (Whether to update the progress by edits or not)
"""
match = event.matches[0].group(1)
force_document = True
if not match:
await event.answer(
"`.ytdl <url>` or `.ytdl <url1> .. <urln> format=<fmt>`")
return
args, kwargs = await client.parse_arguments(match)
fmt = kwargs.get("format", kwargs.get("fmt", False))
auto_delete = kwargs.get("delete", False)
upload = kwargs.get("upload", True)
progress = kwargs.get("progress", True)
round_message = kwargs.get("round_message", kwargs.get("round", False))
update = kwargs.get("update", 10)
supports_streaming = kwargs.get("supports_streaming",
kwargs.get("stream", True))
if not upload and auto_delete:
await event.answer("`The void doesn't make sense!\
\nEither don't upload or delete.`")
return
ffmpeg = await is_ffmpeg_there()
params = copy.deepcopy(ydl_opts)
params["postprocessor_args"] = (
["-preset", "ultrafast", "-tune", "fastdecode"] if ffmpeg else [])
warnings = []
if fmt:
fmt = fmt.strip().lower()
if fmt in audioFormats and ffmpeg:
params.update(format="bestaudio")
params["postprocessors"].append({
"key": "FFmpegExtractAudio",
"preferredcodec": fmt,
"preferredquality": "320",
})
if fmt in ["mp3", "m4a"]:
params["postprocessors"].append({"key": "EmbedThumbnail"})
if fmt in ["mp3"]:
params["postprocessor_args"] += [
"-write_id3v1",
"1",
"-id3v2_version",
"3",
]
elif fmt in videoFormats and ffmpeg:
params["postprocessors"].append({
"key": "FFmpegVideoConvertor",
"preferedformat": fmt
})
if fmt in ["mp4"]:
params["postprocessors"].append({"key": "EmbedThumbnail"})
else:
params.update(format=fmt)
if ffmpeg:
params.update(key="FFmpegMetadata")
else:
fmts = []
for url in args:
info = await extract_info(client.loop,
concurrent.futures.ThreadPoolExecutor(),
params, url)
if isinstance(info, dict):
fmts.append(await list_formats(info))
elif isinstance(info, str):
warnings.append(info)
else:
warnings.append(f"```{await client.get_traceback(info)}```")
if fmts:
text = "**Formats:**\n"
text += ",\n\n".join(f"```{f}```" for f in fmts)
await event.answer(text)
if warnings:
text = "**Warnings:**\n"
text += ",\n\n".join(f"```{w}```" for w in warnings)
reply = bool(fmts)
await event.answer(text, reply=reply)
return
if progress:
event.media = None
progress = ProgressHook(event, update)
params["progress_hooks"].append(progress.hook)
progress_cb = ProgressCallback(event, update=update)
for url in args:
await event.answer(f"`Processing {url}...`")
output = await extract_info(
loop=client.loop,
ydl_opts=params,
url=url,
download=True,
executor=concurrent.futures.ThreadPoolExecutor(),
)
if isinstance(output, str):
result = warning + output if not ffmpeg else output
warnings.append(result)
elif isinstance(output, BaseException):
warnings.append(f"```{await client.get_traceback(output)}```")
elif output is None:
await event.answer("`Oh oh, some thing went wrong!`")
return
else:
path, thumb, info = output
if upload:
title = info.get("title", info.get("id", "Unknown title"))
url = info.get("webpage_url", None)
href = f"[{title}]({url})"
text = success.format(href)
result = warning + text if not ffmpeg else text
dl = io.open(path, "rb")
if progress:
progress_cb.filen = title
uploaded = await client.fast_upload_file(
dl, progress_cb.up_progress if progress else None)
dl.close()
attributes, mime_type = await fix_attributes(
path, info, round_message, supports_streaming)
media = types.InputMediaUploadedDocument(
file=uploaded,
mime_type=mime_type,
attributes=attributes,
thumb=await client.upload_file(thumb) if thumb else None,
)
if supports_streaming and path.suffix == ".mp4":
force_document = False
if round_message:
force_document = False
await client.send_file(
event.chat_id,
media,
caption=href,
reply_to=event.reply_to_msg_id,
force_document=force_document,
)
if thumb:
os.remove(thumb)
if auto_delete:
os.remove(path)
else:
if thumb:
os.remove(thumb)
if warnings:
text = "**Warnings:**\n"
text += ",\n\n".join(warnings)
await event.answer(text)
else:
await event.delete()