upstreams

userge/config.py

@@ -34,6 +34,7 @@ class Config:
     HU_STRING_SESSION = os.environ.get("HU_STRING_SESSION", None)
     OWNER_ID = tuple(filter(lambda x: x, map(int, os.environ.get("OWNER_ID", "0").split())))
     LOG_CHANNEL_ID = int(os.environ.get("LOG_CHANNEL_ID"))
+    AUTH_CHATS = OWNER_ID + (LOG_CHANNEL_ID,)
     DB_URI = os.environ.get("DATABASE_URL")
     LANG = os.environ.get("PREFERRED_LANGUAGE")
     DOWN_PATH = os.environ.get("DOWN_PATH")

userge/core/methods/messages/edit_message_text.py

@@ -83,7 +83,7 @@ async def edit_message_text(self,  # pylint: disable=arguments-differ
         Raises:
             RPCError: In case of a Telegram RPC error.
         """
-        if text and chat_id != Config.LOG_CHANNEL_ID:
+        if text and chat_id not in Config.AUTH_CHATS:
             text = secure_text(str(text))
         msg = await super().edit_message_text(chat_id=chat_id,
                                               message_id=message_id,

userge/core/methods/messages/send_as_file.py

@@ -70,7 +70,7 @@ async def send_as_file(self,
         Returns:
             On success, the sent Message is returned.
         """
-        if text and chat_id != Config.LOG_CHANNEL_ID:
+        if text and chat_id not in Config.AUTH_CHATS:
             text = secure_text(str(text))
         async with aiofiles.open(filename, "w+", encoding="utf8") as out_file:
             await out_file.write(text)

userge/core/methods/messages/send_message.py

@@ -92,7 +92,7 @@ async def send_message(self,  # pylint: disable=arguments-differ
         Returns:
             :obj:`Message`: On success, the sent text message or True is returned.
         """
-        if text and chat_id != Config.LOG_CHANNEL_ID:
+        if text and chat_id not in Config.AUTH_CHATS:
             text = secure_text(str(text))
         msg = await super().send_message(chat_id=chat_id,
                                          text=text,