phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

more variables for SD to allow shutting down the host

Browse source
This commit is contained in:
phixxy 2024-04-17 11:11:31 -07:00
parent b1645bc968
commit 55c81f47fb
2 changed files with 29 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
cogs/stable_diffusion.py
View file

@ -15,7 +15,9 @@ class StableDiffusion(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.stable_diffusion_url = os.getenv("stablediffusion_url") # Change this to stable_diffusion_url
self.stable_diffusion_ip = os.getenv("stable_diffusion_ip")
self.stable_diffusion_port = os.getenv("stable_diffusion_port")
self.stable_diffusion_url = f"http://{self.stable_diffusion_ip}:{self.stable_diffusion_port}"
self.working_dir = "tmp/stable_diffusion/"
self.data_dir = "data/stable_diffusion/"
self.default_neg_prompt = "easynegative, badhandv4, verybadimagenegative_v1.3"

30
cogs/wake_on_lan.py
View file

@ -1,5 +1,6 @@
import logging
import os
import asyncssh
import discord
from discord.ext import commands
import wakeonlan
@ -12,15 +13,14 @@ class WakeOnLan(commands.Cog):
self.admin_id = 242018983241318410
self.logger = logging.getLogger("bot")
self.mac_address = "9C-6B-00-38-08-8D"
self.stable_diffusion_ip = os.getenv("stable_diffusion_ip")
self.stable_diffusion_login = os.getenv("stable_diffusion_login")
self.stable_diffusion_password = os.getenv("stable_diffusion_password")
def wake_on_lan(self, mac_address):
self.logger.info(f"WakeOnLan: Waking up {mac_address}")
wakeonlan.send_magic_packet(mac_address)
def sleep_on_lan(self, mac_address):
self.logger.info(f"WakeOnLan: Sleeping {mac_address}")
wakeonlan.send_magic_packet(mac_address, magic_packet="000000000000")
@commands.command()
async def wake(self, ctx):
if ctx.author.id != self.admin_id:
@ -28,6 +28,28 @@ class WakeOnLan(commands.Cog):
self.wake_on_lan(self.mac_address)
await ctx.send(f"Waking up !imagine server")
@commands.command(
description="shutdown imagine server",
help="shutdown imagine server",
brief="shutdown imagine server",
hidden=True
)
async def sleep(self, ctx, amount=5):
if ctx.author.id == self.admin_id:
#use ssh to login and shutdown
ssh_client = 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")
async def setup(bot):
await bot.add_cog(WakeOnLan(bot))