app.py

from twisted.logger import FileLogObserver, FilteringLogObserver, globalLogPublisher, InvalidLogLevelError, \
    Logger, LogLevel, LogLevelFilterPredicate
from twisted.python.logfile import DailyLogFile
from heufybot.bot import HeufyBot
from heufybot.utils.logutils import consoleLogObserver, logFormat
from signal import signal, SIGINT
import argparse


if __name__ == "__main__":
    # Parse the command line arguments
    parser = argparse.ArgumentParser(description="A modular Twisted IRC bot.")
    parser.add_argument("-c", "--config", help="The configuration file to use", type=str, default="heufybot.yaml")
    parser.add_argument("-f", "--logfile", help="The file the log will be written to", type=str, default="heufybot.log")
    parser.add_argument("-l", "--loglevel", help="The logging level the bot will use", type=str, default="INFO")
    options = parser.parse_args()

    # Start the bot
    heufybot = HeufyBot(options.config)

    # Determine the logging level
    logFilter = LogLevelFilterPredicate()
    try:
        logFilter.setLogLevelForNamespace("heufybot", LogLevel.levelWithName(options.loglevel.lower()))
        invalidLogLevel = False
    except InvalidLogLevelError:
        logFilter.setLogLevelForNamespace("heufybot", LogLevel.info)
        invalidLogLevel = True

    # Set up logging
    logFile = DailyLogFile("heufybot.log", "")
    fileObserver = FileLogObserver(logFile, logFormat)
    fileFilterObserver = FilteringLogObserver(fileObserver, (logFilter,))
    consoleFilterObserver = FilteringLogObserver(consoleLogObserver, (logFilter,))
    heufybot.log = Logger("heufybot")
    globalLogPublisher.addObserver(fileFilterObserver)
    globalLogPublisher.addObserver(consoleFilterObserver)

    heufybot.log.info("Starting bot...")

    # Yell at the user if they specified an invalid log level
    if invalidLogLevel:
        heufybot.log.warn("Picked up invalid log level {invalidLevel}, level has been set to INFO instead.",
                          invalidLevel=options.loglevel.lower())

    signal(SIGINT, lambda signal, stack: heufybot.shutdown())
    heufybot.startup()