Update set_profile_photo and add set_bot_profile_photo method

KurimuzonAkuma · KurimuzonAkuma · commit 279d6c7eafb9 · 2025-08-17T18:18:32.000+03:00
diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py
@@ -752,6 +752,10 @@ def get_title_list(s: str) -> list:
             GiftedStars
             GiftedTon
             UpgradedGiftAttributeId
+            InputChatPhoto
+            InputChatPhotoPrevious
+            InputChatPhotoStatic
+            InputChatPhotoAnimation
         """,
         bot_keyboards="""
         Bot keyboards
diff --git a/pyrogram/methods/users/set_profile_photo.py b/pyrogram/methods/users/set_profile_photo.py
@@ -16,46 +16,32 @@
 #  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, BinaryIO, Optional
+import logging
+from typing import BinaryIO, Optional, Union
 
 import pyrogram
-from pyrogram import raw
+from pyrogram import raw, types
 
+log = logging.getLogger(__name__)
 
 class SetProfilePhoto:
     async def set_profile_photo(
         self: "pyrogram.Client",
+        photo: Optional["types.InputChatPhoto"] = None,
+        is_public: Optional[bool] = None,
         *,
-        photo: Optional[Union[str, BinaryIO]] = None,
-        video: Optional[Union[str, BinaryIO]] = None,
-        is_public: Optional[bool] = None
+        video: Optional[Union[str, BinaryIO]] = None
     ) -> bool:
-        """Set a new profile photo or video (H.264/MPEG-4 AVC video, max 5 seconds).
-
-        The ``photo`` and ``video`` arguments are mutually exclusive.
-        Pass either one as named argument (see examples below).
-
-        .. note::
-
-            This method only works for Users.
-            Bots profile photos must be set using BotFather.
+        """Changes a profile photo for the current user.
 
         .. include:: /_includes/usable-by/users.rst
 
         Parameters:
-            photo (``str`` | ``BinaryIO``, *optional*):
+            photo (:obj:`~pyrogram.types.InputChatPhoto`, *optional*):
                 Profile photo to set.
-                Pass a file path as string to upload a new photo that exists on your local machine or
-                pass a binary file-like object with its attribute ".name" set for in-memory uploads.
-
-            video (``str`` | ``BinaryIO``, *optional*):
-                Profile video to set.
-                Pass a file path as string to upload a new video that exists on your local machine or
-                pass a binary file-like object with its attribute ".name" set for in-memory uploads.
 
             is_public (``bool``, *optional*):
-                If set to True, the chosen profile photo will be shown to users that can't display
-                your main profile photo due to your privacy settings.
+                Pass True to set the public photo, which will be visible even if the main photo is hidden by privacy settings.
 
         Returns:
             ``bool``: True on success.
