[WarnSystem] Use fetch_ban endpoint (#198)

* [warnsystem] use fetch_ban endpoint * fix AttributeError
laggron42 · Jun 17, 2022 · b98e6b7 · b98e6b7
1 parent bc6bc2d
commit b98e6b7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/warnsystem/warnsystem.py b/warnsystem/warnsystem.py
@@ -1290,18 +1290,21 @@ async def wsunmute(self, ctx: commands.Context, member: discord.Member):
             await ctx.send(page)
 
     @commands.command()
+    @commands.bot_has_permissions(ban_members=True)
     @checks.mod_or_permissions(ban_members=True)
-    async def wsunban(self, ctx: commands.Context, member: UnavailableMember):
+    async def wsunban(self, ctx: commands.Context, member_id: int):
         """
         Unban a member banned with WarnSystem.
 
         *wsunban = WarnSystem unban. Feel free to add an alias.*
         """
         guild = ctx.guild
-        bans = await guild.bans()
-        if member.id not in [x.user.id for x in bans]:
+        try:
+            ban_entry = await guild.fetch_ban(discord.Object(member_id))
+        except discord.NotFound:
             await ctx.send(_("That user is not banned."))
             return
+        member = ban_entry.user
         try:
             await guild.unban(member)
         except discord.errors.HTTPException as e: