matrix-webhook/matrix_webhook/conf.py

"""Configuration for Matrix Webhook."""
import argparse
import os
import sys
import yaml


def get_numeric_log_level(log_level):
    """Return a number that will calculate to the verbosity level"""
    if log_level.lower() == "debug":
        return 4
    elif log_level.lower() == "info":
        return 3
    elif log_level.lower() == "warning":
        return 2
    elif log_level.lower() == "error":
        return 1
    elif log_level.lower() == "critical":
        return 0
    else:
        return 2


parser = argparse.ArgumentParser(description=__doc__, prog="python -m matrix_webhook")
parser.add_argument(
    "-H",
    "--host",
    default=os.environ.get("HOST", ""),
    help="host to listen to. Default: `''`. Environment variable: `HOST`",
)
parser.add_argument(
    "-P",
    "--port",
    type=int,
    default=os.environ.get("PORT", 4785),
    help="port to listed to. Default: 4785. Environment variable: `PORT`",
)
parser.add_argument(
    "-u",
    "--matrix-url",
    default=os.environ.get("MATRIX_URL", "https://matrix.org"),
    help="matrix homeserver url. Default: `https://matrix.org`. "
    "Environment variable: `MATRIX_URL`",
)
parser.add_argument(
    "-i",
    "--matrix-id",
    help="matrix user-id. Required. Environment variable: `MATRIX_ID`",
)
parser.add_argument(
    "-p",
    "--matrix-pw",
    help="matrix password. Required. Environment variable: `MATRIX_PW`",
)
parser.add_argument(
    "-k",
    "--api-keys",
    help="comma separated list of shared secrets to use this service. Required. Environment variable: `API_KEYS`",
)
parser.add_argument(
    "-c",
    "--config",
    help="configuration file. Default: `config.yaml`",
)

parser.add_argument(
    "-v", "--verbose", action="count", default=0, help="increment verbosity level"
)

args = parser.parse_args()

if args.config:
    with open(args.config) as f:
        config = yaml.safe_load(f)
    SERVER_ADDRESS = (config["host"], config["port"])
    MATRIX_URL = config["matrix"]["url"]
    MATRIX_ID = config["matrix"]["id"]
    MATRIX_PW = config["matrix"]["pw"]
    API_KEYS = config["api_keys"].keys()
    ROOM_KEYS = config["api_keys"]
    VERBOSE = get_numeric_log_level(config["log"]["level"])
else:
    SERVER_ADDRESS = (args.host, args.port)
    MATRIX_URL = args.matrix_url
    if not args.matrix_id:
        print("Missing matrix user-id. Use -i or --matrix-id or specify in config.yaml")
        sys.exit(1)
    else:
        MATRIX_ID = args.matrix_id

    if not args.matrix_pw:
        print(
            "Missing matrix password. Use -p or --matrix-pw or specify in config.yaml"
        )
        sys.exit(1)
    else:
        MATRIX_PW = args.matrix_pw

    if not args.api_keys:
        print("Missing api keys. Use -k or --api-keys or specify in config.yaml")
        sys.exit(1)
    else:
        API_KEYS = args.api_keys.split(",")
    VERBOSE = args.verbose