mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Improve get_profile_photos and get_profile_photos_count

Browse Source
This commit is contained in:
Dan 2019-06-12 11:37:43 +02:00
parent a21858a262
commit 4f2928e7b5
2 changed files with 40 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pyrogram/client/methods/users/get_profile_photos.py
View File

@ -21,6 +21,7 @@ from typing import Union, List
import pyrogram
from pyrogram.api import functions, types
from pyrogram.client.ext import utils
from ...ext import BaseClient
@ -55,18 +56,7 @@ class GetProfilePhotos(BaseClient):
"""
peer_id = self.resolve_peer(chat_id)
if isinstance(peer_id, (types.InputPeerUser, types.InputPeerSelf)):
r = self.send(
functions.photos.GetUserPhotos(
user_id=peer_id,
offset=offset,
max_id=0,
limit=limit
)
)
return pyrogram.List(pyrogram.Photo._parse(self, photo) for photo in r.photos)
else:
if isinstance(peer_id, types.InputPeerChannel):
r = utils.parse_messages(
self,
self.send(
@ -87,3 +77,14 @@ class GetProfilePhotos(BaseClient):
)
return pyrogram.List([message.new_chat_photo for message in r][:limit])
else:
r = self.send(
functions.photos.GetUserPhotos(
user_id=peer_id,
offset=offset,
max_id=0,
limit=limit
)
)
return pyrogram.List(pyrogram.Photo._parse(self, photo) for photo in r.photos)

28
pyrogram/client/methods/users/get_profile_photos_count.py
View File

@ -18,6 +18,8 @@
from typing import Union
from pyrogram.api import functions, types
from ...ext import BaseClient
@ -38,4 +40,28 @@ class GetProfilePhotosCount(BaseClient):
RPCError: In case of a Telegram RPC error.
"""
return self.get_profile_photos(chat_id, limit=1).total_count
peer_id = self.resolve_peer(chat_id)
if isinstance(peer_id, types.InputPeerChannel):
r = self.send(
functions.messages.GetSearchCounters(
peer=peer_id,
filters=[types.InputMessagesFilterChatPhotos()],
)
)
return r[0].count
else:
r = self.send(
functions.photos.GetUserPhotos(
user_id=peer_id,
offset=0,
max_id=0,
limit=1
)
)
if isinstance(r, types.photos.Photos):
return len(r.photos)
else:
return r.count