postprocess_merge_duplicate_nodes.py

'''
Merge nodes that have identical paths by adding conditional statments to the attribute descriptions.
'''
import re
import sys
from collections import Counter

from dicom_standard import parse_lib as pl

DUPLICATE_PATH_EXCEPTIONS = ['enhanced-mr-image:300a012c']


def add_conditional_to_description(node):
    conditional = node.get('conditional')
    assert conditional is not None, f'Duplicate attribute (path: {node["path"]}) has no conditional statement.'
    conditional = re.sub(r'\.$', ':', conditional)
    formatted_conditional = f'<p style="font-weight: bold">{conditional[0].upper()}{conditional[1:]}</p>'
    node['description'] = formatted_conditional + node['description']


def is_duplicate_node(path, node_list):
    instances = filter(lambda n: n['path'] == path, node_list)
    descriptions = map(lambda n: n['description'], instances)
    return len(set(descriptions)) > 1


def merge_duplicate_nodes(node_list):
    path_list = [d['path'] for d in node_list]
    duplicate_paths = [k for k, v in Counter(path_list).items() if v > 1]
    path_to_node = {}
    for node in node_list:
        path = node['path']
        if path in path_to_node:
            if path not in DUPLICATE_PATH_EXCEPTIONS:
                # Add conditional to description only if the duplicates do not have identical descriptions
                if is_duplicate_node(path, node_list):
                    add_conditional_to_description(node)
                    path_to_node[path]['description'] += node['description']
                    path_to_node[path]['externalReferences'].extend(node['externalReferences'])
        else:
            if path in duplicate_paths and path not in DUPLICATE_PATH_EXCEPTIONS:
                # Add conditional to description only if the duplicates do not have identical descriptions
                if is_duplicate_node(path, node_list):
                    add_conditional_to_description(node)
            path_to_node[path] = node
        path_to_node[path].pop('conditional', None)
    return list(path_to_node.values())


if __name__ == "__main__":
    node_list = pl.read_json_data(sys.argv[1])
    processed_node_list = merge_duplicate_nodes(node_list)
    pl.write_pretty_json(processed_node_list)