neuro_llc/neuro-reply-bot-reworked
2
0
Fork 0
mirror of https://github.com/MrSedan/neuro-reply-bot-reworked.git synced 2026-01-14 21:49:42 +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

6
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:
@@ -18,7 +17,8 @@ class AdminMiddleware(BaseMiddleware):
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)