tgadmin/bot/middlewares/__init__.py
2024-07-25 15:19:15 +03:00

28 lines
883 B
Python

from config import *
from .check_ban import BannedMiddleware
from .create_user import CreateUserMiddleware
from .throttling import ThrottlingMiddleware
from aiogram.utils.callback_answer import CallbackAnswerMiddleware
def setup(dp: Dispatcher):
throttling_middleware = ThrottlingMiddleware()
dp.message.outer_middleware.register(throttling_middleware)
dp.callback_query.outer_middleware.register(throttling_middleware)
create_user_middleware = CreateUserMiddleware()
dp.message.outer_middleware.register(create_user_middleware)
dp.callback_query.outer_middleware.register(create_user_middleware)
banned_middleware = BannedMiddleware()
dp.message.outer_middleware.register(banned_middleware)
dp.callback_query.outer_middleware.register(banned_middleware)
dp.callback_query.middleware(CallbackAnswerMiddleware())