From f47f903f7f85945c04bcb077332477d9e93b7d7f Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Mon, 16 Apr 2018 12:42:31 +0200
Subject: [PATCH] Make get_messages return the correct type

---
 pyrogram/client/client.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index f9bcbfa3..82605595 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -3332,4 +3332,21 @@ class Client:
                 id=message_ids
             )
 
-        return self.send(rpc)
+        r = self.send(rpc)
+
+        users = {i.id: i for i in r.users}
+        chats = {i.id: i for i in r.chats}
+
+        messages = []
+
+        for i in r.messages:
+            if isinstance(i, types.Message):
+                parser = message_parser.parse_message
+            elif isinstance(i, types.MessageService):
+                parser = message_parser.parse_message_service
+            else:
+                continue
+
+            messages.append(parser(self, i, users, chats))
+
+        return messages