mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 13:27:47 +00:00
Code Issues Releases Wiki Activity

Better way of parsing total_count for Messages and UserProfilePhotos

Browse Source
This commit is contained in:
Dan 2018-12-17 16:14:32 +01:00
parent e6dced80cf
commit 6bc2db7157
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyrogram/client/types/messages_and_media/messages.py
View File

@ -43,10 +43,8 @@ class Messages(PyrogramType):
users = {i.id: i for i in messages.users} users = {i.id: i for i in messages.users}
chats = {i.id: i for i in messages.chats} chats = {i.id: i for i in messages.chats}
total_count = getattr(messages, "count", len(messages.messages))
return Messages( return Messages(
total_count=total_count, total_count=getattr(messages, "count", len(messages.messages)),
messages=[Message.parse(client, message, users, chats) for message in messages.messages], messages=[Message.parse(client, message, users, chats) for message in messages.messages],
client=client, client=client,
raw=messages raw=messages

3
pyrogram/client/types/messages_and_media/user_profile_photos.py
View File

@ -16,7 +16,6 @@
# You should have received a copy of the GNU Lesser General Public License # You should have received a copy of the GNU Lesser General Public License
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>. # along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
from pyrogram.api import types
from .photo import Photo from .photo import Photo
from ..pyrogram_type import PyrogramType from ..pyrogram_type import PyrogramType
@ -41,7 +40,7 @@ class UserProfilePhotos(PyrogramType):
@staticmethod @staticmethod
def parse(client, photos) -> "UserProfilePhotos": def parse(client, photos) -> "UserProfilePhotos":
return UserProfilePhotos( return UserProfilePhotos(
total_count=len(photos.photos) if isinstance(photos, types.photos.Photos) else photos.count, total_count=getattr(photos, "count", len(photos.photos)),
photos=[Photo.parse(client, photo) for photo in photos.photos], photos=[Photo.parse(client, photo) for photo in photos.photos],
client=client, client=client,
raw=photos raw=photos