insert/discord-yt-remote
1
0
Fork 0
Code Issues Pull requests Projects Activity

Add timeout

Browse source
This commit is contained in:
insert 2024-07-29 13:47:12 -04:00
parent 27f0bc28ce
commit 5d647ce4c0
Signed by: insert
GPG key ID: A70775C389ACF105
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
bot.py
View file

@ -3,6 +3,7 @@ from selenium import webdriver
from selenium.webdriver.common.keys import Keys from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By from selenium.webdriver.common.by import By
from time import sleep from time import sleep
import time
import disnake import disnake
import random import random
from urllib.parse import urlparse from urllib.parse import urlparse
@ -60,9 +61,10 @@ def play_video(videourl):
pass pass
try: try:
endscreen = driver.find_element(By.CLASS_NAME, "html5-endscreen") endscreen = driver.find_element(By.CLASS_NAME, "html5-endscreen")
timeout = (float(os.getenv("MAX_MIN")) * 60) + time.time() - 10 #subtract 10 since 10 seconds of the video has played by this point
except Exception: except Exception:
return #same as above return #same as above
while str(endscreen.get_attribute('style')) == "display: none;": while str(endscreen.get_attribute('style')) == "display: none;" and time.time() < timeout:
pass pass
sleep(2) sleep(2)
return return
@ -155,9 +157,9 @@ async def button_listener(inter: disnake.MessageInteraction):
except IndexError: except IndexError:
break break
message = message + f"{int(page[0])-1} of {math.ceil((len(queue)-1)/10) if (len(queue)-1)/10 > 1 else 1}" message = message + f"{int(page[0])-1} of {math.ceil((len(queue)-1)/10) if (len(queue)-1)/10 > 1 else 1}"
if (int(page[0])-1) <= 1 and int(page[1] == 1): if (int(page[0])-1) <= 1 and int(int(page[1]) == 1):
await inter.response.edit_message(message) await inter.response.edit_message(message)
elif (int(page[0])-1) <= 1 and int(page[1] > 1): elif (int(page[0])-1) <= 1 and int(int(page[1]) > 1):
await inter.response.edit_message(message, components=[disnake.ui.Button(label=">>", style=disnake.ButtonStyle.primary, custom_id="Forward"),]) await inter.response.edit_message(message, components=[disnake.ui.Button(label=">>", style=disnake.ButtonStyle.primary, custom_id="Forward"),])
else: else:
await inter.response.edit_message(message, components=[disnake.ui.Button(label="<<", style=disnake.ButtonStyle.primary, custom_id="Backward"), await inter.response.edit_message(message, components=[disnake.ui.Button(label="<<", style=disnake.ButtonStyle.primary, custom_id="Backward"),