Fix lots of key errors

This commit is contained in:
Zeva Rose 2021-07-03 17:48:38 -06:00
parent e0b9640dc1
commit 30dae009cc

View file

@ -184,7 +184,7 @@ class AdminCog(commands.Cog):
# Try to get ban information out of Discord
if re.match("^[0-9]{1,}$", user): # User ID
user = int(user)
discord_ban_info = find(lambda x: x.id == user, bans)
discord_ban_info = find(lambda x: x.user.id == user, bans)
else: # User name
if re.match("#[0-9]{4}$", user): # User name has discrim
user, discrim = user.split("#")
@ -228,7 +228,7 @@ class AdminCog(commands.Cog):
await self.discord_apply_unban(ctx, discord_ban_info.user, reason)
else:
discord_ban_info = find(
lambda x: x.id == database_ban_info["id"], bans
lambda x: x.user.id == database_ban_info["id"], bans
)
if discord_ban_info:
await self.discord_apply_unban(
@ -241,9 +241,9 @@ class AdminCog(commands.Cog):
)
self.db.jarvis.unbans.insert_one(
{
"user": discord_ban_info.id,
"username": discord_ban_info.name,
"discrim": discord_ban_info.discriminator,
"user": database_ban_info["user"],
"username": database_ban_info["username"],
"discrim": database_ban_info["discrim"],
"guild": ctx.guild.id,
"admin": ctx.author,
"reason": reason,