New features

handlers/admin/operator_control/__init__.py

@@ -0,0 +1,2 @@
+from . import adding
+from . import delete

handlers/admin/operator_control/adding.py

@@ -0,0 +1,35 @@
+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)

handlers/admin/operator_control/delete.py

@@ -0,0 +1,18 @@
+from aiogram import types
+
+from load import dp, bot, messages
+from keyboard.inline.operator.user import item_list
+from utils.database.base import get_full_operator
+
+
+@dp.message_handler(lambda x: x.text == messages.del_operator, is_admin=True)
+async def del_admin(message: types.Message):
+    users = get_full_operator()
+    user = users[0]
+    result = messages.admin_user.format(**user)
+    await bot.send_message(
+        message.chat.id,
+        result,
+        parse_mode="Markdown",
+        reply_markup=item_list(items=len(users) - 1)
+    )