36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
from aiogram import types
|
||
from aiogram.dispatcher import FSMContext
|
||
|
||
from load import dp, bot, messages
|
||
from utils.database.user import Register, User
|
||
from keyboard.default.admin.main_menu import base_menu
|
||
from keyboard.default.main_menu import back_to_main_menu
|
||
|
||
|
||
@dp.message_handler(lambda x: x.text == messages.add_operator, is_admin=True)
|
||
async def add_operator(message: types.Message, state: FSMContext):
|
||
await bot.send_message(
|
||
message.chat.id,
|
||
("Отправьте user_id пользователя\n"
|
||
"Его можно узнать зайдя в информацию о пользователе"),
|
||
reply_markup=base_menu()
|
||
)
|
||
await state.set_state("get_user_id_add_operator")
|
||
|
||
|
||
@dp.message_handler(is_admin=True, state="get_user_id_add_operator")
|
||
async def add_operator1(message: types.Message, state: FSMContext):
|
||
await state.finish()
|
||
user_id = message.text
|
||
if not user_id.isdigit() or "-100" in user_id:
|
||
await bot.send_message(message.chat.id, "Данные не правильные!", reply_markup=back_to_main_menu)
|
||
return
|
||
|
||
user_id = int(user_id)
|
||
user = User.get_user(user_id)
|
||
if not user:
|
||
await bot.send_message(message.chat.id, "Пользователь не существует!", reply_markup=back_to_main_menu)
|
||
return
|
||
Register.register_operator(user)
|
||
await bot.send_message(message.chat.id, "Оператор добавлен!", reply_markup=back_to_main_menu)
|