Update Middleware
This commit is contained in:
+11
-5
@@ -1,3 +1,5 @@
|
||||
from typing import Union
|
||||
|
||||
from aiogram.filters import BaseFilter
|
||||
from aiogram.types import Message
|
||||
|
||||
@@ -6,9 +8,13 @@ from config import ADMIN_ID
|
||||
|
||||
class IsAdminFilter(BaseFilter):
|
||||
async def __call__(self, message: Message) -> bool:
|
||||
if isinstance(ADMIN_ID, list):
|
||||
return message.from_user.id in ADMIN_ID
|
||||
elif isinstance(ADMIN_ID, int):
|
||||
return message.from_user.id == ADMIN_ID
|
||||
else:
|
||||
try:
|
||||
admin_ids: Union[int, list[int]] = ADMIN_ID
|
||||
|
||||
if isinstance(admin_ids, list):
|
||||
return message.from_user.id in admin_ids
|
||||
|
||||
return message.from_user.id == admin_ids
|
||||
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
Reference in New Issue
Block a user