cswimr/GalaxyCogs
cswimr/GalaxyCogs
1
1
Fork 0
Code Issues 2 Pull requests Projects Wiki Activity Actions

fix(youtubedownloader): fixed downloading the first video stream
Some checks failed
Pylint / Pylint (3.10) (push) Failing after 53s
Details

Browse source
This commit is contained in:
Seaswimmer 2023-10-24 00:28:41 -04:00
parent af27593fa9
commit 229aaa77d5
No known key found for this signature in database
GPG key ID: 5019678FD9CF50D8
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
youtubedownloader/youtubedownloader.py
View file

@ -67,18 +67,16 @@ class YouTubeDownloader(commands.Cog):
yt = YouTube(url=url) yt = YouTube(url=url)
filename = f"{yt.title} ({yt.video_id})" filename = f"{yt.title} ({yt.video_id})"
if audio_only: if audio_only:
yt.streams.filter(only_audio=True, mime_type='audio/mp4') stream = yt.streams.filter(only_audio=True, mime_type='audio/mp4').order_by('abr')
yt.streams.order_by('abr')
filename_format = filename + ".m4a" filename_format = filename + ".m4a"
else: else:
yt.streams.filter(progressive=True, mime_type="video/mp4") stream = yt.streams.filter(progressive=True, mime_type='video/mp4').order_by('res')
yt.streams.order_by('resolution')
filename_format = filename + ".mp4" filename_format = filename + ".mp4"
if os.path.isfile(path + f"{os.sep}{filename_format}"): if os.path.isfile(path + f"{os.sep}{filename_format}"):
previously_existed = True previously_existed = True
else: else:
previously_existed = False previously_existed = False
return yt.streams.first().download(filename=filename, output_path=path), previously_existed, filename_format return stream.first().download(filename=filename, output_path=path), previously_existed, filename_format
data_path = await self.config.save_directory() data_path = await self.config.save_directory()
if subfolder and await self.bot.is_owner(ctx.author): if subfolder and await self.bot.is_owner(ctx.author):
data_path = os.path.join(data_path, subfolder) data_path = os.path.join(data_path, subfolder)