move project files to src

1 files changed, 166 insertions, 0 deletions

diff --git a/src/bot.py b/src/bot.py
new file mode 100644
index 0000000..37e0ac8
--- /dev/null
+++ b/src/bot.py
@@ -0,0 +1,166 @@
+from telegram.ext import Updater, CommandHandler, ConversationHandler, MessageHandler, Filters, CallbackQueryHandler
+from telegram import ReplyKeyboardMarkup, ReplyKeyboardRemove, InlineKeyboardMarkup, InlineKeyboardButton
+
+# Typing
+from telegram import Update, User, Bot
+
+from os import getenv
+import json
+from config import TG_TOKEN, REQUEST_KWARGS
+
+import backend_api
+from keyboards import (
+    MenuKeyboard, TasksKeyboard, TaskChosenKeyboard, ContinueKeyboard,
+    AnsweringKeyboard, AdminKeyboard, BackToMenuKeyboard
+)
+from utils import *
+from states import States, AdminStates
+
+from botlogging import logger
+
+
+def start(bot: Bot, update: Update, user_data: dict):
+    user_data.update({"chosen_task": None})
+
+    update.message.reply_text(
+        "Квиз Базальт СПО на День работодателя КНиИТ СГУ",
+        reply_markup=ReplyKeyboardRemove()
+    )
+
+    user: User = update.message.from_user
+    if user.username is None:
+        return States.wait_for_username(bot, update, user_data)
+
+    else:
+        logger.debug(backend_api.register_user(user.id, user.username, user.full_name))
+        return States.main_menu(bot, update, user_data)
+
+
+def username_check(bot: Bot, update: Update, user_data):
+    user: User = update.message.from_user
+    if user.username is None:
+        return States.wait_for_username(bot, update, user_data)
+
+    else:
+        logger.debug(backend_api.register_user(user.id, user.username, user.full_name))
+        return States.main_menu(bot, update, user_data)
+
+
+def resume_bot(bot: Bot, update: Update, user_data):
+    status_code, state = backend_api.get_state(update.message.from_user.id)
+
+    user_data.update(json.loads(state["user_data"]))
+    last_state = state["last_state"]
+
+    for handler in conversation_handler.states[last_state]:
+        if handler.filters.filter(update):
+            return handler.callback(bot, update, user_data)
+
+    return last_state
+
+
+def stop(bot, update):
+    update.message.reply_text('Пока!', reply_markup=ReplyKeyboardRemove())
+    update.message.reply_text('Для того, чтобы начать работу с ботом заново напишите /start')
+    return ConversationHandler.END
+
+
+def error(bot, update, error):
+    logger.warning('Update "%s" caused error "%s"', update, error)
+
+
+def main():
+    is_prod = getenv('production', '')
+
+    if is_prod:
+        print('PRODUCTION ENVIRONMENT')
+        updater = Updater(TG_TOKEN)
+    else:
+        print('DEVELOPMENT ENVIRONMENT')
+        updater = Updater(TG_TOKEN, request_kwargs=REQUEST_KWARGS)
+
+    dp = updater.dispatcher
+    dp.add_error_handler(error)
+    dp.add_handler(conversation_handler)
+
+    updater.start_polling()
+    updater.idle()
+
+
+conversation_handler = ConversationHandler(
+    entry_points=[
+        CommandHandler('start', start, pass_user_data=True),
+        MessageHandler(Filters.text, resume_bot, pass_user_data=True),
+    ],
+
+    states={
+        WAIT_FOR_USERNAME: [MessageHandler(Filters.text, username_check, pass_user_data=True)],
+
+        MAIN_MENU: [
+            MessageHandler(Filters.regex(MenuKeyboard.CHOOSE_TASK), States.choose_task, pass_user_data=True),
+            MessageHandler(Filters.regex(MenuKeyboard.TOP_10), States.top_10, pass_user_data=True),
+            MessageHandler(Filters.regex(MenuKeyboard.RULES), States.rules, pass_user_data=True),
+            CommandHandler('admin', AdminStates.admin_panel, pass_user_data=True)
+        ],
+        TASK_CHOOSING: [
+            MessageHandler(Filters.regex(TasksKeyboard.CANCEL), States.main_menu, pass_user_data=True),
+            MessageHandler(Filters.text, States.show_task, pass_user_data=True),
+        ],
+        TASK_SHOWN: [
+            MessageHandler(Filters.regex(TaskChosenKeyboard.CANCEL), States.choose_task, pass_user_data=True),
+            MessageHandler(Filters.text, States.accept_answer, pass_user_data=True),
+            # MessageHandler(Filters.regex(TaskChosenKeyboard.TYPE_ANSWER), States.type_answer, pass_user_data=True),
+        ],
+        ANSWERING: [
+            MessageHandler(Filters.regex(AnsweringKeyboard.CANCEL), States.show_task, pass_user_data=True),
+            MessageHandler(Filters.text, States.accept_answer, pass_user_data=True),
+        ],
+        ANSWER_RIGHT: [MessageHandler(Filters.text, States.main_menu, pass_user_data=True)],
+        ANSWER_WRONG: [MessageHandler(Filters.text, States.show_task, pass_user_data=True)],
+
+        ASKING_QUESTION: [
+            MessageHandler(Filters.regex(BackToMenuKeyboard.CANCEL), States.main_menu, pass_user_data=True),
+            MessageHandler(Filters.text, States.ask_question, pass_user_data=True),
+        ],
+
+        # ADMIN PANEL
+
+        ADMIN_MENU: [
+            MessageHandler(Filters.regex(AdminKeyboard.CANCEL), States.main_menu, pass_user_data=True),
+            MessageHandler(Filters.regex(AdminKeyboard.PUBLISH_TASK), AdminStates.choose_task_publish, pass_user_data=True),
+            MessageHandler(Filters.regex(AdminKeyboard.HIDE_TASK), AdminStates.choose_task_hide, pass_user_data=True),
+            MessageHandler(Filters.regex(AdminKeyboard.ANNOUNCE), AdminStates.wait_for_announcement, pass_user_data=True),
+            MessageHandler(Filters.regex(AdminKeyboard.MESSAGE_PLAYER), AdminStates.wait_for_message, pass_user_data=True),
+        ],
+
+        ADMIN_WAIT_FOR_ANNOUNCEMENT: [
+            MessageHandler(Filters.regex(BackToMenuKeyboard.CANCEL), AdminStates.admin_panel, pass_user_data=True),
+            MessageHandler(Filters.text, AdminStates.announce_message, pass_user_data=True),
+        ],
+        ADMIN_WAIT_FOR_MESSAGE: [
+            MessageHandler(Filters.regex(BackToMenuKeyboard.CANCEL), AdminStates.admin_panel, pass_user_data=True),
+            MessageHandler(Filters.text, AdminStates.message_plr, pass_user_data=True),
+        ],
+
+        ADMIN_TASK_CHOOSE_HIDE: [
+            MessageHandler(Filters.regex(TasksKeyboard.CANCEL), AdminStates.admin_panel, pass_user_data=True),
+            MessageHandler(Filters.text, AdminStates.hide_task, pass_user_data=True),
+        ],
+
+        ADMIN_TASK_CHOOSE_PUBLISH: [
+            MessageHandler(Filters.regex(TasksKeyboard.CANCEL), AdminStates.admin_panel, pass_user_data=True),
+            MessageHandler(Filters.text, AdminStates.publish_task, pass_user_data=True),
+        ],
+
+        ADMIN_TASK_PUBLISHED: [MessageHandler(Filters.text, AdminStates.admin_panel, pass_user_data=True)],
+        ADMIN_ACCESS_DENIED: [MessageHandler(Filters.text, States.main_menu, pass_user_data=True)],
+    },
+
+    fallbacks=[
+        CommandHandler('stop', stop),
+        MessageHandler(Filters.regex(MenuKeyboard.HELP), States.prompt_question, pass_user_data=True),
+    ]
+)
+
+if __name__ == '__main__':
+    main()