Nicolas Noble | ddef246 | 2015-01-06 18:08:25 -0800 | [diff] [blame] | 1 | """Allows dot-accessible dictionaries.""" |
| 2 | |
| 3 | |
| 4 | class Bunch(dict): |
| 5 | |
| 6 | def __init__(self, d): |
| 7 | dict.__init__(self, d) |
| 8 | self.__dict__.update(d) |
| 9 | |
| 10 | |
| 11 | # Converts any kind of variable to a Bunch |
| 12 | def to_bunch(var): |
| 13 | if isinstance(var, list): |
| 14 | return [to_bunch(i) for i in var] |
| 15 | if isinstance(var, dict): |
| 16 | ret = {} |
| 17 | for k, v in var.items(): |
| 18 | if isinstance(v, (list, dict)): |
| 19 | v = to_bunch(v) |
| 20 | ret[k] = v |
| 21 | return Bunch(ret) |
| 22 | else: |
| 23 | return var |
| 24 | |
| 25 | |
| 26 | # Merges JSON 'add' into JSON 'dst' |
| 27 | def merge_json(dst, add): |
| 28 | if isinstance(dst, dict) and isinstance(add, dict): |
| 29 | for k, v in add.items(): |
| 30 | if k in dst: |
| 31 | merge_json(dst[k], v) |
| 32 | else: |
| 33 | dst[k] = v |
| 34 | elif isinstance(dst, list) and isinstance(add, list): |
| 35 | dst.extend(add) |
| 36 | else: |
| 37 | raise Exception('Tried to merge incompatible objects %r, %r' % (dst, add)) |