ihkapy.fileio.options_io

 1# utility methods for loading options from Options.toml
 2import os
 3import toml                     # Parameters/config file Options.toml
 4import warnings
 5
 6
 7def load_fio_ops_and_data_ops(options_path="./Options.toml"):
 8    """Load the 'fio_ops' and 'data_ops' dictionaries from options config file
 9
10    Parses the .toml config file at `options_path` into a dictionary. 
11    Returns the sub-dictionaries at the files "fio" and "params.data"
12
13    Parameters
14    ----------
15    `options_path : str`
16        Path to the config file
17
18    Returns
19    -------
20    `dict`
21        A dictionary containing fio params, i.e. paths to data files.
22
23    `dict`
24        A dictionary containing data parameters required for data manip.
25    """
26
27    warnings.warn("Change this relative path once package is configured properly. \nWe need a more reliable way of accessing the options.toml config file")
28    ops = load_ops_as_dict(options_path)
29    fio_ops = ops["fio"]
30    data_ops = ops["params"]["data"]
31    return fio_ops,data_ops
32
33def load_ops_as_dict(options_path="./Options.toml"):
34    """Returns the Options.toml file as a dictionary
35
36    Parses the .toml config file at `options_path` into a dictionary
37    and returns it.
38
39    Parameters
40    ----------
41    `options_path : str`
42        Path to the config file
43    """
44    with open(options_path,"r") as f:
45        ops_string = f.read()
46    ops = toml.loads(ops_string)
47    return ops