mystiatech/Nessa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added a shutdown command

Browse Source
This commit is contained in:
Dan
2024-05-04 20:44:45 -04:00
parent e1151f8423
commit fe52fec2f8
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
dolly/dolly.py
View File

@ -6,6 +6,7 @@ import os
load_dotenv() load_dotenv()
GUILD_ID = int(os.getenv("DISCORD_GUILD_ID")) GUILD_ID = int(os.getenv("DISCORD_GUILD_ID"))
OWNER_ID = int(os.getenv("AUTHORIZED_USER_ID")) # Ensure your .env has your Discord user ID
class Dolly(discord.Client): class Dolly(discord.Client):
@ -16,7 +17,19 @@ class Dolly(discord.Client):
async def setup_hook(self): async def setup_hook(self):
self.tree.add_command(DollyTracker()) self.tree.add_command(DollyTracker())
self.tree.copy_global_to(guild = discord.Object(id=GUILD_ID)) self.tree.copy_global_to(guild = discord.Object(id=GUILD_ID))
await self.tree.sync(guild = discord.Object(id=GUILD_ID)) await self.tree.sync(guild=discord.Object(id=GUILD_ID))
# Adding a guild-specific shutdown command
@self.tree.command(name="shutdown", description="Shut down the bot", guild_ids=[GUILD_ID])
async def shutdown(interaction: discord.Interaction):
if interaction.user.id == OWNER_ID:
await interaction.response.send_message("Shutting down...")
await self.close()
else:
await interaction.response.send_message("You do not have permission to use this command.", ephemeral=True)
async def on_ready(self): async def on_ready(self):
print(f"Logged on as {self.user}!") print(f"Logged on as {self.user}!")
# Ensuring commands are only available in specified guild
self.tree.copy_global_to(guild=discord.Object(id=GUILD_ID))
await self.tree.sync(guild=discord.Object(id=GUILD_ID))