nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Fix audio-only downloads

Browse source
This commit is contained in:
Nexus 2024-01-24 21:17:37 +00:00
parent 6e04f4a97d
commit b815f96c2e
1 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/cogs/ytdl.py
View file

@ -341,8 +341,23 @@ class YTDLCog(commands.Cog):
delete_after=120,
)
try:
file = next(temp_dir.glob("*." + extracted_info["ext"]))
if audio_only is False:
file = next(temp_dir.glob("*." + extracted_info["ext"]))
else:
# can be .opus, .m4a, .mp3, .ogg, .oga
for _file in temp_dir.iterdir():
if _file.suffix in (".opus", ".m4a", ".mp3", ".ogg", ".oga", ".aac", ".wav"):
file = _file
break
else:
raise StopIteration
except StopIteration:
self.log.warning(
"Failed to locate downloaded file. Was supposed to be looking for a file extension of "
"%r amongst files %r, however none were found.",
extracted_info["ext"],
list(map(str, temp_dir.iterdir()))
)
return await ctx.edit(
embed=discord.Embed(
title="Error",
@ -381,7 +396,7 @@ class YTDLCog(commands.Cog):
"-movflags",
"faststart",
"-b:a",
"48k",
"64k",
"-y",
"-strict",
"2",