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

Add compress option

Browse source
This commit is contained in:
nexy7574 2023-05-11 18:34:32 +01:00
parent 9559f40dca
commit 696e884dd2
1 changed files with 39 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
cogs/other.py
View file

@ -1113,9 +1113,17 @@ class OtherCog(commands.Cog):
default=""
) = "",
upload_log: bool = False,
compress_if_possible: bool = False
):
"""Downloads a video using youtube-dl"""
await ctx.defer()
if compress_if_possible and not await self.bot.is_user(ctx.user):
return await ctx.edit(
embed=discord.Embed(
description="You can't compress files.",
color=discord.Color.red()
)
)
formats = await self.list_formats(url)
if list_formats:
embeds = []
@ -1260,6 +1268,37 @@ class OtherCog(commands.Cog):
continue
st = file.stat().st_size
if st / 1024 / 1024 >= MAX_SIZE_MB:
if compress_if_possible:
ffmpeg_command = [
"ffmpeg",
"-i",
str(file),
"-c",
"copy",
"-crf",
"25",
"-preset",
"medium",
str(file.with_name(file.name + '.compressed' + file.suffix))
]
try:
await self.bot.loop.run_in_executor(
None,
partial(
subprocess.run,
ffmpeg_command,
capture_output=True,
check=True
)
)
except subprocess.CalledProcessError as e:
pass
else:
file = file.with_name(file.name + '.compressed' + file.suffix)
st = file.stat().st_size
if st / 1024 / 1024 <= MAX_SIZE_MB:
files.append(discord.File(file, file.name))
continue
units = ["B", "KB", "MB", "GB", "TB"]
st_r = st
while st_r > 1024: