admin_menu.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334
  1. from aiogram import types, Dispatcher
  2. from bot.keyboards import register_kb, main_kb, admin_keyboard
  3. from bot import database
  4. async def enter_admin_menu(message: types.message):
  5. db = database.Database()
  6. if not db.sql_fetchone(sql=f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
  7. not db.sql_fetchone(sql=f"select approved from user_table where tg_id={message.from_user.id}"):
  8. await message.delete()
  9. await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
  10. if db.sql_fetchone(sql=f'select admin from user_table where tg_id = {message.from_user.id}') == "0":
  11. await message.answer("Доступ только для администраторов")
  12. else:
  13. await message.delete()
  14. await message.answer(f"Панель управления лакеем\n\n"
  15. f"Здесь вы можете управлять заявками", reply_markup=admin_keyboard)
  16. async def exit_admin_menu(message: types.message):
  17. db = database.Database()
  18. if not db.sql_fetchone(sql=f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
  19. not db.sql_fetchone(sql=f"select approved from user_table where tg_id={message.from_user.id}"):
  20. await message.delete()
  21. await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
  22. elif not db.sql_fetchone(sql=f'select admin from user_table where tg_id = {message.from_user.id}'):
  23. await message.answer("Доступ только для администраторов", reply_markup=main_kb)
  24. else:
  25. await message.answer("Выход", reply_markup=main_kb)
  26. def register_admin_menu(dp: Dispatcher):
  27. dp.register_message_handler(enter_admin_menu, text="👮 Управление")
  28. dp.register_message_handler(exit_admin_menu, text="Выйти")