From 6e6dd54d4020248072053959b4120ba6d9ba4641 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sat, 8 Jun 2019 19:55:40 +0200
Subject: [PATCH] Add missing attributes to the Chat type

---
 pyrogram/client/types/user_and_chats/chat.py | 19 ++++++++++++++-----
 pyrogram/client/types/user_and_chats/user.py | 14 +++++++-------
 2 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/pyrogram/client/types/user_and_chats/chat.py b/pyrogram/client/types/user_and_chats/chat.py
index 7296a903..2d88d3ed 100644
--- a/pyrogram/client/types/user_and_chats/chat.py
+++ b/pyrogram/client/types/user_and_chats/chat.py
@@ -36,14 +36,17 @@ class Chat(Object):
             Type of chat, can be either "private", "bot", "group", "supergroup" or "channel".
 
         is_verified (``bool``, *optional*):
-            True, if this chat has been verified by Telegram. Supergroups and channels only.
+            True, if this chat has been verified by Telegram. Supergroups, channels and bots only.
 
         is_restricted (``bool``, *optional*):
-            True, if this chat has been restricted. Supergroups and channels only.
+            True, if this chat has been restricted. Supergroups, channels and bots only.
             See *restriction_reason* for details.
 
         is_scam (``bool``, *optional*):
-            True, if this chat has been flagged for scam. Supergroups and channels only.
+            True, if this chat has been flagged for scam. Supergroups, channels and bots only.
+
+        is_support (``bool``):
+            True, if this chat is part of the Telegram support team. Users and bots only.
 
         title (``str``, *optional*):
             Title, for supergroups, channels and basic group chats.
@@ -92,8 +95,8 @@ class Chat(Object):
     """
 
     __slots__ = [
-        "id", "type", "is_verified", "is_restricted", "is_scam", "title", "username", "first_name", "last_name",
-        "photo", "description", "invite_link", "pinned_message", "sticker_set_name", "can_set_sticker_set",
+        "id", "type", "is_verified", "is_restricted", "is_scam", "is_support", "title", "username", "first_name",
+        "last_name", "photo", "description", "invite_link", "pinned_message", "sticker_set_name", "can_set_sticker_set",
         "members_count", "restriction_reason", "permissions"
     ]
 
@@ -106,6 +109,7 @@ class Chat(Object):
         is_verified: bool = None,
         is_restricted: bool = None,
         is_scam: bool = None,
+        is_support: bool = None,
         title: str = None,
         username: str = None,
         first_name: str = None,
@@ -127,6 +131,7 @@ class Chat(Object):
         self.is_verified = is_verified
         self.is_restricted = is_restricted
         self.is_scam = is_scam
+        self.is_support = is_support
         self.title = title
         self.username = username
         self.first_name = first_name
@@ -148,6 +153,10 @@ class Chat(Object):
         return Chat(
             id=peer_id,
             type="bot" if user.bot else "private",
+            is_verified=getattr(user, "verified", None),
+            is_restricted=getattr(user, "restricted", None),
+            is_scam=getattr(user, "scam", None),
+            is_support=getattr(user, "support", None),
             username=user.username,
             first_name=user.first_name,
             last_name=user.last_name,
diff --git a/pyrogram/client/types/user_and_chats/user.py b/pyrogram/client/types/user_and_chats/user.py
index 05877b63..f47e8c42 100644
--- a/pyrogram/client/types/user_and_chats/user.py
+++ b/pyrogram/client/types/user_and_chats/user.py
@@ -52,12 +52,12 @@ class User(Object):
             True, if this user has been restricted. Bots only.
             See *restriction_reason* for details.
 
-        is_support (``bool``):
-            True, if this user is part of the Telegram support team.
-
         is_scam (``bool``):
             True, if this user has been flagged for scam.
 
+        is_support (``bool``):
+            True, if this user is part of the Telegram support team.
+
         first_name (``str``):
             User's or bot's first name.
 
@@ -86,7 +86,7 @@ class User(Object):
 
     __slots__ = [
         "id", "is_self", "is_contact", "is_mutual_contact", "is_deleted", "is_bot", "is_verified", "is_restricted",
-        "is_support", "is_scam", "first_name", "last_name", "status", "username", "language_code", "phone_number",
+        "is_scam", "is_support", "first_name", "last_name", "status", "username", "language_code", "phone_number",
         "photo", "restriction_reason"
     ]
 
@@ -102,8 +102,8 @@ class User(Object):
         is_bot: bool,
         is_verified: bool,
         is_restricted: bool,
-        is_support: bool,
         is_scam: bool,
+        is_support: bool,
         first_name: str,
         last_name: str = None,
         status: UserStatus = None,
@@ -123,8 +123,8 @@ class User(Object):
         self.is_bot = is_bot
         self.is_verified = is_verified
         self.is_restricted = is_restricted
-        self.is_support = is_support
         self.is_scam = is_scam
+        self.is_support = is_support
         self.first_name = first_name
         self.last_name = last_name
         self.status = status
@@ -148,8 +148,8 @@ class User(Object):
             is_bot=user.bot,
             is_verified=user.verified,
             is_restricted=user.restricted,
-            is_support=user.support,
             is_scam=user.scam,
+            is_support=user.support,
             first_name=user.first_name,
             last_name=user.last_name,
             status=UserStatus._parse(client, user.status, user.id, user.bot),