mirror of
https://github.com/MrSedan/neuro-reply-bot-reworked.git
synced 2026-01-14 21:49:42 +03:00
25 lines
917 B
Python
25 lines
917 B
Python
from typing import Any, Awaitable, Callable, Dict
|
|
|
|
from aiogram import BaseMiddleware
|
|
from aiogram.types import Message
|
|
from sqlalchemy.orm import Session
|
|
|
|
from db.data import Admin, User, engine
|
|
|
|
|
|
class AdminMiddleware(BaseMiddleware):
|
|
def __init__(self) -> None:
|
|
pass
|
|
|
|
async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data: Dict[str, Any]) -> Any:
|
|
with Session(engine) as session:
|
|
if not session.get(User, event.from_user.id):
|
|
user = User(id=event.from_user.id, user_name=event.from_user.username)
|
|
session.add(user)
|
|
session.commit()
|
|
isAdmin = session.get(Admin, event.from_user.id)
|
|
if not isAdmin:
|
|
await event.answer('Команда только для админов!')
|
|
return None
|
|
return await handler(event, data)
|
|
|