build.cogs.cogbase

Complexity

Total Complexity

8

Size/Duplication

Total Lines	64
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	BaseCog.__init__()	0	3	1
B	BaseCog.get_monster()	0	32	6
A	BaseCog.get_bot()	0	2	1

import discord
from discord.ext import commands
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.USER, True)
    ]
}


# noinspection PyTypeChecker
class BaseCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        print(f"[{self.__class__.__name__}]: Init")

    def get_bot(self):
        return self.bot

    # Find monster in config
    async def get_monster(self, ctx, name: str):
        """

        :param ctx:
        :type ctx:
        :param name:
        :type name:
        :return:
        :rtype:
        """
        monster = []
        name = name.lower()

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

        if not monster:
            print(f"[{self.__class__.__name__}]: Monster not found")
            return

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

        else:
            monster["role"] = monster["role"].id
        return monster