用递归新增或修改嵌套字典

需求

最近有个需求,需要在字典中修改或添加多级嵌套的键值对,如在dic1中:

dic1 = {
		'test1': '1',
        'test2': 2,
        'test3': {'test3_1': 3},
        'test4': {'test4_1': {'test4_1_1': 4, 'test4_1_2': 4}}
        }

添加
'test5': {'test5_1': {'test5_1_1': 5, 'test5_1_2': 5}}
并修改'test3': {'test3_1': 3}为'test3': {'test3_1': 33}

针对此需求,封装了一个方法,以"."为分隔符,多层传递键值对,如上述添加为 test5.test5_1.test5_1_1=5,上述修改为test3.test3_1=33,类似这样的方式

代码如下

def deepset(dic, keypath, value, ignkeyerror=False):
	"""
	:param dic: 需要修改的字典
	:type dic: dict
	:param keypath: 多级嵌套内容,以.为分隔符
	:type keypath: str
	:param value: 最内层的值
	:type value: Any
	:param ignkeyerror:是否忽略健错误类型,True时为忽略,用于新增;False时为不忽略,用于修改
	:type ignkeyerror: bool
	"""
    if ignkeyerror: #如果是新增
        keys = keypath.split('.')
        for k in keys:
            if k == keys[-1]: #递归出口,如果key已经到了最后一个了,则赋值他的value是传进来的那个值
                dic[k] = value
            else:
                keypath = '.'.join(keys[-len(keys) + 1:]) #每次循环,传入的路径减少一个.分割的内容
                try:
                    dic[k] = deepset(dic[k], keypath, value, ignkeyerror=True) #新进入递归的dic为这一层字典的值
                except KeyError:
                    dic[k] = deepset({}, keypath, value, ignkeyerror=True) #如果键不存在,说明这一层原本没有健,就给下一层传一个空字典
            return dic
    else: #如果是修改
        keys = keypath.split('.')
        for k in keys[:-1]:
            dic = dic[k]
        dic[keys[-1]] = value
        return dic
        
deepset(dic1, 'test3.test3_1', 33)
print(dic1) #{'test1': '1', 'test2': 2, 'test3': {'test3_1': 33}, 'test4': {'test4_1': {'test4_1_1': 4, 'test4_1_2': 4}}}

deepset(dic1, 'test5.test5_1.test5_1_1', 5, ignkeyerror=True)
deepset(dic1, 'test5.test5_1.test5_1_2', 5, ignkeyerror=True)
print(dic1) #{'test1': '1', 'test2': 2, 'test3': {'test3_1': 33}, 'test4': {'test4_1': {'test4_1_1': 4, 'test4_1_2': 4}}, 'test5': {'test5_1': {'test5_1_1': 5, 'test5_1_2': 5}}}