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"""