58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from dotenv import load_dotenv
|
|
import discord
|
|
import os
|
|
from modules.base import BaseModule
|
|
from modules import FourasModule, RhymesModule
|
|
|
|
MODULES = FourasModule, RhymesModule
|
|
|
|
intents = discord.Intents.default()
|
|
intents.members = True
|
|
intents.presences = True
|
|
intents.guilds = True
|
|
intents.messages = True
|
|
intents.message_content = True
|
|
client = discord.Client(intents=intents)
|
|
load_dotenv()
|
|
token = os.getenv("DISCORD_TOKEN", "NO_TOKEN")
|
|
|
|
client.riddles = []
|
|
client.answers = []
|
|
client.rhyme_keys = {}
|
|
client.rhyme_strings = {}
|
|
client.cooldown = {}
|
|
client.ongoing_riddles = {}
|
|
client.modules: list[BaseModule] = []
|
|
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
client.modules = [m(client) for m in MODULES]
|
|
for m in client.modules:
|
|
await m.load()
|
|
print(f"Logged in as {client.user}")
|
|
|
|
|
|
@client.event
|
|
async def on_message(message):
|
|
# don't answer to self
|
|
if (
|
|
message.author == client.user and message.channel in client.ongoing_riddles
|
|
): # need to move a part of that block in FourasModule
|
|
current_riddle = client.ongoing_riddles[message.channel]
|
|
if "message" not in current_riddle:
|
|
current_riddle["message"] = message
|
|
return
|
|
|
|
if isinstance(
|
|
message.channel, (discord.DMChannel, discord.TextChannel, discord.Thread)
|
|
):
|
|
handled = False
|
|
for m in client.modules:
|
|
if not handled:
|
|
handled = await m.handle_message(message)
|
|
|
|
|
|
# Initialise le client
|
|
client.run(token)
|