@@ -64,21 +50,104 @@ async def set_profile_photo(
             .. code-block:: python
 
                 # Set a new profile photo
-                await app.set_profile_photo(photo="new_photo.jpg")
+                await app.set_profile_photo(photo=types.InputChatPhotoStatic("new_photo.jpg"))
 
                 # Set a new profile video
-                await app.set_profile_photo(video="new_video.mp4")
+                await app.set_profile_photo(photo=types.InputChatPhotoAnimation("new_video.mp4"))
+
+                # Set a previous profile photo
+                await app.set_profile_photo(photo=types.InputChatPhotoPrevious(file_id))
 
                 # Set/update your account's public profile photo
-                await app.set_profile_photo(photo="new_photo.jpg", is_public=True)
+                await app.set_profile_photo(photo=types.InputChatPhotoStatic("new_photo.jpg"), is_public=True)
         """
+        if video is not None:
+            log.warning(
+                "`video` is deprecated and will be removed in future updates. Use `photo` instead."
+            )
+
+            photo = types.InputChatPhotoAnimation(animation=video)
+
+        if photo is not None and not isinstance(photo, types.InputChatPhoto):
+            log.warning(
+                "You must pass `photo` as `types.InputChatPhoto`. Passing `photo` as a string "
+                "or binary object is deprecated and will be removed in future updates."
+            )
+
+            photo = types.InputChatPhotoStatic(photo=photo)
+
+        if isinstance(photo, types.InputChatPhotoPrevious):
+            return bool(
+                await self.invoke(
+                    raw.functions.photos.UpdateProfilePhoto(
+                        fallback=is_public,
+                        id=await photo.write(self),
+                    )
+                )
+            )
+        else:
+            return bool(
+                await self.invoke(
+                    raw.functions.photos.UploadProfilePhoto(
+                        fallback=is_public,
+                        file=await photo.write(self) if isinstance(photo, types.InputChatPhotoStatic) else None,
+                        video=await photo.write(self) if isinstance(photo, types.InputChatPhotoAnimation) else None,
+                        video_start_ts=getattr(photo, "main_frame_timestamp", None),
+                    )
+                )
+            )
+
+
+class SetBotProfilePhoto:
+    async def set_bot_profile_photo(
+        self: "pyrogram.Client",
+        bot_user_id: Union[int, str],
+        photo: Optional["types.InputChatPhoto"] = None,
+    ) -> bool:
+        """Changes a profile photo for a bot.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            bot_user_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target bot.
+
+            photo (:obj:`~pyrogram.types.InputChatPhoto`, *optional*):
+                Profile photo to set.
+                Pass None to remove the profile photo.
+
+        Returns:
+            ``bool``: True on success.
+
+        Example:
+            .. code-block:: python
+
+                # Set a new bot profile photo
+                await app.set_bot_profile_photo(bot_user_id="@KurigramBot", photo=types.InputChatPhotoStatic("new_photo.jpg"))
 
-        return bool(
-            await self.invoke(
-                raw.functions.photos.UploadProfilePhoto(
-                    fallback=is_public,
-                    file=await self.save_file(photo),
-                    video=await self.save_file(video)
+                # Set a new bot profile video
+                await app.set_bot_profile_photo(bot_user_id="@KurigramBot", photo=types.InputChatPhotoAnimation("new_video.mp4"))
+
+                # Remove bot profile photo
+                await app.set_bot_profile_photo(bot_user_id="@KurigramBot")
+        """
+        if isinstance(photo, types.InputChatPhotoPrevious):
+            return bool(
+                await self.invoke(
+                    raw.functions.photos.UpdateProfilePhoto(
+                        id=await photo.write(self) if photo else raw.types.InputPhotoEmpty(),
+                        bot=await self.resolve_peer(bot_user_id),
+                    )
+                )
+            )
+        else:
+            return bool(
+                await self.invoke(
+                    raw.functions.photos.UploadProfilePhoto(
+                        bot=await self.resolve_peer(bot_user_id),
+                        file=await photo.write(self) if isinstance(photo, types.InputChatPhotoStatic) else None,
+                        video=await photo.write(self) if isinstance(photo, types.InputChatPhotoAnimation) else None,
+                        video_start_ts=getattr(photo, "main_frame_timestamp", None),
+                    )
                 )
             )
-        )
diff --git a/pyrogram/types/input_content/__init__.py b/pyrogram/types/input_content/__init__.py
@@ -16,6 +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 .input_chat_photo import InputChatPhoto, InputChatPhotoPrevious, InputChatPhotoStatic, InputChatPhotoAnimation
 from .input_checklist import InputChecklist
 from .input_contact_message_content import InputContactMessageContent
 from .input_credentials import InputCredentials
@@ -54,6 +55,10 @@
 from .input_venue_message_content import InputVenueMessageContent
 
 __all__ = [
+    "InputChatPhoto",
+    "InputChatPhotoPrevious",
+    "InputChatPhotoStatic",
+    "InputChatPhotoAnimation",
     "InputChecklist",
     "InputContactMessageContent",
     "InputCredentials",
diff --git a/pyrogram/types/input_content/input_chat_photo.py b/pyrogram/types/input_content/input_chat_photo.py
@@ -0,0 +1,113 @@
+#  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 typing import BinaryIO, Optional, Union
+
+import pyrogram
+from pyrogram import raw, utils
+from pyrogram.file_id import FileType
+
+from ..object import Object
+
+
+class InputChatPhoto(Object):
+    """Describes a photo to be set as a user profile or chat photo.
+
+    It should be one of:
+
+    - :obj:`~pyrogram.types.InputChatPhotoPrevious`
+    - :obj:`~pyrogram.types.InputChatPhotoStatic`
+    - :obj:`~pyrogram.types.InputChatPhotoAnimation`
+    """
+    # TODO: - :obj:`~pyrogram.types.InputChatPhotoSticker`
+
+    def __init__(
+        self,
+    ):
+        super().__init__()
+
+    async def write(self, client: "pyrogram.Client") -> "raw.base.InputFile":
+        raise NotImplementedError
+
+
+class InputChatPhotoPrevious(InputChatPhoto):
+    """A previously used profile photo of the current user.
+
+    Parameters:
+        chat_photo_file_id (``str``):
+            Identifier of the current user's profile photo to reuse.
+    """
+    def __init__(
+        self,
+        chat_photo_file_id: int
+    ):
+        super().__init__()
+
+        self.chat_photo_file_id = chat_photo_file_id
+
+    async def write(self, client: "pyrogram.Client") -> "raw.types.InputPhoto":
+        photo = utils.get_input_media_from_file_id(self.chat_photo_file_id, FileType.PHOTO)
+        
+        return photo.id
+
+
+class InputChatPhotoStatic(InputChatPhoto):
+    """A static photo in JPEG format.
+
+    Parameters:
+        photo (``str`` | ``BinaryIO``):
+            Photo to be set as profile photo.
+    """
+    def __init__(
+        self,
+        photo: Union[str, BinaryIO]
+    ):
+        super().__init__()
+
+        self.photo = photo
+
+    async def write(self, client: "pyrogram.Client") -> "raw.base.InputFile":
+        return await client.save_file(self.photo)
+
+
+class InputChatPhotoAnimation(InputChatPhoto):
+    """An animation in H.264/MPEG-4 AVC format.
+
+    .. note::
+
+        Must be square, at most 10 seconds long, have width between 160 and 1280 and be at most 2MB in size
+
+    Parameters:
+        animation (``str`` | ``BinaryIO``):
+            Animation to be set as profile photo.
+
+        main_frame_timestamp (``float``):
+            Timestamp of the frame, which will be used as static chat photo.
+    """
+    def __init__(
+        self,
+        animation: Union[str, BinaryIO],
+        main_frame_timestamp: Optional[float] = None
+    ):
+        super().__init__()
+
+        self.animation = animation
+        self.main_frame_timestamp = main_frame_timestamp
+
+    async def write(self, client: "pyrogram.Client") -> "raw.base.InputFile":
+        return await client.save_file(self.animation)
