134 lines
4.5 KiB
Python
134 lines
4.5 KiB
Python
import discord
|
|
from config import config
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
TOKEN = config['DISCORD_TOKEN']
|
|
GUILD_ID = config['GUILD_ID']
|
|
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
|
|
|
|
class Selena(discord.Client):
|
|
def __init__(self):
|
|
super().__init__(intents=intents)
|
|
self.tree = discord.app_commands.CommandTree(self)
|
|
self.xp_module = None # Initialize as None
|
|
self.load_modules()
|
|
|
|
async def setup_hook(self):
|
|
logging.info("Setting up modules...")
|
|
self.tree.copy_global_to(guild=discord.Object(id=GUILD_ID))
|
|
await self.tree.sync(guild=discord.Object(id=GUILD_ID))
|
|
logging.info("Modules setup and commands synchronized")
|
|
# Call setup_hook for xp_module here
|
|
if self.xp_module:
|
|
await self.xp_module.setup_hook()
|
|
|
|
def load_modules(self):
|
|
if config['modules']['currency']['enabled']:
|
|
from modules.user.currency import Currency
|
|
currency = Currency(self)
|
|
currency.setup(self.tree)
|
|
logging.info("Currency module loaded")
|
|
|
|
if config['modules']['xp']['enabled']:
|
|
from modules.user.xp import XP
|
|
xp = XP(self)
|
|
xp.setup(self.tree)
|
|
self.xp_module = xp # Set the xp_module attribute
|
|
logging.info("XP module loaded")
|
|
|
|
if config['modules']['birthday']['enabled']:
|
|
from modules.user.birthday import Birthday
|
|
birthday = Birthday(self)
|
|
birthday.setup(self.tree)
|
|
logging.info("Birthday module loaded")
|
|
|
|
if config['modules']['destiny2']['enabled']:
|
|
from modules.social.destiny2 import Destiny2
|
|
destiny2 = Destiny2(self)
|
|
destiny2.setup(self.tree)
|
|
logging.info("Destiny 2 module loaded")
|
|
|
|
if config['modules']['music']['enabled']:
|
|
from modules.music.music import Music
|
|
music = Music(self)
|
|
music.setup(self.tree)
|
|
logging.info("Music module loaded")
|
|
|
|
if config['modules']['youtube']['enabled']:
|
|
from modules.social.youtube import YouTube
|
|
youtube = YouTube(self)
|
|
youtube.setup(self.tree)
|
|
logging.info("YouTube module loaded")
|
|
|
|
if config['modules']['twitch']['enabled']:
|
|
from modules.social.twitch import Twitch
|
|
twitch = Twitch(self)
|
|
twitch.setup(self.tree)
|
|
self.twitch_module = twitch
|
|
logging.info("Twitch module loaded")
|
|
|
|
if config['modules']['update']['enabled']:
|
|
from modules.admin.update import Update
|
|
update = Update(self)
|
|
update.setup(self.tree)
|
|
logging.info("Update module loaded")
|
|
|
|
if config['modules']['data_privacy']['enabled']:
|
|
from modules.admin.data_privacy import DataPrivacy
|
|
data_privacy = DataPrivacy(self)
|
|
data_privacy.setup(self.tree)
|
|
logging.info("Data Privacy module loaded")
|
|
|
|
if config['modules']['terms_privacy']['enabled']:
|
|
from modules.admin.terms_privacy import TermsPrivacy
|
|
terms_privacy = TermsPrivacy(self)
|
|
terms_privacy.setup(self.tree)
|
|
logging.info("Terms and Privacy module loaded")
|
|
|
|
if config['modules']['knucklebones']['enabled']:
|
|
from modules.games.knucklebones import setup as knucklebones_setup
|
|
knucklebones_setup(self)
|
|
logging.info("Knucklebones module loaded")
|
|
|
|
if config['modules']['wordle']['enabled']:
|
|
from modules.games.wordle import setup as wordle_setup
|
|
wordle_setup(self)
|
|
logging.info("Wordle module loaded")
|
|
|
|
if config['modules']['profiles']['enabled']:
|
|
from modules.user.profiles import Profiles
|
|
profiles = Profiles(self)
|
|
profiles.setup(self.tree)
|
|
self.profiles = profiles # Properly set the profiles attribute
|
|
logging.info("Profiles module loaded")
|
|
|
|
if config['modules']['tiktok']['enabled']:
|
|
from modules.social.tiktok import TikTok
|
|
tiktok = TikTok(self)
|
|
tiktok.setup(self.tree)
|
|
logging.info("TikTok module loaded")
|
|
|
|
|
|
bot = Selena()
|
|
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
logging.info(f'{bot.user.name} has connected to Discord!')
|
|
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
logging.debug(f"Message from {message.author}: {message.content}")
|
|
if message.author == bot.user:
|
|
return
|
|
if bot.xp_module:
|
|
await bot.xp_module.handle_message(message)
|
|
|
|
bot.run(TOKEN)
|