From 93102494c0d5e51bae0c201b33294db85dab4893 Mon Sep 17 00:00:00 2001
From: SeaswimmerTheFsh <seaswimmerthefsh@gmail.com>
Date: Wed, 2 Aug 2023 13:38:57 -0400
Subject: [PATCH] fix: added error handling for if you try to use the wrong
 website

---
 musicdownloader/musicdownloader.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/musicdownloader/musicdownloader.py b/musicdownloader/musicdownloader.py
index 86325b3..ec9865b 100644
--- a/musicdownloader/musicdownloader.py
+++ b/musicdownloader/musicdownloader.py
@@ -3,7 +3,7 @@ import re
 import discord
 import os
 import sqlite3
-from yt_dlp import YoutubeDL
+from yt_dlp import YoutubeDL, utils
 from redbot.core import commands, checks, Config, data_manager
 
 class MusicDownloader(commands.Cog):
@@ -149,7 +149,11 @@ class MusicDownloader(commands.Cog):
         else:
             msg = ctx.send
         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])
         while not os.path.isfile(full_filename):
             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]))
                 except ValueError:
                     complete_message = await ctx.send(content="YouTube Downloader completed, but the audio file was too large to upload.")
-                    return
             file.close()
             if delete is True or await self.bot.is_owner(ctx.author) is False:
                 if ytdlp_output[1] is False: