Add restricted_by attribute in ChatMember (#204)

delivrance · delivrance · commit b919ed82420c · 2019-01-21T16:53:54.000+01:00
diff --git a/pyrogram/client/types/user_and_chats/chat_member.py b/pyrogram/client/types/user_and_chats/chat_member.py
@@ -43,6 +43,9 @@ class ChatMember(PyrogramType):
         promoted_by (:obj:`User <pyrogram.User>`, *optional*):
             Administrators only. Information about the user who promoted this member as administrator.
 
+        restricted_by (:obj:`User <pyrogram.User>`, *optional*):
+            Restricted and kicked only. Information about the user who restricted or kicked this member.
+
         until_date (``int``, *optional*):
             Restricted and kicked only. Date when restrictions will be lifted for this user, unix time.
 
@@ -99,6 +102,7 @@ def __init__(self,
                  date: int = None,
                  invited_by: "pyrogram.User" = None,
                  promoted_by: "pyrogram.User" = None,
+                 restricted_by: "pyrogram.User" = None,
                  until_date: int = None,
                  can_be_edited: bool = None,
                  can_change_info: bool = None,
@@ -120,6 +124,7 @@ def __init__(self,
         self.date = date
         self.invited_by = invited_by
         self.promoted_by = promoted_by
+        self.restricted_by = restricted_by
         self.until_date = until_date
         self.can_be_edited = can_be_edited
         self.can_change_info = can_change_info
@@ -181,6 +186,7 @@ def _parse(client, member, users) -> "ChatMember":
                     else "restricted"
                 ),
                 date=member.date,
+                restricted_by=pyrogram.User._parse(client, users[member.kicked_by]),
                 until_date=0 if rights.until_date == (1 << 31) - 1 else rights.until_date,
                 client=client
             )
