From 31b046e5cc436fa223f604bae004997c1fe582f8 Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Sun, 16 Dec 2018 17:58:05 +0100 Subject: [PATCH] Use the refactored types in get_chat_member(s) --- pyrogram/client/methods/chats/get_chat_member.py | 8 +++++--- pyrogram/client/methods/chats/get_chat_members.py | 7 +++++-- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/pyrogram/client/methods/chats/get_chat_member.py b/pyrogram/client/methods/chats/get_chat_member.py index b9b23e54..b5e7cec7 100644 --- a/pyrogram/client/methods/chats/get_chat_member.py +++ b/pyrogram/client/methods/chats/get_chat_member.py @@ -17,7 +17,8 @@ # along with Pyrogram. If not, see . from pyrogram.api import functions, types, errors -from ...ext import BaseClient, utils +from ...ext import BaseClient +import pyrogram class GetChatMember(BaseClient): @@ -51,7 +52,7 @@ class GetChatMember(BaseClient): ) ) - for member in utils.parse_chat_members(full_chat).chat_members: + for member in pyrogram.ChatMembers.parse(self, full_chat).chat_members: if member.user.id == user_id.user_id: return member else: @@ -64,7 +65,8 @@ class GetChatMember(BaseClient): ) ) - return utils.parse_chat_members( + return pyrogram.ChatMembers.parse( + self, types.channels.ChannelParticipants( count=1, participants=[r.participant], diff --git a/pyrogram/client/methods/chats/get_chat_members.py b/pyrogram/client/methods/chats/get_chat_members.py index 5952a39d..df1ca8b1 100644 --- a/pyrogram/client/methods/chats/get_chat_members.py +++ b/pyrogram/client/methods/chats/get_chat_members.py @@ -18,6 +18,7 @@ from pyrogram.api import functions, types from ...ext import BaseClient, utils +import pyrogram class Filters: @@ -83,7 +84,8 @@ class GetChatMembers(BaseClient): peer = self.resolve_peer(chat_id) if isinstance(peer, types.InputPeerChat): - return utils.parse_chat_members( + return pyrogram.ChatMembers.parse( + self, self.send( functions.messages.GetFullChat( peer.chat_id @@ -108,7 +110,8 @@ class GetChatMembers(BaseClient): else: raise ValueError("Invalid filter \"{}\"".format(filter)) - return utils.parse_chat_members( + return pyrogram.ChatMembers.parse( + self, self.send( functions.channels.GetParticipants( channel=peer,