bot.py

import asyncio
import pandas as pd
import discord
import logging
from discord.ext import commands
from datetime import datetime

from configuration import Config

# Add cogs
from comandos.ping import Ping
from comandos.moderacion import Moderacion
from comandos.ayuda import Ayuda
from comandos.flood import FloodSpam
from comandos.limpia import Limpia
from comandos.archivar import Archivar
from comandos.enviar import Enviar

# Global instance of the server
guild = None

# Configuration
config = Config()

# Use '%' as command prefix
intents = discord.Intents().all()
bot = commands.Bot(command_prefix="%", intents=intents)

handler = logging.FileHandler(filename="bot.log", encoding="utf-8", mode="w")
discord.utils.setup_logging(level=logging.INFO, handler=handler)


@bot.event
async def on_message(message):
    # Main log
    with open(config.log_main_file, "a") as f:
        date_str = f"{datetime.now()}"
        line = (
            f'"{date_str}";'
            f'"{message.id}";'
            f'"{message.channel}";'
            f'"{message.author.id}";'
            f'"{message.author}";'
            f'"{message.content}"\n'
        )
        f.write(line)


@bot.event
async def on_ready():
    print("Syncing tree...")
    # await bot.tree.sync()


@bot.event
async def on_command_error(msg, error):
    if isinstance(error, (commands.MissingRole, commands.MissingAnyRole)):
        print(f"MissingRole ERROR: {error}")
    else:
        print(error)


async def main():
    # Reading data
    data_mod = pd.read_csv(str(config.log_mod_file), sep=";", dtype=str)
    data_accepted = pd.read_csv(str(config.log_accepted_file), sep=";", dtype=str)
    data_rejected = pd.read_csv(str(config.log_rejected_file), sep=";", dtype=str)

    # Pending moderation
    # Get 'message_id' from the 'accepted' and 'rejected' files
    ready_ids = set(data_accepted["message_id"]).union(data_rejected["message_id"])

    # keeping the data in the bot instance
    bot.data_mod = data_mod[~data_mod["message_id"].isin(ready_ids)]

    await bot.add_cog(Ping(bot))
    await bot.add_cog(Ayuda(bot))

    await bot.add_cog(Limpia(bot))
    await bot.add_cog(Archivar(bot))

    await bot.add_cog(Moderacion(bot))
    await bot.add_cog(FloodSpam(bot))
    await bot.add_cog(Enviar(bot))

    # Removing the help command
    # bot.remove_command("help")

    print("Running...")
    await bot.start(config.TOKEN)


asyncio.run(main())