21 lines
750 B
Python
21 lines
750 B
Python
from aiogram import BaseMiddleware
|
|
from aiogram.types import Message, CallbackQuery
|
|
from config import ADMIN_ID
|
|
|
|
maintenance_mode = False
|
|
|
|
class MaintenanceModeMiddleware(BaseMiddleware):
|
|
async def __call__(self, handler, event, data):
|
|
if maintenance_mode:
|
|
user_id = None
|
|
if isinstance(event, Message):
|
|
user_id = event.from_user.id
|
|
elif isinstance(event, CallbackQuery):
|
|
user_id = event.from_user.id
|
|
|
|
if user_id and user_id not in ADMIN_ID:
|
|
await event.answer("⚙️ Бот временно недоступен. Ведутся технические работы.")
|
|
return
|
|
|
|
return await handler(event, data)
|