Add for_paid_reactions in get_send_as_chats

Author: SpEcHiDe

pyrogram/methods/chats/get_send_as_chats.py

@@ -16,17 +16,17 @@
 #  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 typing import Union
+from typing import Optional, Union
 
 import pyrogram
-from pyrogram import raw
-from pyrogram import types
+from pyrogram import raw, types
 
 
 class GetSendAsChats:
     async def get_send_as_chats(
         self: "pyrogram.Client",
         chat_id: Union[int, str],
+        for_paid_reactions: Optional[bool] = None
     ) -> list["types.Chat"]:
         """Get the list of "send_as" chats available.
 
@@ -36,6 +36,9 @@ async def get_send_as_chats(
             chat_id (``int`` | ``str``):
                 Unique identifier (int) or username (str) of the target chat.
 
+            for_paid_reactions (``bool``, *optional*):
+                Pass True to get the list of available send_as chats for paid reactions.
+
         Returns:
             List of :obj:`~pyrogram.types.Chat`: The list of chats.
 
@@ -49,6 +52,7 @@ async def get_send_as_chats(
         r = await self.invoke(
             raw.functions.channels.GetSendAs(
                 peer=await self.resolve_peer(chat_id),
+                for_paid_reactions=for_paid_reactions
             )
         )
 
@@ -58,9 +62,10 @@ async def get_send_as_chats(
         send_as_chats = types.List()
 
         for p in r.peers:
-            if isinstance(p, raw.types.PeerUser):
-                send_as_chats.append(types.Chat._parse_chat(self, users[p.user_id]))
+            # TODO
+            if isinstance(p.peer, raw.types.PeerUser):
+                send_as_chats.append(types.Chat._parse_chat(self, users[p.peer.user_id]))
             else:
-                send_as_chats.append(types.Chat._parse_chat(self, chats[p.channel_id]))
+                send_as_chats.append(types.Chat._parse_chat(self, chats[p.peer.channel_id]))
 
         return send_as_chats