rc-tgbot/app/plugin/register.py

from telebot import types
from plugin.user import User
from config import Messages
from plugin.test import TestMessages

from plugin.bot_instance import bot


def start_reg_name(message: types.Message):
    user = User(message.chat.id, message.text)

    bot.send_message(message.chat.id, Messages.ENTER_AGE)

    bot.register_next_step_handler(message, start_reg_age)


def start_reg_age(message: types.Message):
    user = User(message.chat.id)
    user.age = int(message.text)

    markup = types.InlineKeyboardMarkup()
    markup.add(
        types.InlineKeyboardButton(Messages.GENDER_WOMAN, callback_data="gender_woman"),
        types.InlineKeyboardButton(Messages.GENDER_MAN, callback_data="gender_man"),
    )
    bot.send_message(message.chat.id, Messages.ENTER_GENDER)

    bot.register_next_step_handler(message, start_reg_gender)


@bot.callback_query_handler(func=lambda call: "gender" in call.data)
def start_reg_gender(call: types.CallbackQuery):
    user = User(call.message.chat.id)
    user.gender = call.data.split("_")[-1]

    bot.edit_message_text(
        Messages.GENDER_WOMAN if user.gender == "woman" else Messages.GENDER_MAN,
        call.message.chat.id,
        call.message.message_id,
        reply_markup=None,
    )

    bot.send_message(call.message.chat.id, Messages.ENTER_FACULTY)

    bot.register_next_step_handler(call.message, start_reg_faculty)


def start_reg_faculty(message: types.Message):
    user = User(message.chat.id)
    user.faculty = message.text

    bot.send_message(message.chat.id, Messages.ENTER_GROUP)

    bot.register_next_step_handler(message, start_reg_group)


def start_reg_group(message: types.Message):
    user = User(message.chat.id)
    user.group = message.text

    markup = types.InlineKeyboardMarkup()
    markup.add(
        types.InlineKeyboardButton("Всё верно!", callback_data="registration_confirm"),
        types.InlineKeyboardButton("Изменить", callback_data="registration_change"),
    )
    bot.send_message(
        message.chat.id,
        Messages.REGISTRATION_CONFIRM.format(
            user.name,
            user.age,
            user.gender,
            user.faculty,
            user.group,
        ),
        reply_markup=markup,
    )


@bot.callback_query_handler(func=lambda call: call.data == "registration_skip")
def start_skip_reg(call: types.CallbackQuery):
    bot.edit_message_reply_markup(
        call.message.chat.id, call.message.message_id, reply_markup=None
    )

    return


@bot.callback_query_handler(func=lambda call: call.data == "registration_change")
def start_reg_again(call: types.CallbackQuery):
    bot.edit_message_reply_markup(
        call.message.chat.id, call.message.message_id, reply_markup=None
    )

    bot.send_message(call.message.chat.id, Messages.ENTER_NAME)

    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)
    bot.edit_message_reply_markup(
        call.message.chat.id, call.message.message_id, reply_markup=None
    )

    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 = ""

    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