diff --git a/database/users_chats_db.py b/database/users_chats_db.py index 2f6b0675..a636e1ef 100644 --- a/database/users_chats_db.py +++ b/database/users_chats_db.py @@ -44,10 +44,10 @@ class UsersData: return await self.col.count_documents({}) async def remove_ban(self, id): - await self.col.delete_one({"_id": id}) + return await self.col.delete_one({"_id": id}) async def ban_user(self, user_id, ban_reason="No Reason"): - await self.col.insert_one({"_id": user_id, "reason": ban_reason}) + return await self.col.insert_one({"_id": user_id, "reason": ban_reason}) async def get_ban_status(self, id): user = await self.col.find_one({"_id": int(id)}) diff --git a/misskaty/plugins/ban_user_or_chat.py b/misskaty/plugins/ban_user_or_chat.py index c0bba7f4..a9af78a1 100644 --- a/misskaty/plugins/ban_user_or_chat.py +++ b/misskaty/plugins/ban_user_or_chat.py @@ -12,10 +12,10 @@ from misskaty.vars import COMMAND_HANDLER, LOG_CHANNEL, SUDO, SUPPORT_CHAT async def ban_reply(_, ctx: Message): if not ctx.from_user: return - userban = await db.get_ban_status(ctx.from_user.id) - if userban: + isban, alesan = await db.get_ban_status(ctx.from_user.id) + if isban: await ctx.reply_msg( - f'I am sorry, You are banned to use Me. \nBan Reason: {ban["ban_reason"]}' + f'I am sorry, You are banned to use Me. \nBan Reason: {alesan["reason"]}' ) await ctx.stop_propagation()