From a9a6bfb606a6d2092a4260ac995454bf94411691 Mon Sep 17 00:00:00 2001 From: MrMissx Date: Tue, 16 Jul 2024 21:31:50 +0700 Subject: [PATCH] fix(reporting): update reporting setting logic --- anjani/plugins/reporting.py | 20 +++++++------------- 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/anjani/plugins/reporting.py b/anjani/plugins/reporting.py index a417211f5..7212e9320 100644 --- a/anjani/plugins/reporting.py +++ b/anjani/plugins/reporting.py @@ -1,4 +1,5 @@ """ Admin reporting plugin """ + # Copyright (C) 2020 - 2023 UserbotIndo Team, # # This program is free software: you can redistribute it and/or modify @@ -109,19 +110,12 @@ async def on_message(self, message: Message) -> None: async def setting(self, chat_id: int, is_private: bool, setting: bool) -> None: if is_private: - if setting: - await self.user_db.update_one( - {"_id": chat_id}, {"$set": {"setting": True}}, upsert=True - ) - else: - await self.user_db.delete_one({"_id": chat_id}) - else: - if setting: - await self.db.update_one( - {"chat_id": chat_id}, {"$set": {"setting": True}}, upsert=True - ) - else: - await self.db.delete_one({"chat_id": chat_id}) + await self.user_db.update_one( + {"_id": chat_id}, {"$set": {"setting": setting}}, upsert=True + ) + return + + await self.db.update_one({"chat_id": chat_id}, {"$set": {"setting": setting}}, upsert=True) async def is_active(self, uid: int, is_private: bool) -> bool: """Get current setting default to True"""