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
|