bot.py

"""
The bot will be run from this file. Here the handler functions will be assigned.
"""

import logging
import handlers
import const
import database
from telegram.ext import Updater, CallbackQueryHandler, CommandHandler, MessageHandler, Filters, \
    PreCheckoutQueryHandler, InlineQueryHandler, ChosenInlineResultHandler

from bot_tokens import BOT_TOKEN

# Console logger
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger(__name__)


def stop_bot(updater):
    logger.info("Shutting down...")
    updater.stop()
    logger.info("Done (shutdown)")


def main():
    if BOT_TOKEN == "":
        logger.error("TOKEN not defined. Put your bot token on bot_tokens.py")
        return

    updater = Updater(BOT_TOKEN)
    h = updater.dispatcher.add_handler
    const.aux.BOT_USERNAME = updater.bot.get_me().username

    # Assigning handlers
    h(CommandHandler("start", handlers.start, pass_args=True))
    h(CommandHandler("help", handlers.help))
    h(CommandHandler("more", handlers.more))
    h(CommandHandler("ping", handlers.ping))
    h(CommandHandler("donate", handlers.donate, pass_user_data=True))
    h(CommandHandler("support", handlers.support, filters=Filters.private))
    h(CommandHandler("support", handlers.support_group, filters=Filters.group))
    h(CommandHandler("about", handlers.about))

    h(CallbackQueryHandler(handlers.change_donation_quantity, pattern=r"don\*", pass_user_data=True))
    h(CallbackQueryHandler(handlers.send_donation_receipt, pattern=r"donate$", pass_user_data=True))
    h(MessageHandler(filters=Filters.successful_payment, callback=handlers.completed_donation))
    h(PreCheckoutQueryHandler(handlers.approve_transaction))

    h(InlineQueryHandler(handlers.full_query, pattern=".+"))
    h(InlineQueryHandler(handlers.empty_query))
    h(ChosenInlineResultHandler(handlers.chosen_result))
    h(CallbackQueryHandler(handlers.cast_vote, pattern=r"vote\*.*\*"))
    h(CallbackQueryHandler(handlers.get_clist, pattern=r"get_clist\*.*\*"))

    updater.dispatcher.add_error_handler(handlers.error)

    updater.start_polling()

    # CONSOLE
    while True:
        inp = input("")
        if inp:
            input_c = inp.split()[0]
            args = inp.split()[1:]
            strig = ""
            for e in args:
                strig = strig + " " + e

            if input_c == "stop":
                stop_bot(updater)
                break

            else:
                print("Unknown command")


if __name__ == '__main__':
    main()