Убрал бардак из разных скобок. Изменил импорты в app.py

This commit is contained in:
Daniil
2021-01-07 14:37:44 +02:00
parent 1bbe12e9c1
commit 3f4b06933a
8 changed files with 25 additions and 29 deletions

View File

@@ -1,7 +1,8 @@
from aiogram import Dispatcher
from loader import dp
from .throttling import ThrottlingMiddleware
def setup(dp: Dispatcher):
if __name__ == "a.middlewares":
dp.middleware.setup(ThrottlingMiddleware())

View File

@@ -21,8 +21,8 @@ class ThrottlingMiddleware(BaseMiddleware):
handler = current_handler.get()
dispatcher = Dispatcher.get_current()
if handler:
limit = getattr(handler, 'throttling_rate_limit', self.rate_limit)
key = getattr(handler, 'throttling_key', f"{self.prefix}_{handler.__name__}")
limit = getattr(handler, "throttling_rate_limit", self.rate_limit)
key = getattr(handler, "throttling_key", f"{self.prefix}_{handler.__name__}")
else:
limit = self.rate_limit
key = f"{self.prefix}_message"
@@ -34,4 +34,4 @@ class ThrottlingMiddleware(BaseMiddleware):
async def message_throttled(self, message: types.Message, throttled: Throttled):
if throttled.exceeded_count <= 2:
await message.reply('Too many requests! ')
await message.reply("Too many requests!")