Restructure project

1 files changed, 97 insertions, 0 deletions

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()