Added Back Button and fixed the TotalQuestions Database logical error.

Author: advnpzn

__main__.py

@@ -1,4 +1,4 @@
-from os import cpu_count
+from os import cpu_count, terminal_size
 import akinator
 from telegram.files.inputmedia import InputMediaPhoto
 from random import randint
@@ -8,7 +8,7 @@
 from telegram.ext import Updater, CommandHandler, CallbackContext, CallbackQueryHandler
 from config import BOT_TOKEN
 from database import addUser, getChildMode, getCorrectGuess, getLanguage, getTotalGuess, getTotalQuestions, getUnfinishedGuess, getUser, getWrongGuess, totalUsers, updateChildMode, updateCorrectGuess, updateLanguage, updateTotalGuess, updateTotalQuestions, updateWrongGuess
-from strings import AKI_LANG_CODE, AKI_LANG_MSG, CHILDMODE_MSG, ME_MSG, START_MSG
+from strings import AKI_FIRST_QUESTION, AKI_LANG_CODE, AKI_LANG_MSG, CHILDMODE_MSG, ME_MSG, START_MSG
 import akinator
 
 
@@ -53,11 +53,18 @@ def aki_play_callback_handler(update: Update, context:CallbackContext) -> None:
     user_id = update.effective_user.id
     aki = context.user_data[f"aki_{user_id}"]
     q = context.user_data[f"q_{user_id}"]
-    updateTotalQuestions(user_id, context.user_data[f"ques_{user_id}"])
-    context.user_data[f"ques_{user_id}"] = context.user_data[f"ques_{user_id}"] + 1
+    updateTotalQuestions(user_id, 1)
     query = update.callback_query
     a = query.data.split('_')[-1]
-    q = aki.answer(a)
+    if a == '5':
+        updateTotalQuestions(user_id, -1)
+        try:
+            q = aki.back()
+        except akinator.exceptions.CantGoBackAnyFurther:
+            query.answer(text=AKI_FIRST_QUESTION, show_alert=True)
+            return
+    else:
+        q = aki.answer(a)
     query.answer()
     if aki.progression < 80:
         query.message.edit_media(
@@ -167,7 +174,6 @@ def aki_set_child_mode(update: Update, context: CallbackContext) -> None:
 def del_data(context:CallbackContext, user_id: int):
     del context.user_data[f"aki_{user_id}"]
     del context.user_data[f"q_{user_id}"]
-    del context.user_data[f"ques_{user_id}"]
 
 def main():
     updater = Updater(token=BOT_TOKEN)

database.py

@@ -163,5 +163,6 @@ def updateTotalQuestions(user_id: int, total_questions: int) -> None:
     """
     
     """
+    total_questions = total_questions+ getTotalQuestions(user_id)
     my_db["users"].update_one({"user_id": user_id}, {"$set": {"total_questions": total_questions}})
 

keyboard.py

@@ -62,6 +62,8 @@
         [
             InlineKeyboardButton("I don't know", callback_data='aki_play_2'),
             InlineKeyboardButton("Probably Not", callback_data='aki_play_4')
+        ],
+        [   InlineKeyboardButton("Back", callback_data= 'aki_play_5')
         ]
     ]
 )

strings.py

@@ -49,4 +49,6 @@
 CHILDMODE_MSG = """
 If Child mode is enabled, akinator won't show any NSFW content.
 <b>Current Status :</b> <pre>Child mode is {} !</pre>
-"""
+"""
+
+AKI_FIRST_QUESTION = "This is the first question. You can't go back any further!"