add shuffle play

bot.py

@@ -4,6 +4,7 @@ from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By
 from time import sleep
 import disnake
+import random
 from urllib.parse import urlparse
 from disnake.ext import commands
 from disnake import TextInputStyle
@@ -19,6 +20,7 @@ handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(me
 logger.addHandler(handler)
 
 queue = []
+shuffle = False
 load_dotenv()
 options = Options()
 options.profile = webdriver.FirefoxProfile(os.getenv("PROFILE_PATH"))
@@ -35,8 +37,11 @@ async def on_ready():
 def play_video(videourl):
     driver.get(videourl)
     sleep(4)
-    elem = driver.find_element(By.CLASS_NAME, "ytp-fullscreen-button")
-    elem.send_keys(Keys.RETURN)
+    try:
+        elem = driver.find_element(By.CLASS_NAME, "ytp-fullscreen-button")
+        elem.send_keys(Keys.RETURN)
+    except Exception:
+        return #if this errors there is no fullscreen options, such as playlists, so skip the link
     #guess I don't need this
     #sleep(1.5)
     #elem = driver.find_element(By.XPATH, '//button[@aria-keyshortcuts="k"]')
@@ -52,7 +57,10 @@ def play_video(videourl):
         elem.send_keys(Keys.RETURN)
     except Exception:
         pass
-    endscreen = driver.find_element(By.CLASS_NAME, "html5-endscreen")
+    try:
+        endscreen = driver.find_element(By.CLASS_NAME, "html5-endscreen")
+    except Exception:
+        return #same as above
     while str(endscreen.get_attribute('style')) == "display: none;":
         pass
     sleep(2)
@@ -75,6 +83,23 @@ async def play(inter: disnake.AppCmdInter, link: str):
         await inter.edit_original_response(f"This bot only accepts youtube links")
         return
 
+@bot.slash_command(
+    name="shuffle",
+    description="toggles shuffle on or off, the queue cannot be unshuffled once it is shuffled",
+)
+async def shuffleplay(inter: disnake.AppCmdInter, toggle: str = commands.Param(choices=["on", "off"])):
+    await inter.response.defer(ephemeral=True)
+    if toggle == "on":
+        global shuffle
+        shuffle = True
+        await inter.edit_original_response(f"shuffle enabled")
+        return
+    else:
+        global shuffle
+        shuffle = False
+        await inter.edit_original_response(f"shuffle disabled")
+        return
+
 @bot.slash_command(
     name="queue",
     description="list the videos in queue",
@@ -138,7 +163,11 @@ async def remove(inter: disnake.AppCmdInter, toremove: int):
 
 async def queuehandler():
     loop = asyncio.get_running_loop()
+    global queue
     while queue:
+        if shuffle:
+            random.shuffle(queue)
+        print(queue)
         driver.maximize_window()
         await loop.run_in_executor(None, play_video, queue[0])
         queue.pop(0)