Prune dict for dataclass

prune_dict_for_dataclass.py

def prune_dict_for_dataclass(d: dict, cls):
    ret = {}
    for field in fields(cls):
        if field.name in d:
            if is_dataclass(field.type):
                ret[field.name] = prune_dict_for_dataclass(d[field.name], field.type)
            elif typing.get_origin(field.type) is list and type(d[field.name]) is list:
                list_cls = typing.get_args(field.type)[0]
                if is_dataclass(list_cls):
                    ret[field.name] = [prune_dict_for_dataclass(i, list_cls) for i in d[field.name]]
                else:
                    ret[field.name] = d[field.name]
            else:
                ret[field.name] = d[field.name]

    return ret