Delete wraps

This commit is contained in:
Zakhar Izmaylov
2024-11-09 12:01:24 +03:00
parent eec5e8c887
commit 7d51ff3a85
3 changed files with 184 additions and 213 deletions
-27
View File
@@ -28,30 +28,3 @@ class AdminMiddleware(BaseMiddleware):
data["is_admin"] = True
return await handler(event, data)
def admin_only():
def decorator(func):
@wraps(func)
async def wrapper(*args, **kwargs):
# Извлекаем объект события (Message или CallbackQuery)
event = args[0]
# Определяем ID пользователя
user_id = event.from_user.id if hasattr(event, "from_user") else None
if user_id not in int(ADMIN_ID):
# Можно отправить сообщение или просто return
if isinstance(event, Message):
await event.answer("У вас нет доступа к этой команде.")
elif isinstance(event, CallbackQuery):
await event.answer(
"У вас нет доступа к этому действию.", show_alert=True
)
return None
return await func(*args, **kwargs)
return wrapper
return decorator