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

fix: added error handling for if you try to use the wrong website

Browse source
This commit is contained in:
Seaswimmer 2023-08-02 13:38:57 -04:00
parent b7e73178d2
commit 93102494c0
No known key found for this signature in database
GPG key ID: 5019678FD9CF50D8
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
musicdownloader/musicdownloader.py
View file

@ -3,7 +3,7 @@ import re
import discord import discord
import os import os
import sqlite3 import sqlite3
from yt_dlp import YoutubeDL from yt_dlp import YoutubeDL, utils
from redbot.core import commands, checks, Config, data_manager from redbot.core import commands, checks, Config, data_manager
class MusicDownloader(commands.Cog): class MusicDownloader(commands.Cog):
@ -149,7 +149,11 @@ class MusicDownloader(commands.Cog):
else: else:
msg = ctx.send msg = ctx.send
message = await msg("YouTube Downloader started!") message = await msg("YouTube Downloader started!")
ytdlp_output = youtube_download(self, url, data_path, message) try:
ytdlp_output = youtube_download(self, url, data_path, message)
except utils.ExtractorError:
await ctx.send("Please provide a link to YouTube and not another site.")
return
full_filename = os.path.join(data_path, ytdlp_output[0]) full_filename = os.path.join(data_path, ytdlp_output[0])
while not os.path.isfile(full_filename): while not os.path.isfile(full_filename):
await asyncio.sleep(0.2) await asyncio.sleep(0.2)
@ -159,7 +163,6 @@ class MusicDownloader(commands.Cog):
complete_message = await ctx.send(content="YouTube Downloader completed!\nDownloaded file:", file=discord.File(file, ytdlp_output[0])) complete_message = await ctx.send(content="YouTube Downloader completed!\nDownloaded file:", file=discord.File(file, ytdlp_output[0]))
except ValueError: except ValueError:
complete_message = await ctx.send(content="YouTube Downloader completed, but the audio file was too large to upload.") complete_message = await ctx.send(content="YouTube Downloader completed, but the audio file was too large to upload.")
return
file.close() file.close()
if delete is True or await self.bot.is_owner(ctx.author) is False: if delete is True or await self.bot.is_owner(ctx.author) is False:
if ytdlp_output[1] is False: if ytdlp_output[1] is False: