Add set_slow_mode method

delivrance · delivrance · commit d71d9686d7e3 · 2019-10-27T11:02:38.000+01:00
diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py
@@ -212,6 +212,7 @@ def get_title_list(s: str) -> list:
             create_supergroup
             delete_channel
             delete_supergroup
+            set_slow_mode
         """,
         users="""
         Users
diff --git a/compiler/error/source/400_BAD_REQUEST.tsv b/compiler/error/source/400_BAD_REQUEST.tsv
@@ -132,3 +132,4 @@ ADMIN_RANK_EMOJI_NOT_ALLOWED	Emojis are not allowed in custom administrator titl
 FILE_REFERENCE_EMPTY	The file reference is empty
 FILE_REFERENCE_INVALID	The file reference is invalid
 REPLY_MARKUP_TOO_LONG	The reply markup is too long
+SECONDS_INVALID	The seconds interval is invalid, for slow mode try with 0 (off), 10, 30, 60 (1m), 300 (5m), 900 (15m) or 3600 (1h)
diff --git a/compiler/error/source/420_FLOOD.tsv b/compiler/error/source/420_FLOOD.tsv
@@ -1,3 +1,4 @@
 id	message
 FLOOD_WAIT_X	A wait of {x} seconds is required
 TAKEOUT_INIT_DELAY_X	You have to confirm the data export request using one of your mobile devices or wait {x} seconds
+SLOWMODE_WAIT_X	A wait of {x} seconds is required to send messages in this chat.
diff --git a/pyrogram/client/methods/chats/__init__.py b/pyrogram/client/methods/chats/__init__.py
@@ -49,7 +49,7 @@
 from .unban_chat_member import UnbanChatMember
 from .unpin_chat_message import UnpinChatMessage
 from .update_chat_username import UpdateChatUsername
-
+from .set_slow_mode import SetSlowMode
 
 class Chats(
     GetChat,
@@ -84,6 +84,7 @@ class Chats(
     DeleteChannel,
     DeleteSupergroup,
     GetNearbyChats,
-    SetAdministratorTitle
+    SetAdministratorTitle,
+    SetSlowMode
 ):
     pass
diff --git a/pyrogram/client/methods/chats/set_slow_mode.py b/pyrogram/client/methods/chats/set_slow_mode.py
@@ -0,0 +1,57 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017-2019 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 typing import Union
+
+from pyrogram.api import functions
+from ...ext import BaseClient
+
+
+class SetSlowMode(BaseClient):
+    def set_slow_mode(
+        self,
+        chat_id: Union[int, str],
+        seconds: int,
+    ) -> bool:
+        """Set the slow mode interval for a chat.
+
+        Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+
+            seconds (``int`` | ``str``):
+                Seconds in which members will be able to send only one message per this interval.
+                Valid values are: 0 (off), 10, 30, 60 (1m), 300 (5m), 900 (15m) or 3600 (1h).
+
+        Returns:
+            ``bool``: True on success.
+
+        Example:
+            .. code-block:: python
+
+                app.set_slow_mode("pyrogramchat", 60)
+        """
+
+        self.send(
+            functions.channels.ToggleSlowMode(
+                channel=self.resolve_peer(chat_id),
+                seconds=seconds
+            )
+        )
+
+        return True
