Fix history TTL Service Message Parse (hydrogram#21)

omg-xtao · ColinShark · delivrance · web-flow · commit 787690624309 · 2024-02-04T10:25:08.000+03:00
Co-authored-by: Colin &lt;colin@colinshark.de&gt;
Co-authored-by: Dan &lt;14043624+delivrance@users.noreply.github.com&gt;
Co-authored-by: Akkia &lt;AkkiaS7@outlook.com&gt;
diff --git a/pyrogram/enums/message_service_type.py b/pyrogram/enums/message_service_type.py
@@ -101,3 +101,6 @@ class MessageServiceType(AutoName):
 
     REQUESTED_CHAT = auto()
     "Requested chat"
+
+    CHAT_TTL_CHANGED = auto()
+    "Chat TTL changed"
diff --git a/pyrogram/methods/chats/set_chat_ttl.py b/pyrogram/methods/chats/set_chat_ttl.py
@@ -54,11 +54,19 @@ async def set_chat_ttl(
                 # Disable TTL for this chat
                 app.set_chat_ttl(chat_id, 0)
         """
-        await self.invoke(
+        r = await self.invoke(
             raw.functions.messages.SetHistoryTTL(
                 peer=await self.resolve_peer(chat_id),
                 period=ttl_seconds,
             )
         )
 
-        return True
+        for i in r.updates:
+            if isinstance(i, (raw.types.UpdateNewMessage,
+                              raw.types.UpdateNewChannelMessage)):
+                return await types.Message._parse(
+                    self,
+                    i.message,
+                    {i.id: i for i in r.users},
+                    {i.id: i for i in r.chats},
+                )
diff --git a/pyrogram/types/messages_and_media/message.py b/pyrogram/types/messages_and_media/message.py
@@ -466,6 +466,7 @@ def __init__(
         gift_code: "types.GiftCode" = None,
         requested_chats: List["types.Chat"] = None,
         giveaway_launched: bool = None,
+        chat_ttl_period: int = None,
         reply_markup: Union[
             "types.InlineKeyboardMarkup",
             "types.ReplyKeyboardMarkup",
@@ -567,6 +568,7 @@ def __init__(
         self.requested_chats = requested_chats
         self.giveaway_launched = giveaway_launched
         self.reactions = reactions
+        self.chat_ttl_period = chat_ttl_period
 
     @staticmethod
     async def _parse(
@@ -629,6 +631,7 @@ async def _parse(
             gift_code = None
             giveaway_launched = None
             requested_chats = None
+            chat_ttl_period = None
 
             service_type = None
 
@@ -732,6 +735,9 @@ async def _parse(
                 requested_chats = types.List(_requested_chats) or None
 
                 service_type = enums.MessageServiceType.REQUESTED_CHAT
+            elif isinstance(action, raw.types.MessageActionSetMessagesTTL):
+                chat_ttl_period = action.period
+                service_type = enums.MessageServiceType.CHAT_TTL_CHANGED
 
             from_user = types.User._parse(client, users.get(user_id, None))
             sender_chat = types.Chat._parse(client, message, users, chats, is_chat=False) if not from_user else None
@@ -768,6 +774,7 @@ async def _parse(
                 giveaway_launched=giveaway_launched,
                 gift_code=gift_code,
                 requested_chats=requested_chats,
+                chat_ttl_period=chat_ttl_period,
                 client=client
                 # TODO: supergroup_chat_created
             )
diff --git a/pyrogram/types/user_and_chats/chat.py b/pyrogram/types/user_and_chats/chat.py
@@ -661,7 +661,7 @@ async def set_photo(
             video_start_ts=video_start_ts
         )
 
-    async def set_ttl(self, ttl_seconds: int) -> bool:
+    async def set_ttl(self, ttl_seconds: int) -> "types.Message":
         """Bound method *set_ttl* of :obj:`~pyrogram.types.Chat`.
 
         Use as a shortcut for:
@@ -679,7 +679,7 @@ async def set_ttl(self, ttl_seconds: int) -> bool:
                 await chat.set_ttl(86400)
 
         Returns:
-            ``bool``: True on success.
+            :obj:`~pyrogram.types.Message`: On success, the generated service message is returned.
         """
         return await self._client.set_chat_ttl(
             chat_id=self.id,
