TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor Messages

Browse Source
This commit is contained in:
Dan 2018-12-17 14:18:15 +01:00
parent 5bc9e0fc75
commit 7b90a0e7c5
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyrogram/client/types/messages_and_media/messages.py
View File

@ -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 pyrogram.api import types
from .message import Message
from ..pyrogram_type import PyrogramType
@ -35,3 +37,17 @@ class Messages(PyrogramType):
self.total_count = total_count
self.messages = messages
@staticmethod
def parse(client, messages: types.messages.Messages) -> "Messages":
users = {i.id: i for i in messages.users}
chats = {i.id: i for i in messages.chats}
total_count = getattr(messages, "count", len(messages.messages))
return Messages(
total_count=total_count,
messages=[Message.parse(client, message, users, chats) for message in messages.messages],
client=client,
raw=messages
)