diff --git a/database/users_chats_db.py b/database/users_chats_db.py index 50eefa82..f840c65a 100644 --- a/database/users_chats_db.py +++ b/database/users_chats_db.py @@ -51,7 +51,7 @@ class UsersData: async def get_ban_status(self, id): user = await self.col.find_one({"_id": int(id)}) - return (True, user) if user else (False, None) + return user if user else False async def get_all_users(self): return self.col.find({}) diff --git a/misskaty/plugins/ban_user_or_chat.py b/misskaty/plugins/ban_user_or_chat.py index 5b9bef4e..08c352c5 100644 --- a/misskaty/plugins/ban_user_or_chat.py +++ b/misskaty/plugins/ban_user_or_chat.py @@ -121,7 +121,7 @@ async def unban_a_user(bot, message): return await message.reply(f"Error - {e}") else: jar = await db.get_ban_status(k.id) - if not jar["is_banned"]: + if not jar: return await message.reply(f"{k.mention} is not yet banned.") await db.remove_ban(k.id) await message.reply(f"Successfully unbanned user {k.mention}!!!")