KazTron.py

#!/usr/bin/env python
# coding=utf8

import sys
import logging

import kaztron
from kaztron import runner
from kaztron.config import get_kaztron_config
from kaztron.logging import setup_logging

# In the loving memory of my time as a moderator of r/worldbuilding network
# To the future dev, this whole thing is a mess that somehow works. Sorry for the inconvenience.
# (Assuming this is from Kazandaki -- Laogeodritt)

# load configuration
try:
    config = get_kaztron_config(kaztron.cfg_defaults)
except OSError as e:
    print(str(e), file=sys.stderr)
    sys.exit(runner.ErrorCodes.CFG_FILE)


def stop_daemon():
    import os
    import signal
    # noinspection PyPackageRequirements
    from daemon import pidfile
    print("Reading pidfile...")
    pidf = pidfile.TimeoutPIDLockFile(config.get('core', 'daemon_pidfile'))
    pid = pidf.read_pid()
    print("Stopping KazTron (PID={:d})...".format(pid))
    os.kill(pid, signal.SIGINT)
    time.sleep(2)  # time for process to finish - TODO: monitor PID directly
    print("Stopped.")


if __name__ == '__main__':
    import asyncio
    import os
    import signal
    import time

    try:
        cmd = sys.argv[1].lower()
    except IndexError:
        cmd = None

    if cmd == 'start' and config.get('core', 'daemon', False):
        print("Starting KazTron (daemon mode)...")
        with runner.get_daemon_context(config):
            print("Starting KazTron daemon...")
            setup_logging(logging.getLogger(), config, console=False)
            loop = asyncio.get_event_loop()
            runner.run_reboot_loop(loop)

    elif cmd == 'start':  # non-daemon
        print("Starting KazTron (non-daemon mode)...")
        setup_logging(logging.getLogger(), config, console=True)
        loop = asyncio.get_event_loop()
        runner.run_reboot_loop(loop)

    elif cmd == 'debug':
        # override logging levels
        print("Starting in debug mode...")
        setup_logging(logging.getLogger(), config, debug=True)

        # run in console (non-daemon)
        loop = asyncio.get_event_loop()
        runner.run_reboot_loop(loop)

    elif cmd == 'stop':
        if not config.get('core', 'daemon', False):
            print("[ERROR] Cannot stop: daemon mode disabled", file=sys.stderr)
            sys.exit(runner.ErrorCodes.DAEMON_NOT_RUNNING)

        try:
            stop_daemon()
        except TypeError:
            print("[ERROR] Cannot stop: daemon not running", file=sys.stderr)
            sys.exit(runner.ErrorCodes.DAEMON_NOT_RUNNING)

    elif cmd == 'restart':
        if not config.get('core', 'daemon', False):
            print("[ERROR] Cannot restart: daemon mode disabled", file=sys.stderr)
            sys.exit(runner.ErrorCodes.DAEMON_NOT_RUNNING)

        try:
            stop_daemon()
        except TypeError:
            print("[WARNING] Cannot stop: daemon not running", file=sys.stderr)

        print("Starting KazTron (daemon mode)...")
        with runner.get_daemon_context(config):
            print("Starting KazTron daemon...")
            setup_logging(logging.getLogger(), config, console=False)
            loop = asyncio.get_event_loop()
            runner.run_reboot_loop(loop)

    else:
        print("Usage: ./kaztron.py <start|stop|debug|help>\n")