From 9d94a01b631676b54f680c2571ae4dbba9ccd037 Mon Sep 17 00:00:00 2001 From: jantuomi Date: Tue, 19 Jul 2016 14:50:22 +0300 Subject: Restructure project --- radiodiodibot | 97 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 97 insertions(+) create mode 100755 radiodiodibot (limited to 'radiodiodibot') diff --git a/radiodiodibot b/radiodiodibot new file mode 100755 index 0000000..4bd53e0 --- /dev/null +++ b/radiodiodibot @@ -0,0 +1,97 @@ +#!/usr/bin/env python3 +import argparse +import configparser +import logging +import signal +import sys + +from project import botmanager + + +def sigint_handler(signal, frame): + """Exit gracefully when receiving an interrupt signal""" + print("Exiting src...") + sys.exit(0) + +signal.signal(signal.SIGINT, sigint_handler) + +try: # Python 2.7+ + from logging import NullHandler +except ImportError: + class NullHandler(logging.Handler): + def emit(self, record): + pass + +logging.getLogger(__name__).addHandler(NullHandler()) + +# Read configs, unsuccessful reads are +# silently ignored +CONFIG_FILE = "bot.config" +config = configparser.ConfigParser() +config.read(CONFIG_FILE) + +# Get the bot token as an argument from the user +parser = argparse.ArgumentParser(description="Telegram bot for the Radiodiodi Student Radio broadcast.") +parser.add_argument("-t", "--token", help="Telegram Bot API token") +parser.add_argument("-U", "--shoutbox-api-url", help="Shoutbox API URL") +parser.add_argument("-C", "--telegram-chat-id", help="Telegram Chat ID") +parser.add_argument("-v", "--verbose", help="Verbose output", action="store_true") +parser.add_argument("-i", "--interval", help="Delay between API update calls in seconds", type=int, default=10) +args = parser.parse_args() + +# Show verbose output with the -v option +if args.verbose: + logging_level = logging.INFO +else: + logging_level = logging.WARNING + +logging.basicConfig(level=logging_level, format="%(asctime)s %(levelname)s %(message)s") + +# Try to import telepot +# If not installed, inform user and fail gracefully +try: + import telepot +except: + print("src needs the telepot module to communicate with Telegram.") + print("Please install telepot before using src.") + sys.exit(1) + + +# Entry point +def main(): + print("=== Radiodiodibot ===") + print("Press CTRL-C to exit.") + # Get default parameter values from the config file + telegram_bot_token = config["GENERAL"]["TelegramBotToken"] + shoutbox_api_url = config["GENERAL"]["ShoutboxApiUrl"] + telegram_chat_id = config["GENERAL"]["TelegramChatID"] + api_call_interval = config["GENERAL"]["ApiCallInterval"] + + # If the user has specified the parameters as command line + # arguments, use them to override the config values + if args.token is not None: + telegram_bot_token = args.token + + if args.shoutbox_api_url is not None: + shoutbox_api_url = args.shoutbox_api_url + + if args.telegram_chat_id is not None: + telegram_chat_id = args.telegram_chat_id + + if args.interval is not None: + api_call_interval = args.interval + + # Store the final values in a manager instance + manager = botmanager.BotManager(telegram_bot_token) + manager.set_parameters(shoutbox_api_url, telegram_chat_id, api_call_interval) + + logging.info("Using Shoutbox API URL: {}".format(shoutbox_api_url)) + logging.info("Using Telegram Chat ID: {}".format(telegram_chat_id)) + logging.info("Using API call interval of {} seconds.".format(api_call_interval)) + + # Start listening + manager.start() + + +if __name__ == "__main__": + main() -- cgit v1.3