nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Embed yt_dl

Browse source
This commit is contained in:
Nexus 2023-04-28 12:05:11 +01:00
parent 0215986e54
commit a2ccb56316
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 32 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
cogs/other.py
View file

@ -752,6 +752,7 @@ class OtherCog(commands.Cog):
upload_log: bool = True
):
"""Downloads a video from <URL> using youtube-dl"""
await ctx.defer()
with tempfile.TemporaryDirectory(prefix="jimmy-ytdl-") as tempdir:
video_format = video_format.lower()
OUTPUT_FILE = str(Path(tempdir) / f"{ctx.user.id}.%(ext)s")
@ -770,8 +771,15 @@ class OtherCog(commands.Cog):
else:
options.extend(["--format", f"(bv*+ba/b/ba)[filesize<={MAX_SIZE}M]"])
await ctx.defer()
await ctx.edit(content="Downloading video...")
await ctx.edit(
embed=discord.Embed(
description="\u200b"
).set_author(
name="Downloading...",
icon_url="https://cdn.discordapp.com/emojis/1101463077586735174.gif?v=1",
url=url
)
)
try:
venv = Path.cwd() / "venv" / ("Scripts" if os.name == "nt" else "bin")
if venv:
@ -786,14 +794,20 @@ class OtherCog(commands.Cog):
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
stdout, stderr = await process.communicate()
stdout_log = io.BytesIO(stdout)
stdout_log_file = discord.File(stdout_log, filename="stdout.txt")
stderr_log = io.BytesIO(stderr)
stderr_log_file = discord.File(stderr_log, filename="stderr.txt")
await process.wait()
async with ctx.channel.typing():
stdout, stderr = await process.communicate()
stdout_log = io.BytesIO(stdout)
stdout_log_file = discord.File(stdout_log, filename="stdout.txt")
stderr_log = io.BytesIO(stderr)
stderr_log_file = discord.File(stderr_log, filename="stderr.txt")
await process.wait()
except FileNotFoundError:
return await ctx.edit(content="Downloader not found.")
return await ctx.edit(
embed=discord.Embed(
description="Downloader not found.",
color=discord.Color.red()
)
)
if process.returncode != 0:
files = [
@ -822,7 +836,7 @@ class OtherCog(commands.Cog):
"\t- Resolution: {0[resolution]}\n"
"\t- Size: {1!s}MB".format(fmt, fs)
)
await ctx.edit(content="Invalid format. Available formats:")
await ctx.edit(content="Invalid format. Available formats:", embed=None)
for page in paginator.pages:
await ctx.send(page)
return await ctx.send(files=files)
@ -848,8 +862,14 @@ class OtherCog(commands.Cog):
continue
if not files:
return await ctx.edit(content="No files found.")
await ctx.edit(content="Here's your video!", files=files)
return await ctx.edit(embed=discord.Embed(description="No files found.", color=discord.Colour.red()))
await ctx.edit(
embed=discord.Embed(
title="Here's your video!",
color=discord.Colour.green()
),
files=files
)
@commands.slash_command(name="text-to-mp3")
@commands.cooldown(5, 600, commands.BucketType.user)