From ff493d27e7b4c48df1217bde1fcb6356ce48ba48 Mon Sep 17 00:00:00 2001 From: Yasir Aris M Date: Sat, 21 Oct 2023 22:49:59 +0700 Subject: [PATCH] Tes add ban user to db Signed-off-by: Yasir Aris M --- database/users_chats_db.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/database/users_chats_db.py b/database/users_chats_db.py index 35682698..c30b3f8d 100644 --- a/database/users_chats_db.py +++ b/database/users_chats_db.py @@ -44,12 +44,11 @@ class UsersData: return await self.col.count_documents({}) async def remove_ban(self, id): - ban_status = dict(is_banned=False, ban_reason="") - await self.col.update_one({"id": id}, {"$set": {"ban_status": ban_status}}) + await self.col.delete_one({"id": id}) async def ban_user(self, user_id, ban_reason="No Reason"): ban_status = dict(is_banned=True, ban_reason=ban_reason) - await self.col.update_one({"id": user_id}, {"$set": {"ban_status": ban_status}}) + await self.col.insert_one({"id": user_id}, {"$set": {"ban_status": ban_status}}) async def get_ban_status(self, id): default = dict(is_banned=False, ban_reason="")