Passed
Push — main ( f7b31c...1a3ff0 )
by Bartosz
02:52 queued 01:23
created

build.cogs.cogbase.BaseCog.get_bot()   A

Complexity

Conditions 1

Size

Total Lines 2
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
eloc 2
dl 0
loc 2
rs 10
c 0
b 0
f 0
cc 1
nop 1
1
import discord
2
from discord.ext import commands
3
from discord_slash.model import SlashCommandPermissionType
4
from discord_slash.utils.manage_commands import create_permission
5
from modules.get_settings import get_settings
6
7
GUILD_IDS = get_settings("guild")
8
MODERATION_IDS = get_settings("MOD_ROLES")
9
PERMISSION_MODS = {
10
    GUILD_IDS[0]: [
11
        create_permission(MODERATION_IDS[0], SlashCommandPermissionType.ROLE, True),
12
        create_permission(MODERATION_IDS[1], SlashCommandPermissionType.ROLE, True)
13
    ]
14
}
15
PERMISSION_ADMINS = {
16
    GUILD_IDS[0]: [
17
        create_permission(get_settings("ADMIN"), SlashCommandPermissionType.USER, True)
18
    ]
19
}
20
21
22
# noinspection PyTypeChecker
23
class BaseCog(commands.Cog):
24
    def __init__(self, bot):
25
        self.bot = bot
26
        print(f"[{self.__class__.__name__}]: Init")
27
28
    def get_bot(self):
29
        return self.bot
30
31
    # Find monster in config
32
    async def get_monster(self, ctx, name: str):
33
        """
34
35
        :param ctx:
36
        :type ctx:
37
        :param name:
38
        :type name:
39
        :return:
40
        :rtype:
41
        """
42
        monster = []
43
        name = name.lower()
44
45
        for monsters in self.bot.config["commands"]:
46
            if monsters["name"].lower() == name:
47
                monster = monsters
48
                break
49
            if name in monsters["triggers"]:
50
                monster = monsters
51
52
        if not monster:
53
            print(f"[{self.__class__.__name__}]: Monster not found")
54
            return
55
56
        monster["role"] = discord.utils.get(ctx.guild.roles, name=monster["name"])
57
        if not monster["role"]:
58
            print(f"[{self.__class__.__name__}]: Failed to fetch roleID for monster {monster['name']}")
59
            return
60
61
        else:
62
            monster["role"] = monster["role"].id
63
        return monster
64