summaryrefslogtreecommitdiffstats
path: root/src/pybind/mgr/dashboard/tests/helper.py
blob: 67dd09fdfe1dad29adb80c94886c7b67d3c8d793 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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
# -*- coding: utf-8 -*-

try:
    from typing import Any, Dict
except ImportError:
    pass


def update_dict(data, update_data):
    # type: (Dict[Any, Any], Dict[Any, Any]) -> Dict[Any]
    """ Update a dictionary recursively.

    Eases doing so by providing the option to separate the key to be updated by dot characters.  If
    a key provided does not exist, it will raise an KeyError instead of just updating the
    dictionary.

    Limitations

    Please note that the functionality provided by this method can only be used if the dictionary to
    be updated (`data`) does not contain dot characters in its keys.

    :raises KeyError:

    >>> update_dict({'foo': {'bar': 5}}, {'foo.bar': 10})
    {'foo': {'bar': 10}}

    >>> update_dict({'foo': {'bar': 5}}, {'xyz': 10})
    Traceback (most recent call last):
        ...
    KeyError: 'xyz'

    >>> update_dict({'foo': {'bar': 5}}, {'foo.xyz': 10})
    Traceback (most recent call last):
        ...
    KeyError: 'xyz'
    """
    for k, v in update_data.items():
        keys = k.split('.')
        element = None
        for i, key in enumerate(keys):
            last = False
            if len(keys) == i + 1:
                last = True

            if not element:
                element = data[key]
            elif not last:
                element = element[key]  # pylint: disable=unsubscriptable-object

            if last:
                if key not in element:
                    raise KeyError(key)

                element[key] = v
    return data