diff --git a/docs/source/pyrogram/Client.rst b/docs/source/pyrogram/Client.rst index e3ef2d69..5adf4956 100644 --- a/docs/source/pyrogram/Client.rst +++ b/docs/source/pyrogram/Client.rst @@ -114,6 +114,7 @@ Users get_me get_users get_user_profile_photos + get_user_profile_photos_count set_user_profile_photo delete_user_profile_photos update_username diff --git a/pyrogram/client/methods/users/__init__.py b/pyrogram/client/methods/users/__init__.py index f8c39650..d67a18bd 100644 --- a/pyrogram/client/methods/users/__init__.py +++ b/pyrogram/client/methods/users/__init__.py @@ -19,6 +19,7 @@ from .delete_user_profile_photos import DeleteUserProfilePhotos from .get_me import GetMe from .get_user_profile_photos import GetUserProfilePhotos +from .get_user_profile_photos_count import GetUserProfilePhotosCount from .get_users import GetUsers from .set_user_profile_photo import SetUserProfilePhoto from .update_username import UpdateUsername @@ -30,6 +31,7 @@ class Users( DeleteUserProfilePhotos, GetUsers, GetMe, - UpdateUsername + UpdateUsername, + GetUserProfilePhotosCount ): pass diff --git a/pyrogram/client/methods/users/get_user_profile_photos_count.py b/pyrogram/client/methods/users/get_user_profile_photos_count.py new file mode 100644 index 00000000..fdb81790 --- /dev/null +++ b/pyrogram/client/methods/users/get_user_profile_photos_count.py @@ -0,0 +1,54 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-2019 Dan Tès +# +# 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 . + +from typing import Union + +from pyrogram.api import functions, types +from ...ext import BaseClient + + +class GetUserProfilePhotosCount(BaseClient): + def get_user_profile_photos_count(self, user_id: Union[int, str]) -> int: + """Use this method to get the total count of profile pictures for a user. + + Args: + user_id (``int`` | ``str``): + Unique identifier (int) or username (str) of the target chat. + For your personal cloud (Saved Messages) you can simply use "me" or "self". + For a contact that exists in your Telegram address book you can use his phone number (str). + + Returns: + On success, an integer is returned. + + Raises: + :class:`RPCError ` in case of a Telegram RPC error. + """ + + r = self.send( + functions.photos.GetUserPhotos( + user_id=self.resolve_peer(user_id), + offset=0, + max_id=0, + limit=1 + ) + ) + + if isinstance(r, types.photos.Photos): + return len(r.photos) + else: + return r.count