From deca5879667cb29aaea64fda8050adf3a290a132 Mon Sep 17 00:00:00 2001 From: unknown Date: Sat, 3 Jun 2023 16:35:06 +0200 Subject: [PATCH] Removed hardcoded values and added config file --- recruitmentCog/config.ini | 4 ++++ recruitmentCog/recruitment.py | 16 +++++++++++++--- 2 files changed, 17 insertions(+), 3 deletions(-) create mode 100644 recruitmentCog/config.ini diff --git a/recruitmentCog/config.ini b/recruitmentCog/config.ini new file mode 100644 index 0000000..ace64eb --- /dev/null +++ b/recruitmentCog/config.ini @@ -0,0 +1,4 @@ +[DEFAULT] +identifier_id = 101101101101001110101 +guild_id = 274657393936302080 +trial_role_id = 531181363420987423 \ No newline at end of file diff --git a/recruitmentCog/recruitment.py b/recruitmentCog/recruitment.py index 3ccd617..96beaca 100644 --- a/recruitmentCog/recruitment.py +++ b/recruitmentCog/recruitment.py @@ -3,6 +3,7 @@ import datetime import re from typing import List from datetime import timedelta +import configparser import discord from redbot.core import Config, checks, commands @@ -24,9 +25,18 @@ class Recruitment(commands.Cog): def __init__(self, bot: Red): self.bot = bot self.message: str = '' - self.config = Config.get_conf(self, identifier=101101101101001110101) # Replace with your own unique identifier - default_guild = {"guild_id": 274657393936302080, "application_channel_id": None} + config = configparser.ConfigParser() + config.read('config.ini') + + identifier_id = int(config['DEFAULT']['identifier_id']) + self.config = Config.get_conf(self, identifier=identifier_id) + + self.guild_id = int(config['DEFAULT']['guild_id']) + self.trial_role_id = int(config['DEFAULT']['trial_role_id']) + + default_guild = {"guild_id": self.guild_id, "application_channel_id": None} self.config.register_guild(**default_guild) + self.antispam = {} async def cog_check(self, ctx: commands.Context): @@ -163,7 +173,7 @@ class Recruitment(commands.Cog): return # Assign the Trial role to the author - role = guild.get_role(531181363420987423) + role = guild.get_role(self.trial_role_id) try: await member.add_roles(role) except discord.Forbidden: