From 9a9a74f19f117b86396c8a9098da303afc32e31b Mon Sep 17 00:00:00 2001
From: phixxy <bottlecap@protonmail.com>
Date: Wed, 17 Apr 2024 11:56:31 -0700
Subject: [PATCH] maybe fixed ssh connection

---
 cogs/wake_on_lan.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/cogs/wake_on_lan.py b/cogs/wake_on_lan.py
index 8eb5645..0e16b44 100644
--- a/cogs/wake_on_lan.py
+++ b/cogs/wake_on_lan.py
@@ -37,18 +37,19 @@ class WakeOnLan(commands.Cog):
     async def sleep(self, ctx, amount=5):
         if ctx.author.id == self.admin_id:
             #use ssh to login and shutdown
-            ssh_client = asyncssh.connect(
+            async with asyncssh.connect(
                 self.stable_diffusion_ip,
                 username=self.stable_diffusion_login,
                 password=self.stable_diffusion_password,
                 timeout=10,
-            )
-            try:
-                await ssh_client.run("shutdown /s")
-                #await ssh_client.run("sudo shutdown -h now")
-            except:
-                self.logger.exception("WakeOnLan: Sleeping failed")
-                await ctx.send("Sleeping failed")
+            ) as ssh_client:
+                try:
+                    result = await ssh_client.run("shutdown /s")
+                    await ctx.send(result.stdout)
+                    #await ssh_client.run("sudo shutdown -h now")
+                except:
+                    self.logger.exception("WakeOnLan: Sleeping failed")
+                    await ctx.send("Sleeping failed")
 
 
 async def setup(bot):