From ed6ae575d19a1229856c163acd865363be4bd5c4 Mon Sep 17 00:00:00 2001
From: Alex Root Junior <jroot.junior@gmail.com>
Date: Sat, 2 Dec 2017 22:57:35 +0200
Subject: [PATCH] Add example for throttling manager (anti-flood)

---
 examples/throtling_example.py | 43 +++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 examples/throtling_example.py

diff --git a/examples/throtling_example.py b/examples/throtling_example.py
new file mode 100644
index 00000000..e1bda994
--- /dev/null
+++ b/examples/throtling_example.py
@@ -0,0 +1,43 @@
+"""
+Example for throttling manager.
+
+You can use that for flood controlling.
+"""
+
+import asyncio
+import logging
+
+from aiogram import Bot, types
+from aiogram.contrib.fsm_storage.memory import MemoryStorage
+from aiogram.dispatcher import Dispatcher
+from aiogram.utils.exceptions import Throttled
+from aiogram.utils.executor import start_polling
+
+API_TOKEN = 'BOT TOKEN HERE'
+
+logging.basicConfig(level=logging.INFO)
+
+loop = asyncio.get_event_loop()
+bot = Bot(token=API_TOKEN, loop=loop)
+
+# Throttling manager does not working without Leaky Bucket.
+# Then need to use storage's. For example use simple in-memory storage.
+storage = MemoryStorage()
+dp = Dispatcher(bot, storage=storage)
+
+
+@dp.message_handler(commands=['start', 'help'])
+async def send_welcome(message: types.Message):
+    try:
+        # Execute throttling manager with rate-limit equals to 2 seconds for key "start"
+        await dp.throttle('start', rate=2)
+    except Throttled:
+        # If request is throttled the `Throttled` exception will be raised.
+        await message.reply('Too many requests!')
+    else:
+        # Otherwise do something.
+        await message.reply("Hi!\nI'm EchoBot!\nPowered by aiogram.")
+
+
+if __name__ == '__main__':
+    start_polling(dp, loop=loop, skip_updates=True)