55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
"""Command line interface for hbd package."""
|
|
|
|
import argparse
|
|
|
|
from .config import load_config
|
|
from .server import run as run_server
|
|
|
|
PUSHSRVS = ["all", "pushover", "mattermost"]
|
|
|
|
|
|
def build_parser():
|
|
parser = argparse.ArgumentParser(
|
|
prog="hbd",
|
|
description="HeartBeatDaemon - Wait for heartbeat messages and act on them (or their absence)",
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
parser.add_argument(
|
|
"-c", "--config", dest="configfile", help="Config file path (YAML)"
|
|
)
|
|
parser.add_argument(
|
|
"-f", "--foreground", action="store_true", help="Run in foreground"
|
|
)
|
|
parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
|
|
parser.add_argument(
|
|
"-p", "--pushsrv", dest="pushsrv", choices=PUSHSRVS, help="Push service to use"
|
|
)
|
|
parser.add_argument(
|
|
"-x", "--debug", action="count", default=0, help="Increase debug level"
|
|
)
|
|
return parser
|
|
|
|
|
|
def main(argv=None):
|
|
parser = build_parser()
|
|
args = parser.parse_args(argv)
|
|
|
|
config = load_config(args.configfile)
|
|
|
|
# Apply CLI overrides
|
|
if args.foreground:
|
|
config["foreground"] = True
|
|
if args.verbose:
|
|
config["verbose"] = True
|
|
if args.pushsrv:
|
|
config["pushsrv"] = args.pushsrv
|
|
if args.debug:
|
|
config.setdefault("debug", 0)
|
|
config["debug"] += args.debug
|
|
|
|
run_server(config)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|