main.py

import discord
import os
import datetime

from discord.ext.commands import Bot, when_mentioned_or
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.cron import CronTrigger
from discord.ext.commands import CommandNotFound, CommandOnCooldown, MissingPermissions, MissingRequiredArgument, BadArgument, MemberNotFound

from utils import tasks
from constants import AUTO_UPDATE_TIME

intents = discord.Intents.default()
intents.members = False
client = Bot(case_insensitive=True, description="Lockout Bot", command_prefix=when_mentioned_or("."), intents=intents)

logging_channel = None


@client.event
async def on_ready():
    await client.change_presence(activity=discord.Game(name="in matches ⚔️"))
    global logging_channel
    logging_channel = await client.fetch_channel(os.environ.get("LOGGING_CHANNEL"))
    await logging_channel.send(f"Bot ready")

    scheduler = AsyncIOScheduler()
    scheduler.add_job(update, 'interval', seconds=AUTO_UPDATE_TIME)
    scheduler.add_job(tasks.create_backup, CronTrigger(hour="0, 6, 12, 18", timezone="Asia/Kolkata"), [client])
    scheduler.add_job(tasks.update_ratings, CronTrigger(minute="30", timezone="Asia/Kolkata"), [client])
    scheduler.add_job(tasks.update_problemset, CronTrigger(hour="8", timezone="Asia/Kolkata"), [client])
    scheduler.add_job(tasks.scrape_authors, CronTrigger(day_of_week="0", timezone="Asia/Kolkata"), [client])
    scheduler.start()


async def update():
    await tasks.update_matches(client)
    await tasks.update_rounds(client)


@client.event
async def on_command_error(ctx: discord.ext.commands.Context, error: Exception):
    if isinstance(error, CommandNotFound):
        pass

    elif isinstance(error, CommandOnCooldown):
        tot = error.cooldown.per
        rem = error.retry_after
        msg = f"{ctx.author.mention} That command has a default cooldown of {str(datetime.timedelta(seconds=tot)).split('.')[0]}.\n"
        msg += f"Please retry after {str(datetime.timedelta(seconds=rem)).split('.')[0]}."
        embed = discord.Embed(description=msg, color=discord.Color.red())
        embed.set_author(name=f"Slow down!")
        await ctx.send(embed=embed)

    elif isinstance(error, MemberNotFound):
        command = ctx.command
        command.reset_cooldown(ctx)
        await ctx.send(embed=discord.Embed(description=f"`{str(error)}`\nTry mentioning the user instead of typing name/id", color=discord.Color.gold()))

    elif isinstance(error, BadArgument) or isinstance(error, MissingRequiredArgument):
        command = ctx.command
        command.reset_cooldown(ctx)
        usage = f"`.{str(command)} "
        params = []
        for key, value in command.params.items():
            if key not in ['self', 'ctx']:
                params.append(f"[{key}]" if "NoneType" in str(value) else f"<{key}>")
        usage += ' '.join(params)
        usage += '`'
        if command.help:
            usage += f"\n\n{command.help}"
        await ctx.send(embed=discord.Embed(description=f"The correct usage is: {usage}", color=discord.Color.gold()))

    elif isinstance(error, MissingPermissions):
        await ctx.send(f"{str(error)}")

    else:
        desc = f"{ctx.author.name}({ctx.author.id}) {ctx.guild.name}({ctx.guild.id}) {ctx.message.content}\n"
        desc += f"**{str(error)}**"
        await logging_channel.send(desc)


if __name__ == "__main__":
    for filename in os.listdir('./cogs'):
        if filename.endswith('.py'):
            try:
                client.load_extension(f'cogs.{filename[:-3]}')
            except Exception as e:
                print(f'Failed to load file {filename}: {str(e)}')
                print(str(e))

    token = os.environ.get('LOCKOUT_BOT_TOKEN')
    client.run(token)