better steamid handling

This commit is contained in:
Skillet 2023-04-17 22:36:32 -04:00
parent 9205638625
commit b095eacade

View File

@ -25,14 +25,17 @@ class Whitelist(commands.Cog):
return return
updatecur = self.client.squadjs.cursor(buffered=True) updatecur = self.client.squadjs.cursor(buffered=True)
try: try:
updatecur.execute(self.client.squadjs_updateDiscordID, (interaction.user.id, steam64)) if updatecur.execute(self.client.squadjs_updateDiscordID, (interaction.user.id, steam64)) <= 0:
await interaction.response.send_message("Cound not find SteamID!")
self.client.squadjs.commit()
return
except mysql.connector.Error as err: except mysql.connector.Error as err:
log.error("MYSQL error!") log.error("MYSQL error!")
await interaction.response.send_message("Could not find steamID!") await interaction.response.send_message("There was an internal server error, pls contact skillet")
return return
for urole in interaction.user.roles: for urole in interaction.user.roles:
if urole.id in self.client.whitelistGrps.keys(): if urole.id in self.client.whitelistGrps.keys():
disusername = interaction.user.nick if interaction.user.nick is not None else interaction.user.name 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.whitelistGrps[urole.id].addMember(config.WhitelistMember(interaction.user.id, disusername, steam64))
self.client.squadjs.commit() self.client.squadjs.commit()
await interaction.response.send_message("SteamID is linked, roles updated.") await interaction.response.send_message(f"discord is linked to steamID {steam64}, roles updated.")