Basealt adaptation

1 files changed, 17 insertions, 12 deletions

diff --git a/states.py b/states.py
index aa40286..316357f 100644
--- a/states.py
+++ b/states.py
@@ -43,8 +43,8 @@ class States:
     @save_state
     def prompt_question(bot: Bot, update: Update, user_data: dict):
         update.message.reply_text(
-            "Введите свой вопрос, админы вам ответят, как только появится возможность, "
-            "поэтому следите за сообщениями от бота.",
+            "Введите свой вопрос, он будет перенаправлен представителям компании. "
+            "Вам ответят как только появится возможность, поэтому следите за сообщениями от бота.",
             reply_markup=ReplyKeyboardMarkup(BackToMenuKeyboard.get_keyboard())
         )
 
@@ -70,11 +70,11 @@ class States:
                 )
 
         update.message.reply_text(
-            "Ваш вопрос был отправлен на рассмотрение, ожидайте ответ.",
+            "Ваш вопрос отправлен на рассмотрение, ожидайте ответ.",
             reply_markup=ReplyKeyboardMarkup(ContinueKeyboard.get_keyboard())
         )
 
-        return ANSWER_RIGHT
+        return States.main_menu(bot, update, user_data)
 
 
     @staticmethod
@@ -100,9 +100,12 @@ class States:
         menu_text = [
 
         ]
+        solved = False
 
         full_score = Decimal(0.0)
         if status_code == 200:
+            if len(response) == 5:
+                solved = True
             if len(response) != 0:
                 menu_text.append("Твои решенные задачи:")
 
@@ -118,7 +121,8 @@ class States:
                         fp = dt.datetime.strptime(attempt["task"]["first_published"], "%Y-%m-%dT%H:%M:%SZ")
 
                     t = Decimal((ts - fp).total_seconds()) / Decimal(60)
-                    plr_score = calc_score(t, attempt["task"]["base_score"])
+                    # plr_score = calc_score(t, attempt["task"]["base_score"])
+                    plr_score = int(attempt["task"]["base_score"])
                     full_score += plr_score
 
                     menu_text.append(
@@ -138,8 +142,13 @@ class States:
         update.message.reply_text("\n".join(menu_text), parse_mode="Markdown")
 
         update.message.reply_text(
+            "*У тебя решены все задачи! Подойди к нам, чтобы получить приз.*",
+            parse_mode="Markdown"
+        )
+
+        update.message.reply_text(
             "Выбери следующее действие...",
-            reply_markup=ReplyKeyboardMarkup(MenuKeyboard.get_keyboard(update.message.from_user.id))
+            reply_markup=ReplyKeyboardMarkup(MenuKeyboard.get_keyboard(update.message.from_user.id, solved))
         )
 
         return MAIN_MENU
@@ -306,8 +315,6 @@ class States:
         message = '\n'.join([
             f"*{task['title']}*",
             f"{task['statement']}",
-            "",
-            f"_Теги: {task['tags']}_",
         ])
         keyboard = TaskChosenKeyboard.get_keyboard()
 
@@ -355,11 +362,9 @@ class States:
 
             if answer == task["answer"]:
                 update.message.reply_text(
-                    "Ты ввел правильный ответ! Возвращайся к другим задачам",
-                    reply_markup=ReplyKeyboardMarkup(ContinueKeyboard.get_keyboard())
+                    "Ты ввел правильный ответ! Возвращаемся в главное меню."
                 )
-
-                return ANSWER_RIGHT
+                return States.main_menu(bot, update, user_data)
             else:
                 update.message.reply_text(
                     "К сожалению, твой ответ неверный =( Попробуй ввести другой ответ.",