Removed hardcoded values and added config file

This commit is contained in:
unknown 2023-06-03 16:35:06 +02:00
parent 8d5570fe43
commit deca587966
2 changed files with 17 additions and 3 deletions

View File

@ -0,0 +1,4 @@
[DEFAULT]
identifier_id = 101101101101001110101
guild_id = 274657393936302080
trial_role_id = 531181363420987423

View File

@ -3,6 +3,7 @@ import datetime
import re import re
from typing import List from typing import List
from datetime import timedelta from datetime import timedelta
import configparser
import discord import discord
from redbot.core import Config, checks, commands from redbot.core import Config, checks, commands
@ -24,9 +25,18 @@ class Recruitment(commands.Cog):
def __init__(self, bot: Red): def __init__(self, bot: Red):
self.bot = bot self.bot = bot
self.message: str = '' self.message: str = ''
self.config = Config.get_conf(self, identifier=101101101101001110101) # Replace with your own unique identifier config = configparser.ConfigParser()
default_guild = {"guild_id": 274657393936302080, "application_channel_id": None} 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.config.register_guild(**default_guild)
self.antispam = {} self.antispam = {}
async def cog_check(self, ctx: commands.Context): async def cog_check(self, ctx: commands.Context):
@ -163,7 +173,7 @@ class Recruitment(commands.Cog):
return return
# Assign the Trial role to the author # Assign the Trial role to the author
role = guild.get_role(531181363420987423) role = guild.get_role(self.trial_role_id)
try: try:
await member.add_roles(role) await member.add_roles(role)
except discord.Forbidden: except discord.Forbidden: