Source code for config

#!/usr/bin/env python3
"""
Settings manager for the data tracker.

Read settings from `./config.yaml`, `../config.yaml` or from the provided path.
"""

import logging
import os
import sys

import yaml


def read_config(path: str = ""):
    """
    Look for settings.yaml and parse the settings from there.

    The file is expected to be found in the current, parent or provided folder.

    Args:
        path (str): The yaml file to use

    Returns:
        dict: The loaded settings

    Raises:
        FileNotFoundError: No settings file found

    """
    file_locations = [os.getcwd(), os.pardir]
    if not path:
        for location in file_locations:
            fpath = os.path.join(location, "config.yaml")
            if os.path.exists(fpath):
                path = fpath
                break

    with open(path, "r") as in_file:
        return yaml.load(in_file, Loader=yaml.FullLoader)


def init() -> dict:
    """
    Read the config from a config.yaml file.

    Returns:
        dict: The config.
    """
    config_file = ""
    arg = "--config_file"
    if arg in sys.argv:
        try:
            config_file = sys.argv[sys.argv.index(arg) + 1]
        except IndexError:
            logging.error("No argument for --config_file")
            sys.exit(1)

    config = read_config(config_file)
    if config["dev_mode"]["testing"]:
        logging.getLogger().setLevel(logging.DEBUG)
        config["DEBUG"] = True
        config["TESTING"] = True
        config["ENV"] = "development"

    if config.get("oidc"):
        for oidc_entry in config["oidc"]:
            base_name = oidc_entry.upper()
            for conf_part in config["oidc"][oidc_entry]:
                config[f"{base_name}_{conf_part.upper()}"] = config["oidc"][oidc_entry][conf_part]
    config["oidc_names"] = config["oidc"].keys()
    del config["oidc"]

    config["SESSION_COOKIE_NAME"] = "dt_session"
    config["SECRET_KEY"] = config["flask"]["secret"]
    config["SESSION_COOKIE_SAMESITE"] = "Lax"
    return config