Created
February 12, 2019 12:45
-
-
Save Yamp/928861974c59ab29c4235e5698bfe6ed to your computer and use it in GitHub Desktop.
d_dict.a.b.c.d = 1234
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from collections import defaultdict | |
class DeepDict(defaultdict): | |
def __init__(self, d=None): | |
defaultdict.__init__(self, DeepDict) | |
if d is not None: | |
self.from_dict(d) | |
def __getattr__(self, key): | |
return self[key] | |
def __setattr__(self, key, value): | |
self[key] = value | |
def from_dict(self, d: dict): | |
for k, v in d.items(): | |
self[k] = DeepDict() | |
if not type(v) == dict: | |
self[k] = v | |
else: | |
self[k] = DeepDict(v) | |
# self[k] = self[k].from_dict(v) | |
return self | |
def merge(self, target): | |
for k, v in target.items(): | |
if isinstance(target[k], DeepDict): | |
if isinstance(self[k], DeepDict): | |
self[k].merge(target[k]) | |
else: | |
self[k] = DeepDict() | |
self.merge(target[k]) | |
else: | |
self[k] = target[k] | |
def clear_empty(self): | |
for k, v in list(self.items()): | |
if isinstance(v, DeepDict): | |
v.clear_empty() | |
if len(v) == 0: | |
del self[k] | |
return self | |
def get_dict(self): | |
res = {} | |
for k, v in self.items(): | |
if type(v) == DeepDict: | |
res[k] = v.get_dict() | |
else: | |
res[k] = v | |
return res |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment