From 5b8b950aafe69ff733fa0b4e516da908a407c295 Mon Sep 17 00:00:00 2001 From: Zakhar Izmaylov Date: Sun, 17 Nov 2024 15:39:27 +0300 Subject: [PATCH] Fix admin middleware --- middlewares/admin.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/middlewares/admin.py b/middlewares/admin.py index 1e532d7c..d7f0c99b 100644 --- a/middlewares/admin.py +++ b/middlewares/admin.py @@ -4,6 +4,7 @@ from aiogram import BaseMiddleware from aiogram.types import TelegramObject from config import ADMIN_ID +from logger import logger class AdminMiddleware(BaseMiddleware): @@ -13,12 +14,15 @@ class AdminMiddleware(BaseMiddleware): event: TelegramObject, data: Dict[str, Any], ) -> Any: - if isinstance(ADMIN_ID, list): - if event.from_user.id in ADMIN_ID: - data["admin"] = True - elif isinstance(ADMIN_ID, int): - if event.from_user.id == ADMIN_ID: - data["admin"] = True - else: + try: + if isinstance(ADMIN_ID, list): + if event.from_user.id in ADMIN_ID: + data["admin"] = True + elif isinstance(ADMIN_ID, int): + if event.from_user.id == ADMIN_ID: + data["admin"] = True + except Exception as e: + logger.error(e) + finally: data["admin"] = False return await handler(event, data)