DiscordWhitelist/autowl/Cogs/Whitelist.py

import discord
import logging

import mysql.connector

from autowl import config
from autowl.bot import Bot
from discord.ext import commands
from discord import app_commands


log = logging.getLogger(__name__)


class Whitelist(commands.Cog):
    def __init__(self, client: Bot):
        self.client = client

    @app_commands.command()
    async def link(self, interaction: discord.Interaction, steam64: str):
        self.client.squadjs.connect()
        if not interaction.guild:
            await interaction.response.send_message(
                "This command must be ran within a discord server!"
            )
            return
        updatecur = self.client.squadjs.cursor(buffered=True)
        try:
            updatecur.execute(self.client.squadjs_updateDiscordID, (interaction.user.id, steam64))
            rowsaffected = updatecur.rowcount
            if rowsaffected <= 0:
                updatecur.execute(self.client.squadjs_findByDiscordID, [interaction.user.id])
                if updatecur.rowcount <= 0:
                    await interaction.response.send_message("Cound not find SteamID!")
                else:
                    for urole in interaction.user.roles:
                        if urole.id in self.client.whitelistGrps.keys():
                            self.client.whitelistGrps[urole.id].members[f"{interaction.user.id}"].steam64 = steam64
                            self.client.whitelistGrps[urole.id].updateGroup()
                    await interaction.response.send_message("SteamID already linked, roles updated.")
                self.client.squadjs.commit()
                return
        except mysql.connector.Error as err:
            log.error("MYSQL error!")
            await interaction.response.send_message("There was an internal server error, pls contact skillet")
            return
        for urole in interaction.user.roles:
            if urole.id in self.client.whitelistGrps.keys():
                disusername = interaction.user.nick if interaction.user.nick is not None else interaction.user.name
                self.client.whitelistGrps[urole.id].addMember(config.WhitelistMember(interaction.user.id, disusername, steam64))
        self.client.squadjs.commit()
        self.client.squadjs.close()
        await interaction.response.send_message(f"discord is linked to steamID, roles updated.")