mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-20 02:26:32 +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)
|
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}")
|
||||||
|
|
Loading…
Reference in a new issue