handlers.py

import functions
from telegram.ext import (
    CommandHandler,
    ConversationHandler,
    MessageHandler,
    filters,
)

# States for the bot
TYPING_BARCODE, CHOOSING, TYPING_PAPER, TYPING_ISSUE, CHOOSING_CLASS, CHOOSING_OPERATION, CHOOSING_ANALYTIC, CHOOSING_CLASS_ANALYTIC = range(
    8)


conv_handler = ConversationHandler(
    entry_points=[CommandHandler("start", functions.start)],
    states={
        CHOOSING: [
            MessageHandler(
                filters.Regex("^🧾  Get Papers$"), functions.choose_class
            ),
            MessageHandler(
                filters.Regex("^📍 Paper Issues$"), functions.paper_issue
            ),
            MessageHandler(
                filters.Regex("^📊  See Analysis$"), functions.get_analytics
            ),
            MessageHandler(
                filters.Regex("^❌  Close"), functions.done
            ),
            MessageHandler(
                filters.TEXT, functions.invalid_choice_1
            ),
        ],
        TYPING_BARCODE: [
            MessageHandler(
                filters.Regex("^[0-9]{8,10}$"), functions.enter_barcode
            ),
            CommandHandler("close", functions.done),
            MessageHandler(
                filters.TEXT, functions.invalid_input_1
            )
        ],
        TYPING_PAPER: [
            MessageHandler(
                filters.Regex("^[0-9]*$"), functions.enter_paper_no
            ),
            CommandHandler("close", functions.done),
            MessageHandler(
                filters.TEXT, functions.invalid_input_2
            )
        ],
        TYPING_ISSUE: [
            MessageHandler(
                filters.Regex("^❌  Cancel$"), functions.cancel_issue
            ),
            MessageHandler(
                filters.TEXT, functions.enter_paper_issue
            )
        ],
        CHOOSING_CLASS: [
            CommandHandler("close", functions.done),
            MessageHandler(
                filters.TEXT, functions.choose_operation
            )
        ],
        CHOOSING_OPERATION: [
            MessageHandler(
                filters.Regex("^🔖  Get Marks$"), functions.get_marks
            ),
            MessageHandler(
                filters.Regex(
                    "^📝  Get Marked Paper$"), functions.get_papers
            ),
            MessageHandler(
                filters.Regex("^❌  Close"), functions.done
            ),
            MessageHandler(
                filters.TEXT, functions.invalid_choice_2
            ),
        ],
        CHOOSING_ANALYTIC: [
            MessageHandler(
                filters.Regex(
                    "^📊  Get Data Analysis$"), functions.get_data_analytics
            ),
            MessageHandler(
                filters.Regex(
                    "^📈  Get Graph Analysis$"), functions.get_graph_analytics
            ),
            MessageHandler(
                filters.Regex("^❌  Close"), functions.done
            ),
            MessageHandler(
                filters.TEXT, functions.invalid_choice_3
            ),
        ],
        CHOOSING_CLASS_ANALYTIC: [
            MessageHandler(
                filters.TEXT, functions.showAnalytics
            ),
        ]
    },
    fallbacks=[
        MessageHandler(filters.Regex("^❌  Close"), functions.done)],
)