dict reformulator

reformulator.py

from typing import Dict, List
from itertools import chain

to_convert = {
    "someCommonKey": 123,
    "listOne": ["foo", "bar", "baz"],
    "listTwo": ["lorem", "ipsum", "doler"],
    "listThree": ["jorem", "jpsum", "joler"],
}
ignore = ["someCommonKey"]


def reformulator(source_dict: Dict, ignore_keys: List[str]) -> Dict:
    if len(set(ignore_keys)) != len(ignore_keys):
        raise ValueError("Duplicate Keys are not allowed")

    key_count = len(set(to_convert.keys() - ignore_keys))
    intermediate_res = list(
        chain.from_iterable(
            [
                list(zip([k for _ in range(0, len(v))], v))
                for k, v in source_dict.items()
                if k not in ignore
            ]
        )
    )
    final_result = {ik: source_dict[ik] for ik in ignore_keys}
    final_result["children"] = [
        dict(intermediate_res[n::key_count]) for n in range(0, key_count)
    ]

    return final_result


expected_result = {
    "someCommonKey": 123,
    "children": [
        {"listOne": "foo", "listTwo": "lorem", "listThree": "jorem"},
        {"listOne": "bar", "listTwo": "ipsum", "listThree": "jpsum"},
        {"listOne": "baz", "listTwo": "doler", "listThree": "joler"},
    ],
}


assert expected_result == reformulator(to_convert, ignore)