Main.py - Functional source of the bot, handles all the discord bot functionality

Commands.py - Allows the commands to be offloaded into another file for easy of edit/registering without overloading the main core.
This commit is contained in:
Dan 2025-01-13 21:32:31 -05:00
parent 16355e93a2
commit 0fa15014d8
2 changed files with 74 additions and 0 deletions

14
commands.py Normal file
View File

@ -0,0 +1,14 @@
import discord
# Function to register all commands to the CommandTree
def register_commands(tree: discord.app_commands.CommandTree, guild: discord.Object = None):
# Ping Command
@tree.command(name="ping", description="Check if Amber is responsive.", guild=guild)
async def ping(interaction: discord.Interaction):
await interaction.response.send_message(f"Pong! Latency: {round(interaction.client.latency * 1000)}ms")
# Example Command
@tree.command(name="hello", description="Say hello to Amber.", guild=guild)
async def hello(interaction: discord.Interaction):
await interaction.response.send_message("Hello! Amber here to help.")

60
main.py Normal file
View File

@ -0,0 +1,60 @@
import discord
import os
from dotenv import load_dotenv
from commands import register_commands # Import the command registration function
# Load environment variables from .env
load_dotenv()
# Get the bot token from the .env file
TOKEN = os.getenv("DISCORD_TOKEN")
# Define whether to sync globally or for a specific guild
DEF_PARAM = "GUILD" # Options: "GLOBAL" or "GUILD"
# Set the guild ID if using guild-specific sync
GUILD_ID = os.getenv("GUILD_ID")
class AmberClient(discord.Client):
def __init__(self):
intents = discord.Intents.default()
intents.guilds = True # Required for guild interactions
super().__init__(intents=intents)
# Command tree for slash commands
self.tree = discord.app_commands.CommandTree(self)
async def setup_hook(self):
# Register all commands from commands.py
if DEF_PARAM == "GUILD" and GUILD_ID:
guild = discord.Object(id=int(GUILD_ID))
register_commands(self.tree, guild=guild) # Register with specific guild
else:
register_commands(self.tree) # Register globally
# Sync commands after registering
try:
if DEF_PARAM == "GLOBAL":
# Sync globally
synced = await self.tree.sync()
print(f"Synced {len(synced)} commands globally!")
elif DEF_PARAM == "GUILD" and GUILD_ID:
# Sync to a specific guild
guild = discord.Object(id=int(GUILD_ID))
synced = await self.tree.sync(guild=guild)
print(f"Synced {len(synced)} commands to guild {GUILD_ID}!")
else:
print("Error: DEF_PARAM is set to 'GUILD' but GUILD_ID is not provided.")
except Exception as e:
print(f"Failed to sync commands: {e}")
# Create client instance
amber = AmberClient()
# Run the bot
if TOKEN:
amber.run(TOKEN)
else:
print("Error: DISCORD_TOKEN is not set in the .env file.")