Безопасная админка: /admin <пароль> из .env, user_id проверка, молчаливый отказ

This commit is contained in:
devproject
2026-06-14 21:56:31 +04:00
parent b6d5b59132
commit c1f06d299e
2 changed files with 32 additions and 27 deletions
+1
View File
@@ -4,6 +4,7 @@ from dotenv import load_dotenv
load_dotenv() load_dotenv()
BOT_TOKEN = os.getenv("BOT_TOKEN") BOT_TOKEN = os.getenv("BOT_TOKEN")
ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "admin123")
if not BOT_TOKEN: if not BOT_TOKEN:
raise ValueError("BOT_TOKEN not found in environment") raise ValueError("BOT_TOKEN not found in environment")
+27 -23
View File
@@ -1,56 +1,60 @@
"""Admin handler for spa-telegram-bot — secure with password."""
from aiogram import Router, F from aiogram import Router, F
from aiogram.types import Message, CallbackQuery from aiogram.types import Message, CallbackQuery
from aiogram.filters import Command from aiogram.filters import Command, CommandObject
from app.db.memory import load_bookings from app.db.memory import load_bookings
from app.config import ADMIN_IDS from app.config import ADMIN_IDS, ADMIN_PASSWORD
from app.keyboards.menu import menu
from app.logger import logger from app.logger import logger
router = Router() router = Router()
def is_admin(user_id: int) -> bool: def is_admin(user_id: int, password: str | None = None) -> bool:
return user_id in ADMIN_IDS """Check if user is admin: by user_id or by password."""
if user_id in ADMIN_IDS:
return True
if password and password == ADMIN_PASSWORD:
return True
return False
@router.message(Command("admin")) @router.message(Command("admin"))
async def admin_panel(message: Message): async def admin_panel(message: Message, command: CommandObject):
if not is_admin(message.from_user.id): uid = message.from_user.id
await message.answer(
"⛔️ Доступ запрещен" # Get password from command args
) password = command.args
if not is_admin(uid, password=password):
# Silent — no hint about how to get in
return return
logger.info("Admin {uid} opened admin panel", uid=message.from_user.id) logger.info("Admin {uid} opened admin panel", uid=uid)
bookings = load_bookings() bookings = load_bookings()
active = [b for b in bookings if b.get("status") == "active"] active = [b for b in bookings if b.get("status") == "active"]
if not active: if not active:
await message.answer( await message.answer("📋 Активных записей нет")
"📋 Активных записей нет"
)
return return
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
for booking in active: for booking in active:
buttons = [ buttons = [[
[
InlineKeyboardButton( InlineKeyboardButton(
text="❌ Отменить", text="❌ Отменить",
callback_data=f"admin_cancel_{booking['id']}" callback_data=f"admin_cancel_{booking['id']}"
) )
] ]]
]
user_name = booking.get("user_name") or f"id{booking['user_id']}" user_name = booking.get("user_name") or f"id{booking['user_id']}"
await message.answer( await message.answer(
f"📋 Запись #{booking['id']}\n" f"📋 Запись #{booking['id']}\n"
f"👤 Клиент: {user_name}\n" f"👤 Клиент: {user_name}\n"
f"💆 Услуга: {booking['service']}\n" f"💆 Услуги: {booking.get('service', '?')}\n"
f"👤 Мастер: {booking['master']}\n" f"👤 Мастер: {booking.get('master', '?')}\n"
f"📅 Дата: {booking['date']}\n" f"📅 Дата: {booking.get('date', '?')}\n"
f"🕒 Время: {booking['time']}", f"🕒 Время: {booking.get('time', '?')}",
reply_markup=InlineKeyboardMarkup(inline_keyboard=buttons) reply_markup=InlineKeyboardMarkup(inline_keyboard=buttons)
) )
@@ -58,7 +62,7 @@ async def admin_panel(message: Message):
@router.callback_query(F.data.startswith("admin_cancel_")) @router.callback_query(F.data.startswith("admin_cancel_"))
async def admin_cancel_booking(callback: CallbackQuery): async def admin_cancel_booking(callback: CallbackQuery):
if not is_admin(callback.from_user.id): if not is_admin(callback.from_user.id):
await callback.answer(" Нет доступа") await callback.answer("⛔ Нет доступа")
return return
booking_id = int(callback.data.split("_")[2]) booking_id = int(callback.data.split("_")[2])