Add MessageImportInfo

Author: SpEcHiDe

compiler/docs/compiler.py

@@ -499,6 +499,7 @@ def get_title_list(s: str) -> list:
             Location
             Message
             MessageEntity
+            MessageImportInfo
             MessageOrigin
             MessageOriginChannel
             MessageOriginChat

pyrogram/types/message_origin/__init__.py

@@ -21,8 +21,10 @@
 from .message_origin_chat import MessageOriginChat
 from .message_origin_hidden_user import MessageOriginHiddenUser
 from .message_origin_user import MessageOriginUser
+from .message_import_info import MessageImportInfo
 
 __all__ = [
+    "MessageImportInfo",
     "MessageOrigin",
     "MessageOriginChannel",
     "MessageOriginChat",

pyrogram/types/message_origin/message_import_info.py

@@ -0,0 +1,50 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  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 datetime import datetime
+
+from .message_origin import MessageOrigin
+
+import pyrogram
+from pyrogram import types
+
+
+class MessageImportInfo(MessageOrigin):
+    """Contains information about a message created with `importMessages <https://t.me/telegram/142>`_.
+
+    Parameters:
+        date (:py:obj:`~datetime.datetime`):
+            Date the message was sent originally in Unix time
+
+        sender_user_name (``str``):
+            Name of the original sender
+
+    """
+
+    def __init__(
+        self,
+        *,
+        date: datetime = None,
+        sender_user_name: str = None
+    ):
+        super().__init__(
+            type="import_info",
+            date=date
+        )
+
+        self.sender_user_name = sender_user_name

pyrogram/types/message_origin/message_origin.py

@@ -55,6 +55,7 @@ def _parse(
         if not forward_header:
             return None
         forward_date = utils.timestamp_to_datetime(forward_header.date)
+        forward_signature = getattr(forward_header, "post_author", None)
         if forward_header.from_id:
             raw_peer_id = utils.get_raw_peer_id(forward_header.from_id)
             peer_id = utils.get_peer_id(forward_header.from_id)
@@ -67,7 +68,6 @@ def _parse(
             else:
                 forward_from_chat = types.Chat._parse_channel_chat(client, chats[raw_peer_id])
                 forward_from_message_id = forward_header.channel_post
-                forward_signature = forward_header.post_author
                 if forward_from_message_id:
                     return types.MessageOriginChannel(
                         date=forward_date,
@@ -87,3 +87,8 @@ def _parse(
                 date=forward_date,
                 sender_user_name=forward_sender_name
             )
+        elif forward_header.imported:
+            return types.MessageImportInfo(
+                date=forward_date,
+                sender_user_name=forward_signature
+            )