nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-20 02:26:32 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix SS command

Browse source
This commit is contained in:
nex 2023-01-03 14:29:33 +00:00
parent 0b357c69e8
commit a1a21a9042
1 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cogs/other.py
View file

@ -67,13 +67,14 @@ class OtherCog(commands.Cog):
driver = webdriver.Firefox(service=service, options=options) driver = webdriver.Firefox(service=service, options=options)
await ctx.edit(content="Loading website...") await ctx.edit(content="Loading website...")
driver.get(website) await asyncio.to_thread(driver.get(website))
await ctx.edit(content=f"Waiting {render_time:,} seconds to render...") await ctx.edit(content=f"Waiting {render_time:,} seconds to render...")
time_sleep(render_time) await asyncio.sleep(render_time)
await ctx.edit(content="Taking screenshot...") await ctx.edit(content="Taking screenshot...")
domain = re.sub(r"https?://", "", website) domain = re.sub(r"https?://", "", website)
data = await asyncio.to_thread(driver.get_screenshot_as_png())
_io = io.BytesIO() _io = io.BytesIO()
_io.write(driver.get_screenshot_as_png()) _io.write(data)
_io.seek(0) _io.seek(0)
driver.quit() driver.quit()
return discord.File(_io, f"{domain}.png") return discord.File(_io, f"{domain}.png")
@ -368,15 +369,12 @@ class OtherCog(commands.Cog):
await ctx.respond("Taking screenshot...") await ctx.respond("Taking screenshot...")
try: try:
screenshot = await self.bot.loop.run_in_executor( screenshot = await self.screenshot_website(
None,
self.screenshot_website(
ctx, ctx,
url, url,
browser, browser,
render_timeout render_timeout
) )
)
except Exception as e: except Exception as e:
console.log(f"Error taking screenshot: {e}") console.log(f"Error taking screenshot: {e}")
return await ctx.edit(content=f"Error: {e}") return await ctx.edit(content=f"Error: {e}")