BonHowi/discordPogmareBot

import discord
from discord.ext import commands
from discord.utils import get
from discord_slash.model import SlashCommandPermissionType
from discord_slash.utils.manage_commands import create_permission

from modules.get_settings import get_settings

GUILD_IDS = get_settings("guild")
MODERATION_IDS = get_settings("MOD_ROLES")
PERMISSION_MODS = {
    GUILD_IDS[0]: [
        create_permission(MODERATION_IDS[0], SlashCommandPermissionType.ROLE, True),
        create_permission(MODERATION_IDS[1], SlashCommandPermissionType.ROLE, True)
    ]
}
PERMISSION_ADMINS = {
    GUILD_IDS[0]: [
        create_permission(get_settings("ADMIN"), SlashCommandPermissionType.ROLE, True)
    ]
}


# noinspection PyTypeChecker
class BaseCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.create_log_msg("Init")
        self.legend_multiplier = 5

    def get_bot(self):
        return self.bot

    # Find monster in config
    def get_monster(self, ctx, name: str):
        name = name.lower()
        monster_found = None

        for monster in self.bot.config["commands"]:
            if monster["name"].lower() == name or name in monster["triggers"]:
                monster_found = monster
                break

        if not monster_found:
            self.create_log_msg(f"Monster not found ({ctx.author}: {name})")
            return

        monster_found["role"] = discord.utils.get(ctx.guild.roles, name=monster_found["name"])
        if not monster_found["role"]:
            self.create_log_msg(f"Failed to fetch roleID for monster {monster_found['name']}")
            return

        monster_found["role"] = monster_found["role"].id
        return monster_found

    def create_log_msg(self, message: str) -> None:
        dt_string = self.bot.get_current_time()
        # That's just my pedantic way to approach string aligning as .format did not work here as if I wanted to
        log: str = f"({dt_string}) [{self.__class__.__name__}]:"
        log = f"{log}\t{message}" if len(log) >= 40 else f"{log}\t\t{message}"
        print(log)
        logs_txt_dir: str = "logs/logs.txt"
        with open(logs_txt_dir, "a+") as file_object:
            file_object.write(f"{log}\n")

    # Create role if not on server
    async def create_new_role(self, guild, role) -> None:
        if get(guild.roles, name=role):
            return
        await guild.create_role(name=role)
        self.create_log_msg(f"{role} role created")