Argparse Telegram bot

argparse_bot.py

import argparse
from telegram.ext import Updater
from telegram.ext.commandhandler import CommandHandler

import logging

# Enable logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)

logger = logging.getLogger(__name__)


class ArgumentParser(argparse.ArgumentParser):
    """
    The default (and only) argparse behavior is to throw error to sys.stderr via the error method.
    """

    def error(self, message):
        raise ParserError(message)


class ParserError(Exception): pass


# Example of standard usage
command_1_parser = ArgumentParser(description='Command 1 Parser')
command_1_parser.add_argument('-f', '--foo')
command_1_parser.add_argument('-b', '--bar', required=True)

# Using subparsers, as in postional arguments, provides a more natural way to use this via a chat
command_2_parser = ArgumentParser(description='Command 2 Parser')
subparsers = command_2_parser.add_subparsers()

foo_parser = subparsers.add_parser('foo')
foo_parser.add_argument('-b')


def command_1(bot, update, args):
    try:
        options = command_1_parser.parse_args(args)
        message = repr(options)
    except ParserError as e:
        message = 'ERROR!: %s' % str(e)
    bot.sendMessage(update.message.chat_id, text=message)


def command_2(bot, update, args):
    try:
        options = command_2_parser.parse_args(args)
        message = repr(options)
    except ParserError as e:
        message = 'ERROR!: %s' % str(e)
    bot.sendMessage(update.message.chat_id, text=message)


def error(bot, update, error):
    logger.warn('Update "%s" caused error "%s"' % (update, error))


def main():
    # Create the EventHandler and pass it your bot's token.
    updater = Updater("TOKEN")

    # Get the dispatcher to register handlers
    dp = updater.dispatcher

    dp.add_handler(CommandHandler('command1', command_1, pass_args=True))
    dp.add_handler(CommandHandler('command2', command_2, pass_args=True))

    dp.add_error_handler(error)

    # Start the Bot
    updater.start_polling()

    # Run the bot until the you presses Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT. This should be used most of the time, since
    # start_polling() is non-blocking and will stop the bot gracefully.
    updater.idle()


if __name__ == '__main__':
    main()