aswi2020sebela-gitlab/modules/crawler/Utilities/ConfigureFunctions.py @ 9540d301

import yaml

# Path to dataset configuration files
CONFIG_FILES_PATH = "DatasetConfigs/"
# Config file type
CONFIG_FILE_TYPE = ".yaml"


def load_configuration(dataset_name):
    """
    Loads yaml configuration file into memory

    Args:
        dataset_name: name of dataset that has existing configuration file

    Returns:
        yaml configuration file as dictionary
    """
    with open(CONFIG_FILES_PATH + dataset_name + CONFIG_FILE_TYPE, "r") as f:
        data = yaml.load(f, Loader=yaml.FullLoader)

    devices_dic = dict()

    if data["devices"] is not None:
        for item in data["devices"]:
            devices_dic.update(item)

    data["devices"] = devices_dic

    return data


def update_configuration(dataset_name, new_devices):
    """
    Open dataset and appends new_devices to the end

    Args:
        dataset_name: name of dataset that has existing configuration file
        new_devices: list or set of new devices for dataset
    """
    with open(CONFIG_FILES_PATH + dataset_name + CONFIG_FILE_TYPE, "a") as file:
        for device in new_devices:
            file.write("  - "+device+":\n")
            file.write("      x: UNKNOWN!\n")
            file.write("      y: UNKNOWN!\n")
            file.write("\n")