Admin check

This commit is contained in:
2023-11-04 05:32:00 +03:00
parent 29743114dd
commit 3a7e80adcb

View File

@@ -4,9 +4,8 @@ from aiogram import BaseMiddleware
from aiogram.types import Message
from sqlalchemy.orm import Session
from db.data import User, engine
from db.data import Admin, User, engine
ADMIN_LIST = [248770879, 395543883]
class AdminMiddleware(BaseMiddleware):
def __init__(self) -> None:
@@ -18,7 +17,8 @@ class AdminMiddleware(BaseMiddleware):
user = User(id=event.from_user.id, user_name=event.from_user.username)
session.add(user)
session.commit()
if event.from_user.id not in ADMIN_LIST:
isAdmin = session.get(Admin, event.from_user.id)
if not isAdmin:
await event.answer('Команда только для админов!')
return None
return await handler(event, data)