Skip to content

Configuration

Configuration settings are explained here.

sleepecg.get_config(key=None)

Read SleepECG preferences from the configuration file.

For parameters not set in the user configuration file (~/.sleepecg/config.yml), this falls back to the default values defined in site-packages/sleepecg/config.yml. See configuration for a list of possible settings.

Parameters:

  • key (str) –

    The configuration key to look for. If None, all configuration settings are returned in a dictionary, by default None.

Returns:

  • Any –

    The configuration value.

Source code in sleepecg/config.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
def get_config(key: Optional[str] = None) -> Any:
    """
    Read SleepECG preferences from the configuration file.

    For parameters not set in the user configuration file (`~/.sleepecg/config.yml`), this
    falls back to the default values defined in `site-packages/sleepecg/config.yml`. See
    [configuration](../configuration.md) for a list of possible settings.

    Parameters
    ----------
    key : str, optional
        The configuration key to look for. If `None`, all configuration settings are
        returned in a dictionary, by default `None`.

    Returns
    -------
    typing.Any
        The configuration value.
    """
    config = _read_yaml(_DEFAULT_CONFIG_PATH)
    user_config = _read_yaml(_USER_CONFIG_PATH)
    if invalid_keys := set(user_config) - set(config):
        raise ValueError(
            f"Invalid key(s) found in user config at {_USER_CONFIG_PATH}: {invalid_keys}"
        )
    config.update(user_config)

    if key is None:
        return config
    if key not in config:
        options = ", ".join(config)
        raise ValueError(
            f"Trying to get invalid config key: {key!r}, possible options: {options}"
        )
    return config[key]

sleepecg.set_config(**kwargs)

Set SleepECG preferences and store them to the user configuration file.

If a value is None, the corresponding key is deleted from the user configuration. See configuration for a list of possible settings.

Parameters:

  • **kwargs (Any) –

    The configuration keys and values to set.

Examples:

>>> set_config(data_dir='~/.sleepecg/datasets')
Source code in sleepecg/config.py 
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
def set_config(**kwargs: Any) -> None:
    """
    Set SleepECG preferences and store them to the user configuration file.

    If a value is `None`, the corresponding key is deleted from the user configuration. See
    [configuration](../configuration.md) for a list of possible settings.

    Parameters
    ----------
    **kwargs: dict, optional
        The configuration keys and values to set.

    Examples
    --------
    >>> set_config(data_dir='~/.sleepecg/datasets')
    """
    default_config = _read_yaml(_DEFAULT_CONFIG_PATH)
    user_config = _read_yaml(_USER_CONFIG_PATH)

    # validate all parameters before setting anything
    for key, value in kwargs.items():
        if key not in default_config:
            options = ", ".join(default_config)
            raise ValueError(
                f"Trying to set invalid config key: {key!r}, possible options: {options}"
            )

    for key, value in kwargs.items():
        if value is None:
            user_config.pop(key, None)
        else:
            user_config[key] = value

    _USER_CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
    with open(_USER_CONFIG_PATH, "w") as user_config_file:
        yaml.dump(user_config, user_config_file)