85 lines
2.0 KiB
Python
85 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from benedict.utils import type_util
|
|
|
|
|
|
def _get_index(key):
|
|
if type_util.is_integer(key):
|
|
return key
|
|
return None
|
|
|
|
|
|
def _get_item_value(item, key):
|
|
if type_util.is_list(item):
|
|
index = _get_index(key)
|
|
if index is not None:
|
|
return item[index]
|
|
elif type_util.is_dict(item):
|
|
return item[key]
|
|
raise KeyError
|
|
|
|
|
|
def _new_item_value(key):
|
|
index = _get_index(key)
|
|
return {} if index is None else []
|
|
|
|
|
|
def get_item(d, keys):
|
|
items = get_items(d, keys)
|
|
return items[-1] if items else (None, None, None, )
|
|
|
|
|
|
def get_items(d, keys):
|
|
items = []
|
|
item = d
|
|
value = None
|
|
for key in keys:
|
|
try:
|
|
value = _get_item_value(item, key)
|
|
if type_util.is_list(item):
|
|
index = _get_index(key)
|
|
items.append((item, index, value, ))
|
|
else:
|
|
items.append((item, key, value, ))
|
|
item = value
|
|
except (IndexError, KeyError, ):
|
|
items.append((None, None, None, ))
|
|
break
|
|
return items
|
|
|
|
|
|
def set_item_value(item, key, value):
|
|
index = _get_index(key)
|
|
if index is not None:
|
|
try:
|
|
# overwrite existing index
|
|
item[index] = value
|
|
except IndexError:
|
|
# insert index
|
|
item += ([None] * (index - len(item)))
|
|
item.insert(index, value)
|
|
else:
|
|
item[key] = value
|
|
|
|
|
|
def set_item(d, keys, value):
|
|
item = d
|
|
i = 0
|
|
j = len(keys)
|
|
while i < j:
|
|
key = keys[i]
|
|
if i < (j - 1):
|
|
try:
|
|
subitem = _get_item_value(item, key)
|
|
if not type_util.is_dict_or_list(subitem):
|
|
raise TypeError
|
|
except (IndexError, KeyError, TypeError, ):
|
|
subkey = keys[i + 1]
|
|
subitem = item[key] = _new_item_value(subkey)
|
|
item = subitem
|
|
i += 1
|
|
continue
|
|
set_item_value(item, key, value)
|
|
break
|
|
|