mirror of
https://github.com/neoarz/Syntrel.git
synced 2025-12-25 11:40:12 +01:00
f28ae00d7d
Added graceful shutdown support via signals in bot.py, improved shutdown command to use SIGTERM, and enhanced logging for shutdown events. Updated UDID command description for clarity.
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
import os
|
|
import signal
|
|
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import Context
|
|
|
|
|
|
class Shutdown(commands.Cog, name="shutdown"):
|
|
def __init__(self, bot) -> None:
|
|
self.bot = bot
|
|
|
|
async def send_embed(self, context: Context, embed: discord.Embed, *, ephemeral: bool = False) -> None:
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if interaction.response.is_done():
|
|
await interaction.followup.send(embed=embed, ephemeral=ephemeral)
|
|
else:
|
|
await interaction.response.send_message(embed=embed, ephemeral=ephemeral)
|
|
else:
|
|
await context.send(embed=embed)
|
|
|
|
@commands.hybrid_command(
|
|
name="shutdown",
|
|
description="Make the bot shutdown.",
|
|
)
|
|
@commands.is_owner()
|
|
async def shutdown(self, context: Context) -> None:
|
|
"""
|
|
Shuts down the bot.
|
|
|
|
:param context: The hybrid command context.
|
|
"""
|
|
# Log command execution before shutdown starts
|
|
if context.guild is not None:
|
|
self.bot.logger.info(
|
|
f"Executed shutdown command in {context.guild.name} (ID: {context.guild.id}) by {context.author} (ID: {context.author.id})"
|
|
)
|
|
else:
|
|
self.bot.logger.info(
|
|
f"Executed shutdown command by {context.author} (ID: {context.author.id}) in DMs"
|
|
)
|
|
|
|
embed = discord.Embed(
|
|
title="Shutdown",
|
|
description="Shutting down. Bye! <a:PandaThanos:1417483671253811262>",
|
|
color=0x7289DA,
|
|
).set_author(name="Owner", icon_url="https://yes.nighty.works/raw/zReOib.webp")
|
|
|
|
await self.send_embed(context, embed)
|
|
os.kill(os.getpid(), signal.SIGTERM)
|
|
|
|
async def cog_command_error(self, context: Context, error) -> None:
|
|
if isinstance(error, commands.NotOwner):
|
|
embed = discord.Embed(
|
|
title="Permission Denied",
|
|
description="You are not the owner of this bot!",
|
|
color=0xE02B2B,
|
|
)
|
|
embed.set_author(name="Owner", icon_url="https://yes.nighty.works/raw/zReOib.webp")
|
|
await self.send_embed(context, embed, ephemeral=True)
|
|
else:
|
|
raise error
|
|
|
|
|
|
async def setup(bot) -> None:
|
|
await bot.add_cog(Shutdown(bot))
|