nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +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)
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...")
time_sleep(render_time)
await asyncio.sleep(render_time)
await ctx.edit(content="Taking screenshot...")
domain = re.sub(r"https?://", "", website)
data = await asyncio.to_thread(driver.get_screenshot_as_png())
_io = io.BytesIO()
_io.write(driver.get_screenshot_as_png())
_io.write(data)
_io.seek(0)
driver.quit()
return discord.File(_io, f"{domain}.png")
@ -368,15 +369,12 @@ class OtherCog(commands.Cog):
await ctx.respond("Taking screenshot...")
try:
screenshot = await self.bot.loop.run_in_executor(
None,
self.screenshot_website(
screenshot = await self.screenshot_website(
ctx,
url,
browser,
render_timeout
)
)
except Exception as e:
console.log(f"Error taking screenshot: {e}")
return await ctx.edit(content=f"Error: {e}")