Pyrofork: Add support to reply message in another chat

Signed-off-by: wulan17 <[email protected]>

pyrogram/methods/messages/send_animation.py

@@ -50,6 +50,7 @@ async def send_animation(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -129,6 +130,10 @@ async def send_animation(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -191,8 +196,16 @@ async def progress(current, total):
         file = None
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_audio.py

@@ -48,6 +48,7 @@ async def send_audio(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -122,6 +123,10 @@ async def send_audio(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -185,8 +190,16 @@ async def progress(current, total):
         file = None
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_cached_media.py

@@ -39,6 +39,7 @@ async def send_cached_media(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -94,6 +95,10 @@ async def send_cached_media(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -118,8 +123,16 @@ async def send_cached_media(
         """
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_contact.py

@@ -36,6 +36,7 @@ async def send_contact(
         disable_notification: bool = None,
         message_thread_id: int = None,
         reply_to_message_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -79,6 +80,10 @@ async def send_contact(
             reply_to_message_id (``int``, *optional*):
                 If the message is a reply, ID of the original message.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -103,8 +108,16 @@ async def send_contact(
         """
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
 
         r = await self.invoke(
             raw.functions.messages.SendMedia(

pyrogram/methods/messages/send_dice.py

@@ -34,6 +34,7 @@ async def send_dice(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -75,6 +76,10 @@ async def send_dice(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -106,8 +111,16 @@ async def send_dice(
         """
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_document.py

@@ -46,6 +46,7 @@ async def send_document(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -114,6 +115,10 @@ async def send_document(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -172,8 +177,16 @@ async def progress(current, total):
         file = None
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_location.py

@@ -34,6 +34,7 @@ async def send_location(
         disable_notification: bool = None,
         message_thread_id: int = None,
         reply_to_message_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -68,6 +69,10 @@ async def send_location(
                 Unique identifier for the target message thread (topic) of the forum.
                 for forum supergroups only.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             reply_to_message_id (``int``, *optional*):
                 If the message is a reply, ID of the original message
 
@@ -95,8 +100,16 @@ async def send_location(
         """
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
 
         r = await self.invoke(
             raw.functions.messages.SendMedia(

pyrogram/methods/messages/send_media_group.py

@@ -49,6 +49,7 @@ async def send_media_group(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -80,6 +81,10 @@ async def send_media_group(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -110,8 +115,16 @@ async def send_media_group(
         multi_media = []
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_message.py

@@ -37,6 +37,7 @@ async def send_message(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -84,6 +85,10 @@ async def send_message(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -139,8 +144,16 @@ async def send_message(
         message, entities = (await utils.parse_text_entities(self, text, parse_mode, entities)).values()
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)

pyrogram/methods/messages/send_photo.py

@@ -44,6 +44,7 @@ async def send_photo(
         message_thread_id: int = None,
         reply_to_message_id: int = None,
         reply_to_story_id: int = None,
+        reply_to_chat_id: int = None,
         quote_text: str = None,
         schedule_date: datetime = None,
         protect_content: bool = None,
@@ -105,6 +106,10 @@ async def send_photo(
             reply_to_story_id (``int``, *optional*):
                 Unique identifier for the target story.
 
+            reply_to_chat_id (``int``, *optional*):
+                Unique identifier for the origin chat.
+                for reply to message from another chat.
+
             quote_text (``str``, *optional*):
                 Text to quote.
                 for reply_to_message only.
@@ -163,8 +168,16 @@ async def send_photo(
         file = None
 
         reply_to = None
+        reply_to_chat = None
         if reply_to_message_id or message_thread_id:
-            reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id, quote_text=quote_text)
+            if reply_to_chat_id is not None:
+                reply_to_chat = await self.resolve_peer(reply_to_chat_id)
+            reply_to = types.InputReplyToMessage(
+                reply_to_message_id=reply_to_message_id,
+                message_thread_id=message_thread_id,
+                reply_to_chat=reply_to_chat,
+                quote_text=quote_text
+            )
         if reply_to_story_id:
             user_id = await self.resolve_peer(chat_id)
             reply_to = types.InputReplyToStory(user_id=user_id, story_id=reply_to_story_id)