feat: allow to set emoji status for channel

Author: KurimuzonAkuma

pyrogram/methods/users/set_emoji_status.py

@@ -16,7 +16,7 @@
 #  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 Optional
+from typing import Optional, Union
 
 import pyrogram
 from pyrogram import raw, types
@@ -25,15 +25,25 @@
 class SetEmojiStatus:
     async def set_emoji_status(
         self: "pyrogram.Client",
+        chat_id: Optional[Union[int, str]] = None,
         emoji_status: Optional["types.EmojiStatus"] = None
     ) -> bool:
         """Set the emoji status.
 
+        .. note::
+
+            For Telegram Premium users only.
+
         .. include:: /_includes/usable-by/users.rst
 
         Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+                Defaults to the current user.
+
             emoji_status (:obj:`~pyrogram.types.EmojiStatus`, *optional*):
-                The emoji status to set. None to remove.
+                New emoji status.
+                Pass None to remove.
 
         Returns:
             ``bool``: On success, True is returned.
@@ -43,20 +53,40 @@ async def set_emoji_status(
 
                 from pyrogram import types
 
-                # Set emoji status
-                await app.set_emoji_status(types.EmojiStatus(custom_emoji_id="1234567890987654321"))
+                # Set emoji status of the current user
+                await app.set_emoji_status(emoji_status=types.EmojiStatus(custom_emoji_id="1234567890987654321"))
 
-                # Set collectible emoji status
-                await app.set_emoji_status(types.EmojiStatus(collectible_id=1234567890987654321))
+                # Set collectible emoji status for a channel
+                await app.set_emoji_status(
+                    chat_id="channel_username",
+                    emoji_status=types.EmojiStatus(gift_id=1234567890987654321)
+                )
         """
-        await self.invoke(
-            raw.functions.account.UpdateEmojiStatus(
-                emoji_status=(
-                    emoji_status.write()
-                    if emoji_status
-                    else raw.types.EmojiStatusEmpty()
+        if chat_id is None:
+            peer = raw.types.InputPeerSelf()
+        else:
+            peer = await self.resolve_peer(chat_id)
+
+        if isinstance(peer, raw.types.InputPeerChannel):
+            await self.invoke(
+                raw.functions.channels.UpdateEmojiStatus(
+                    channel=peer,
+                    emoji_status=(
+                        emoji_status.write()
+                        if emoji_status
+                        else raw.types.EmojiStatusEmpty()
+                    )
+                )
+            )
+        else:
+            await self.invoke(
+                raw.functions.account.UpdateEmojiStatus(
+                    emoji_status=(
+                        emoji_status.write()
+                        if emoji_status
+                        else raw.types.EmojiStatusEmpty()
+                    )
                 )
             )
-        )
 
         return True