Add .archive() and .unarchive() bound methods to Chat

delivrance · delivrance · commit 6e3d8ca20b25 · 2019-06-08T19:28:47.000+02:00
diff --git a/docs/source/api/bound-methods.rst b/docs/source/api/bound-methods.rst
@@ -59,6 +59,15 @@ Message
     - :meth:`~Message.reply_video_note`
     - :meth:`~Message.reply_voice`
 
+Chat
+^^^^
+
+.. hlist::
+    :columns: 2
+
+    - :meth:`~Chat.archive`
+    - :meth:`~Chat.unarchive`
+
 CallbackQuery
 ^^^^^^^^^^^^^
 
@@ -109,6 +118,10 @@ Details
 .. automethod:: Message.reply_video_note()
 .. automethod:: Message.reply_voice()
 
+.. Chat
+.. automethod:: Chat.archive()
+.. automethod:: Chat.unarchive()
+
 .. CallbackQuery
 .. automethod:: CallbackQuery.answer()
 
diff --git a/pyrogram/client/types/user_and_chats/chat.py b/pyrogram/client/types/user_and_chats/chat.py
@@ -257,3 +257,49 @@ def _parse_chat(client, chat: Union[types.Chat, types.User, types.Channel]) -> "
             return Chat._parse_user_chat(client, chat)
         else:
             return Chat._parse_channel_chat(client, chat)
+
+    def archive(self):
+        """Bound method *archive* of :obj:`Chat`.
+
+        Use as a shortcut for:
+
+        .. code-block:: python
+
+            client.archive_chats(-100123456789)
+
+        Example:
+            .. code-block:: python
+
+                chat.archive()
+
+        Returns:
+            True on success.
+
+        Raises:
+            RPCError: In case of a Telegram RPC error.
+        """
+
+        return self._client.archive_chats(self.id)
+
+    def unarchive(self):
+        """Bound method *unarchive* of :obj:`Chat`.
+
+        Use as a shortcut for:
+
+        .. code-block:: python
+
+            client.unarchive_chats(-100123456789)
+
+        Example:
+            .. code-block:: python
+
+                chat.unarchive()
+
+        Returns:
+            True on success.
+
+        Raises:
+            RPCError: In case of a Telegram RPC error.
+        """
+
+        return self._client.unarchive_chats(self.id)
