From e5915505a15516b9014e1beda9e3199e41d458c8 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Thu, 19 Jul 2018 23:26:20 +0200
Subject: [PATCH] Add get_chat_member method

---
 pyrogram/client/methods/chats/__init__.py     |  2 +
 .../client/methods/chats/get_chat_member.py   | 59 +++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 pyrogram/client/methods/chats/get_chat_member.py

diff --git a/pyrogram/client/methods/chats/__init__.py b/pyrogram/client/methods/chats/__init__.py
index 038440f6..ce56f16b 100644
--- a/pyrogram/client/methods/chats/__init__.py
+++ b/pyrogram/client/methods/chats/__init__.py
@@ -19,6 +19,7 @@
 from .delete_chat_photo import DeleteChatPhoto
 from .export_chat_invite_link import ExportChatInviteLink
 from .get_chat import GetChat
+from .get_chat_member import GetChatMember
 from .get_chat_members import GetChatMembers
 from .join_chat import JoinChat
 from .kick_chat_member import KickChatMember
@@ -43,6 +44,7 @@ class Chats(
     RestrictChatMember,
     PromoteChatMember,
     GetChatMembers,
+    GetChatMember,
     SetChatPhoto,
     DeleteChatPhoto,
     SetChatTitle,
diff --git a/pyrogram/client/methods/chats/get_chat_member.py b/pyrogram/client/methods/chats/get_chat_member.py
new file mode 100644
index 00000000..082d1671
--- /dev/null
+++ b/pyrogram/client/methods/chats/get_chat_member.py
@@ -0,0 +1,59 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017-2018 Dan Tès <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 pyrogram.api import functions, types, errors
+from ...ext import BaseClient, utils
+
+
+class GetChatMember(BaseClient):
+    def get_chat_member(self,
+                        chat_id: int or str,
+                        user_id: int or str):
+        # TODO: docstrings
+        chat_id = self.resolve_peer(chat_id)
+        user_id = self.resolve_peer(user_id)
+
+        if isinstance(chat_id, types.InputPeerChat):
+            full_chat = self.send(
+                functions.messages.GetFullChat(
+                    chat_id=chat_id.chat_id
+                )
+            )
+
+            for member in utils.parse_chat_members(full_chat).chat_members:
+                if member.user.id == user_id.user_id:
+                    return member
+            else:
+                raise errors.UserNotParticipant
+        elif isinstance(chat_id, types.InputPeerChannel):
+            r = self.send(
+                functions.channels.GetParticipant(
+                    channel=chat_id,
+                    user_id=user_id
+                )
+            )
+
+            return utils.parse_chat_members(
+                types.channels.ChannelParticipants(
+                    count=1,
+                    participants=[r.participant],
+                    users=r.users
+                )
+            ).chat_members[0]
+        else:
+            raise ValueError("The chat_id \"{}\" belongs to a user".format(chat_id))