Stuffy.py

import sys
import os

import settings
import discord
import message_handler

from apscheduler.schedulers.asyncio import AsyncIOScheduler
from events.base_event              import BaseEvent
from events                         import *
from multiprocessing                import Process

# Set to remember if the bot is already running, since on_ready may be called
# more than once on reconnects
this = sys.modules[__name__]
this.running = False

# Scheduler that will be used to manage events
sched = AsyncIOScheduler()


###############################################################################

def main():
    # Initialize the client
    print("Starting up...")
    _intents = discord.Intents.default()
    _intents.message_content = True
    client = discord.Client(intents=_intents)

    # Define event handlers for the client
    # on_ready may be called multiple times in the event of a reconnect,
    # hence the running flag
    @client.event
    async def on_ready():
        if this.running:
            return

        this.running = True
        

        # Set the playing status
        if settings.NOW_PLAYING:
            print("Displaying help status", flush=True)
            await client.change_presence(
                activity=discord.Activity(type=discord.ActivityType.listening, name=settings.NOW_PLAYING))
        print("Logged in!", flush=True)
        channel = client.get_channel(818611323755036702)
        await channel.send(content = "Logged in")

        # Load all events
        print("Loading events...", flush=True)
        n_ev = 0
        for ev in BaseEvent.__subclasses__():
            event = ev()
            sched.add_job(event.run, 'interval', (client,), 
                          minutes=event.interval_minutes)
            n_ev += 1
        sched.start()
        print(f"{n_ev} events loaded", flush=True)

    # The message handler for both new message and edits
    async def common_handle_message(message):
        text = message.content
        for x in range(len(settings.COMMAND_PREFIX)):
          if text.startswith(settings.COMMAND_PREFIX[x]) and text != settings.COMMAND_PREFIX[x]:
              cmd_split = text[len(settings.COMMAND_PREFIX[x]):].split()
              try:
                  await message_handler.handle_command(cmd_split[0].lower(), 
                                        cmd_split[1:], message, client)
              except:
                  print("Error while handling message", flush=True)
                  raise

    @client.event
    async def on_message(message):
        await common_handle_message(message)

    @client.event
    async def on_message_edit(before, after):
        await common_handle_message(after)

    # Finally, set the bot running
    client.run(os.getenv('BOT_TOKEN'))

###############################################################################


if __name__ == "__main__":
    main()