neuro_llc/neuro-reply-bot-reworked
2
0
Fork 0
mirror of https://github.com/MrSedan/neuro-reply-bot-reworked.git synced 2026-01-15 05:59:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

Admin check

Browse Source
This commit is contained in:
Sergey Elpashev
2023-11-04 05:32:00 +03:00
parent 29743114dd
commit 3a7e80adcb
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
handlers/middlewares/user.py
View File

@@ -4,9 +4,8 @@ from aiogram import BaseMiddleware
from aiogram.types import Message from aiogram.types import Message
from sqlalchemy.orm import Session 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): class AdminMiddleware(BaseMiddleware):
def __init__(self) -> None: def __init__(self) -> None:
@@ -14,11 +13,12 @@ class AdminMiddleware(BaseMiddleware):
async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data: Dict[str, Any]) -> Any: async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data: Dict[str, Any]) -> Any:
with Session(engine) as session: with Session(engine) as session:
if not session.get(User, event.from_user.id): if not session.get(User, event.from_user.id):
user = User(id=event.from_user.id, user_name=event.from_user.username) user = User(id=event.from_user.id, user_name=event.from_user.username)
session.add(user) session.add(user)
session.commit() 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('Команда только для админов!') await event.answer('Команда только для админов!')
return None return None
return await handler(event, data) return await handler(event, data)