Decorator that lets you apply per chat rate limits to telethon handlers

telethon_ratelimiter.py

import asyncio
import logging
import time
from collections import defaultdict

from telethon import TelegramClient
from telethon import events

logging.basicConfig(level=logging.INFO)

client = TelegramClient('user', 6, 'eb06d4abfb49dc3eeb1aeb98ae0f581e').start()


def handler_ratelimit(timeout):
    def wrapper(function):
        last_called = defaultdict(int)

        async def wrapped(event, *args, **kwargs):
            current_time = time.time()
            if current_time - last_called[event.chat_id] < timeout:
                return
            last_called[event.chat_id] = current_time
            return await function(event, *args, **kwargs)
        return wrapped
    return wrapper


@client.on(events.NewMessage(incoming=True, pattern='^hi$'))
@handler_ratelimit(60)
async def hi(event):
    await event.reply('hi')


client.run_until_disconnected()