Skip to content

reformat_json

format_sort_round_json(current_filename, new_filename)

A function that takes a JSON file and reformats it to be sorted and rounded to 9 digits. This is helpful when comparing the new and old models. The function will create a new file.

Parameters:

Name Type Description Default
current_filename str or Path

The name of the JSON file to be converted.

 required
new_filename str or Path

The name of the new JSON file to be created.

 required
Source code in emodpy_hiv/countries/converting/reformat_json.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
def format_sort_round_json(current_filename: Union[str, Path],
                           new_filename: Union[str, Path]):
    """
    A function that takes a JSON file and reformats it to be sorted and rounded to 9 digits.
    This is helpful when comparing the new and old models. The function will create a new file.

    Args:
        current_filename (str or Path):
            The name of the JSON file to be converted.

        new_filename (str or Path):
            The name of the new JSON file to be created.
    """
    if os.path.exists(new_filename):
        ValueError(f"File {new_filename} already exists. Please delete it before running this function.")
    if not os.path.exists(current_filename):
        raise ValueError(f"File {current_filename} does not exist. Please check the file path.")

    tmp_json = None
    with open(current_filename, 'r') as file:
        tmp_json = json.load(file)

    with open(new_filename, "w") as file:
        # This process of writing and reading and writing allows us to write a file where
        # the floating point values are rounded to 9 digits. This helps us not have rounding
        # issues between different platforms.
        tmp_json = json.dumps(tmp_json, indent=4, sort_keys=True)
        tmp_json = json.loads(tmp_json, parse_float=lambda x: round(float(x), 9))
        json.dump(tmp_json, file, indent=4, sort_keys=True)