From fa4ffef8ff1e41bbac2fa83f727fc116597aa779 Mon Sep 17 00:00:00 2001 From: wulan17 Date: Thu, 11 Dec 2025 00:00:56 +0700 Subject: [PATCH] pyrofork: Refactor TodoList Signed-off-by: wulan17 --- pyrogram/types/messages_and_media/todo_list.py | 2 +- pyrogram/types/messages_and_media/todo_task.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pyrogram/types/messages_and_media/todo_list.py b/pyrogram/types/messages_and_media/todo_list.py index 5c30a4f0..9ffd54cb 100644 --- a/pyrogram/types/messages_and_media/todo_list.py +++ b/pyrogram/types/messages_and_media/todo_list.py @@ -68,7 +68,7 @@ class TodoList(Object): ) -> "TodoList": todo_list = todo.todo completions = todo.completions - entities = [types.MessageEntity._parse(client, entity, None) for entity in todo_list.title.entities] + entities = [types.MessageEntity._parse(client, entity, users) for entity in todo_list.title.entities] entities = types.List(filter(lambda x: x is not None, entities)) tasks = [ types.TodoTask._parse(client, task, users, completions) diff --git a/pyrogram/types/messages_and_media/todo_task.py b/pyrogram/types/messages_and_media/todo_task.py index 01a7c97d..cb2678ec 100644 --- a/pyrogram/types/messages_and_media/todo_task.py +++ b/pyrogram/types/messages_and_media/todo_task.py @@ -65,15 +65,15 @@ class TodoTask(Object): @staticmethod def _parse( client: "pyrogram.Client", - todo_task: "raw.types.TodoTask", + todo_task: "raw.types.TodoList", users: Dict, - completions: List["raw.types.TodoTaskCompletion"] = None + completions: List["raw.types.TodoCompletion"] = None ) -> "TodoTask": - entities = [types.MessageEntity._parse(client, entity, None) for entity in todo_task.title.entities] + entities = [types.MessageEntity._parse(client, entity, users) for entity in todo_task.title.entities] entities = types.List(filter(lambda x: x is not None, entities)) complete = {i.id: i for i in completions} if completions else {} todo_completion = complete.get(todo_task.id) - completed_by = types.User._parse(client, users.get(todo_completion.completed_by, None)) if todo_completion else None + completed_by = types.User._parse(client, users.get(getattr(todo_completion.completed_by, "user_id", None), None)) if todo_completion else None complete_date = utils.timestamp_to_datetime(todo_completion.date) if todo_completion else None return TodoTask( id=todo_task.id,