Update Checklist and get_send_as_chats for new Layer

Author: SpEcHiDe

pyrogram/methods/chats/get_send_as_chats.py

@@ -26,7 +26,8 @@ class GetSendAsChats:
     async def get_send_as_chats(
         self: "pyrogram.Client",
         chat_id: Union[int, str],
-        for_paid_reactions: Optional[bool] = None
+        for_paid_reactions: Optional[bool] = None,
+        for_live_stories: Optional[bool] = None,
     ) -> list["types.Chat"]:
         """Get the list of "send_as" chats available.
 
@@ -39,6 +40,9 @@ async def get_send_as_chats(
             for_paid_reactions (``bool``, *optional*):
                 Pass True to get the list of available send_as chats for paid reactions.
 
+            for_live_stories (``bool``, *optional*):
+                Pass True to get the list of available send_as chats for viewing live stories.
+
         Returns:
             List of :obj:`~pyrogram.types.Chat`: The list of chats.
 
@@ -52,7 +56,8 @@ async def get_send_as_chats(
         r = await self.invoke(
             raw.functions.channels.GetSendAs(
                 peer=await self.resolve_peer(chat_id),
-                for_paid_reactions=for_paid_reactions
+                for_paid_reactions=for_paid_reactions,
+                for_live_stories=for_live_stories
             )
         )
 

pyrogram/types/input_message_content/external_reply_info.py

@@ -327,7 +327,7 @@ async def _parse(
                     media_type = enums.MessageMediaType.PAID_MEDIA
                 elif isinstance(media, raw.types.MessageMediaToDo):
                     media_type = enums.MessageMediaType.CHECKLIST
-                    checklist = types.Checklist._parse(client, media, users)
+                    checklist = types.Checklist._parse(client, media, users, chats)
                 else:
                     media_type = enums.MessageMediaType.UNKNOWN
 

pyrogram/types/messages_and_media/checklist.py

@@ -69,6 +69,7 @@ def _parse(
         client: "pyrogram.Client",
         checklist: "raw.types.MessageMediaToDo",
         users: Dict[int, "raw.base.User"],
+        chats: Dict[int, "raw.base.User"],
     ) -> "Checklist":
         completions = {i.id: i for i in getattr(checklist, "completions", [])}
 
@@ -80,7 +81,8 @@ def _parse(
                     client,
                     task,
                     completions.get(task.id),
-                    users
+                    users,
+                    chats
                 )
             )
 

pyrogram/types/messages_and_media/checklist_task.py

@@ -40,7 +40,7 @@ class ChecklistTask(Object):
             Special entities that appear in the task text.
             May contain only Bold, Italic, Underline, Strikethrough, Spoiler, CustomEmoji, Url, EmailAddress, Mention, Hashtag, Cashtag and PhoneNumber entities.
 
-        completed_by_user (:obj:`~pyrogram.types.User`, *optional*):
+        completed_by_user (:obj:`~pyrogram.types.Chat`, *optional*):
             The user that completed the task.
             None if the task isn't completed.
 
@@ -56,7 +56,7 @@ def __init__(
         id: int,
         text: str,
         text_entities: Optional[list["types.MessageEntity"]] = None,
-        completed_by_user: Optional["types.User"] = None,
+        completed_by_user: Optional["types.Chat"] = None,
         completion_date: Optional[datetime] = None,
     ):
         super().__init__()
@@ -73,6 +73,7 @@ def _parse(
         item: "raw.types.TodoItem",
         completion: "raw.types.TodoCompletion",
         users: Dict[int, "raw.base.User"],
+        chats: Dict[int, "raw.base.Chat"],
     ) -> "ChecklistTask":
         text_entities = [
             types.MessageEntity._parse(client, entity, users)
@@ -81,10 +82,19 @@ def _parse(
         text_entities = types.List(filter(lambda x: x is not None, text_entities))
         text = Str(item.title.text).init(text_entities) or None
 
+        completed_by_peer = getattr(completion, "completed_by", None)
+        completed_by_chat = None
+        if completed_by_peer:
+            completed_by_peer_id = utils.get_raw_peer_id(completed_by_peer)
+            if isinstance(completed_by_peer, raw.types.PeerUser):
+                completed_by_chat = types.Chat._parse_chat(client, users.get(completed_by_peer_id))
+            else:
+                completed_by_chat = types.Chat._parse_chat(client, chats.get(completed_by_peer_id))
+
         return ChecklistTask(
             id=item.id,
             text=text,
             text_entities=text_entities,
-            completed_by_user=types.User._parse(client, users.get(getattr(completion, "completed_by", None))),
+            completed_by_user=completed_by_chat,
             completion_date=utils.timestamp_to_datetime(getattr(completion, "date", None))
         )

pyrogram/types/messages_and_media/checklist_tasks_added.py

@@ -16,6 +16,8 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+from typing import Dict
+
 import pyrogram
 from pyrogram import raw, types
 
@@ -48,10 +50,15 @@ def __init__(
         self.tasks = tasks
 
     @staticmethod
-    def _parse(client: "pyrogram.Client", message: "raw.types.MessageService") -> "ChecklistTasksAdded":
+    def _parse(
+        client: "pyrogram.Client",
+        message: "raw.types.MessageService",
+        users: Dict[int, "raw.base.User"],
+        chats: Dict[int, "raw.base.Chat"],
+    ) -> "ChecklistTasksAdded":
         action: "raw.types.MessageActionTodoAppendTasks" = message.action
 
         return ChecklistTasksAdded(
             checklist_message_id=getattr(message.reply_to, "reply_to_msg_id", None),
-            tasks=types.List([types.ChecklistTask._parse(client, task, None, {}) for task in action.list])
+            tasks=types.List([types.ChecklistTask._parse(client, task, None, users, chats) for task in action.list])
         )

pyrogram/types/messages_and_media/message.py

@@ -1097,7 +1097,7 @@ async def _parse(
 
             elif isinstance(action, raw.types.MessageActionTodoAppendTasks):
                 service_type = enums.MessageServiceType.CHECKLIST_TASKS_ADDED
-                checklist_tasks_added = types.ChecklistTasksAdded._parse(client, message)
+                checklist_tasks_added = types.ChecklistTasksAdded._parse(client, message, users, chats)
 
             parsed_message = Message(
                 id=message.id,
@@ -1363,7 +1363,7 @@ async def _parse(
                     media_type = enums.MessageMediaType.PAID_MEDIA
                 elif isinstance(media, raw.types.MessageMediaToDo):
                     media_type = enums.MessageMediaType.CHECKLIST
-                    checklist = types.Checklist._parse(client, media, users)
+                    checklist = types.Checklist._parse(client, media, users, chats)
                 else:
                     media = None
                     media_type = enums.MessageMediaType.UNKNOWN