mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-19 18:16:34 +01:00
Fix SS command
This commit is contained in:
parent
0b357c69e8
commit
a1a21a9042
1 changed files with 9 additions and 11 deletions
|
@ -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}")
|
||||
|
|
Loading…
Reference in a new issue