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.
@@ -186,7 +191,14 @@ async def progress(current, total):
 
         reply_to = 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.
@@ -119,7 +124,14 @@ async def send_cached_media(
 
         reply_to = 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.
@@ -104,7 +109,14 @@ async def send_contact(
 
         reply_to = 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.
@@ -107,7 +112,14 @@ async def send_dice(
 
         reply_to = 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.
@@ -173,7 +178,14 @@ async def progress(current, total):
 
         reply_to = 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
 
@@ -96,7 +101,14 @@ async def send_location(
 
         reply_to = 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.
@@ -111,7 +116,14 @@ async def send_media_group(
 
         reply_to = 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.
@@ -164,7 +169,14 @@ async def send_photo(
 
         reply_to = 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_poll.py

@@ -45,6 +45,7 @@ async def send_poll(
         protect_content: 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,
         reply_markup: Union[
@@ -124,6 +125,10 @@ async def send_poll(
             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.
@@ -146,7 +151,14 @@ async def send_poll(
 
         reply_to = 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
+            )
 
         solution, solution_entities = (await utils.parse_text_entities(
             self, explanation, explanation_parse_mode, explanation_entities