Python: создайте вложенный dict из строк пар ключ: значение

Question

У меня есть такой запрос:

{"key1:key2[0]:key3[0]": "1234",
 "key1:key2[0]:key4[0]:key5": "4567",
 "key1:key2[1]:key3[0]": "789",
 "key1:key2[1]:key4[1]:key5": "12345"}

Ключи - это описательный способ представления происхождения каждого окончательного значения в заданном дикте. : отделяет родительский ключ от его дочернего ключа, [] означает, что значение предыдущих ключей является списком, а индекс находится между фигурными скобками.

Учитывая это, как я могу создать диктовку, такую ​​как

{
   "key1":{
      "key2":[
         {
            "key3":["1234"],
            "key4":[{"key5":"4567"}]
         },
         {
            "key3":["789"],
            "key4":[{"key5":"12345"}]
         }
      ]
   }
}

Я пытался сделать что-то вроде этого:

result_dict = {}

def populate(target_path, value):
    current_point_in_path = None
    t = result_dict
    target_path = target_path.split(":")
    for i, each_key in enumerate(target_path):
        list_index = re.findall(r'\[(.*?)\]', each_key)
        if len(list_index) > 1:
            raise Exception("not allowed")
        elif len(list_index) == 1:
            index = int(list_index[0])
            key_before = each_key.split(index)[0]
            if not isinstance(result_dict[key_before], list):
                t = t.setdefault(key_before, [])
                if i+1 == len(target_path):
                    # the issue is that this insert won't return a pointer to the current index element like setdefault would do
                    # alternate soultions are wc
                    t.insert(index, value)
                else:
                    t.insert(index, {})

        else:
            if i + 1 == len(target_path):
                t = t.setdefault(each_key, value)
            else:
                t = t.setdefault(each_key, {})

Я не могу завершить части кода здесь. Возможно, мне понадобится лучший дизайн с моим описательным языком. Любые предложения приветствуются.

Answer 1

Вы можете использовать этого монстра:

def populate(result_dict, target_path, value):
    # split path
    target_path = re.findall(r"[^:]+?(?=\[|:|$)|\[\d+?\]", target_path)
    # prepare path
    for i, element in enumerate(target_path):
        if element[0] == "[" and element[-1] == "]":
            element = int(element[1:-1])
        target_path[i] = element
    current = result_dict
    for i, element in enumerate(target_path[:-1]):
        if isinstance(element, str):  # dict index
            if element not in current:  # create new entry
                if isinstance(target_path[i + 1], str):  # next is a dict
                    current[element] = {}
                else:  # next is a list
                    current[element] = []
        elif isinstance(element, int):  # list index
            if element >= len(current):  # create new entry
                current.extend(None for _ in range(element-len(current)+1))
            if current[element] is None:
                if isinstance(target_path[i + 1], str):  # next is a dict
                    current[element] = {}
                else:  # next is a list
                    current[element] = []
        current = current[element]
    if isinstance(target_path[-1], int):
        current.append(value)
    else:
        current[target_path[-1]] = value

Вы можете использовать этот код для заполнения текста:

result_dict = {}
for key, value in {"key1:key2[0]:key3[0]": "1234",
                   "key1:key2[0]:key4[0]:key5": "4567",
                   "key1:key2[1]:key3[0]": "789",
                   "key1:key2[1]:key4[1]:key5": "12345"}.items():
    populate(result_dict, key, value)
print(json.dumps(result_dict, indent=4))

печать:

{
    "key1": {
        "key2": [
            {
                "key3": [
                    "1234"
                ],
                "key4": [
                    {
                        "key5": "4567"
                    }
                ]
            },
            {
                "key3": [
                    "789"
                ],
                "key4": [
                    null,
                    {
                        "key5": "12345"
                    }
                ]
            }
        ]
    }
}

Answer 2

Вы можете использовать itertools.groupby с рекурсией:

import re, itertools
d = {"key1:key2[0]:key3": "1234", "key1:key2[0]:key4": "4567", "key1:key2[1]:key3": "789", "key1:key2[1]:key4": "12345"}
new_d = [(re.findall('\w+', a), b) for a, b in d.items()]
def last_group(d):
  return [{a[-1]:c for a, c in list(b)} for _, b in itertools.groupby(sorted(d, key=lambda x:x[0][1]), key=lambda x:x[0][1])]

def group_data(d):
   return {a:(lambda x:group_data([(c[1:], d) for c, d in x]) if all(len(c) > 3 for c, _ in x) else last_group(x))(list(b)) for a, b in itertools.groupby(sorted(d, key=lambda x:x[0][0]), key=lambda x:x[0][0])}

print(group_data(new_d))

Выход:

{'key1': {'key2': [{'key3': '1234', 'key4': '4567'}, {'key3': '789', 'key4': '12345'}]}}