heart_of_bot.py

import logging
import asyncio
import argparse


logger = logging.getLogger('nojam')
logging.basicConfig(level=logging.INFO)


if __name__ == '__main__':
    # Use argument parser
    parser = argparse.ArgumentParser(
        description='This is a script to run nojambot'
    )
    parser.add_argument(
        'mode',
        help='Choose whether to run on Slack or Twitter',
        choices=['twitter', 'slack'],
        type=str
    )

    args = parser.parse_args()

    if args.mode == 'twitter':
        logger.info('Run on Twitter')

        from twitter_bot import Twitter

        twt = Twitter()
        try:
            twt.execute_bot()
        except KeyboardInterrupt:
            logger.info('Terminate nojambot on Twitter')

    else:
        logger.info('Run on Slack')

        from slack_bot import Slack

        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)

        slk = Slack()
        try:
            asyncio.get_event_loop().run_until_complete(slk.execute_bot())
            asyncio.get_event_loop().run_forever()
        except KeyboardInterrupt:
            logger.info('Terminate nojambot on Slack')