add admin logic

app/plugin/register.py

@@ -1,4 +1,3 @@
-import telebot
 from telebot import types
 from plugin.user import User
 from config import Messages
@@ -19,7 +18,7 @@ def start_reg_age(message: types.Message):
     user = User(message.chat.id)
     user.age = int(message.text)
 
-    markup = types.InlineKeyboardMarkup(resize_keyboard=True)
+    markup = types.InlineKeyboardMarkup()
     markup.add(
         types.InlineKeyboardButton(Messages.GENDER_WOMAN, callback_data="gender_woman"),
         types.InlineKeyboardButton(Messages.GENDER_MAN, callback_data="gender_man"),
@@ -59,7 +58,7 @@ def start_reg_group(message: types.Message):
     user = User(message.chat.id)
     user.group = message.text
 
-    markup = types.InlineKeyboardMarkup(resize_keyboard=True)
+    markup = types.InlineKeyboardMarkup()
     markup.add(
         types.InlineKeyboardButton("Всё верно!", callback_data="registration_confirm"),
         types.InlineKeyboardButton("Изменить", callback_data="registration_change"),
@@ -97,6 +96,9 @@ def start_reg_again(call: types.CallbackQuery):
     bot.register_next_step_handler(call.message, start_reg_name)
 
 
+# --------------------------------- Test ---------------------------------
+
+
 @bot.callback_query_handler(func=lambda call: call.data == "registration_confirm")
 def start_test(call: types.CallbackQuery):
     user = User(call.message.chat.id)
@@ -104,6 +106,65 @@ def start_test(call: types.CallbackQuery):
         call.message.chat.id, call.message.message_id, reply_markup=None
     )
 
-    bot.send_message(call.message.chat.id, TestMessages.TEST_WELCOME)
+    markup = types.InlineKeyboardMarkup()
+    markup.add(
+        types.InlineKeyboardButton("Начнём!", callback_data="test_yes"),
+    )
+    bot.send_message(
+        call.message.chat.id, TestMessages.TEST_WELCOME, reply_markup=markup
+    )
+    user.type = ""
 
-    pass
+    return
+
+
+@bot.callback_query_handler(func=lambda call: "test" in call.data)
+def test_question(call: types.CallbackQuery):
+    user = User(call.message.chat.id)
+
+    bot.edit_message_reply_markup(
+        call.message.chat.id, call.message.message_id, reply_markup=None
+    )
+
+    if len(user.type) == 4:
+        show_result(call.message)
+        return
+
+    if call.data != "test_yes":
+        user.type += call.data.split("_")[-1]
+
+        bot.edit_message_text(
+            call.message.text
+            + "\n\n"
+            + TestMessages.TEST_QUESTIONS[len(user.type) - 1]["answers"][
+                int(call.data[-1])
+            ][0],
+            call.message.chat.id,
+            call.message.message_id,
+        )  # упростить
+
+    question = TestMessages.TEST_QUESTIONS[len(user.type)]
+    markup = types.InlineKeyboardMarkup()
+    for answer in question["answers"]:
+        markup.add(
+            types.InlineKeyboardButton(answer[0], callback_data=f"test_{answer[1]}")
+        )
+
+    return
+
+
+def show_result(message: types.Message):
+    user = User(message.chat.id)
+    text = TestMessages.TEST_RESULT.format(
+        TestMessages.TEST_RESULTS[user.type][0],
+        TestMessages.TEST_RESULTS[user.type][1],
+    )
+    bot.send_photo(
+        message.chat.id,
+        open(f"pics/{user.type}.jpg", "rb"),
+        caption=text,
+    )
+
+    bot.send_message(message.chat.id, TestMessages.TEST_FINISH)
+
+    return