Skip to content

Vis.js Node Enricher Example

Prompt

I have a JSON file that represents the nodes and edges in a network graph. I would like you to generate a Python program that reads in this JSON file and returns it with one addition. If the node has a group of 1, I want you to add another property "x": -300. 

All other elements in the JSON file should remain unchained. Here is a sample of the nodes: { "nodes": \[ { "id": 1, "label": "Ohm's Law", "group": 1 }, { "id": 2, "label": "Voltage", "group": 1 }, { "id": 3, "label": "Current", "group": 1 }, { "id": 4, "label": "Resistance", "group": 1 }, { "id": 5, "label": "Power", "group": 1 }, { "id": 6, "label": "Energy", "group": 1 } You should return: "nodes": \[ { "id": 1, "label": "Ohm's Law", "group": 1, "x": 0 } etc.

ChatGPT Response

Certainly! Below is a Python program that accomplishes your requirements. This script reads a JSON file representing a network graph, adds an "x": -300 property to each node with "group": 1, and writes the modified data back to a new JSON file. All other elements in the JSON remain unchanged.

Step-by-Step Explanation

  1. Import Necessary Modules:

    • json: To handle JSON data.
    • sys: To handle command-line arguments (optional, for flexibility).

    • Define the Functionality:

    • Reading the JSON File: The script reads the input JSON file containing the network graph.

    • Modifying Nodes: It iterates through each node in the "nodes" list. If a node has "group": 1, it adds a new property "x": -300 to that node.
    • Writing the Modified JSON: The script writes the updated JSON data to an output file.

    • Usage:

    • You can run the script from the command line, specifying the input and output file paths. If no arguments are provided, it defaults to input.json and output.json.

Python Code

 1
 2
 3
 4
 5
 6
 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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import json
import sys

def add_x_property(input_file, output_file):
    try:
        # Read the input JSON file
        with open(input_file, 'r', encoding='utf-8') as f:
            data = json.load(f)

        # Check if 'nodes' key exists
        if 'nodes' not in data:
            print(f"Error: The key 'nodes' was not found in {input_file}.")
            return

        # Iterate through each node and add "x": -300 if group is 1
        for node in data['nodes']:
            if node.get('group') == 1:
                node['x'] = -300

        # Write the modified data to the output JSON file
        with open(output_file, 'w', encoding='utf-8') as f:
            json.dump(data, f, indent=4)

        print(f"Successfully wrote the modified JSON to {output_file}.")

    except FileNotFoundError:
        print(f"Error: The file {input_file} does not exist.")
    except json.JSONDecodeError:
        print(f"Error: The file {input_file} is not a valid JSON file.")
    except Exception as e:
        print(f"An unexpected error occurred: {e}")

def main():
    # Default input and output file names
    input_file = 'input.json'
    output_file = 'output.json'

    # If command-line arguments are provided, use them
    if len(sys.argv) == 3:
        input_file = sys.argv[1]
        output_file = sys.argv[2]
    elif len(sys.argv) != 1:
        print("Usage: python script.py [input_file output_file]")
        return

    add_x_property(input_file, output_file)

if __name__ == "__main__":
    main()

How to Use the Script

  1. Save the Script:

    • Save the above code to a file, for example, modify_graph.py.

    • Prepare Your JSON File:

    • Ensure your JSON file (e.g., input.json) is properly formatted. Here's an example based on your sample:

       1
 2
 3
 4
 5
 6
 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
36
37
38
39
40
41
      json
Copy code

`{
    "nodes": [
        {
            "id": 1,
            "label": "Ohm's Law",
            "group": 1
        },
        {
            "id": 2,
            "label": "Voltage",
            "group": 1
        },
        {
            "id": 3,
            "label": "Current",
            "group": 1
        },
        {
            "id": 4,
            "label": "Resistance",
            "group": 1
        },
        {
            "id": 5,
            "label": "Power",
            "group": 1
        },
        {
            "id": 6,
            "label": "Energy",
            "group": 1
        }
    ],
    "edges": [
        // Your edge definitions here
    ]
}
`

  2. Run the Script:

    • Open your terminal or command prompt.

    • Navigate to the directory containing modify_graph.py and your input.json.

    • Execute the script using Python. You can specify input and output files or use the defaults.

      Using Default File Names (input.json and output.json):

      1
2
3
4
5
      bash
Copy code

`python modify_graph.py
`

      Specifying Input and Output Files:

      1
2
3
4
5
      bash
Copy code

`python modify_graph.py your_input.json your_output.json
`

  3. Check the Output:

    • After running the script, check the output.json file. Nodes with "group": 1 will have an additional property "x": -300.

    • Example Output:

       1
 2
 3
 4
 5
 6
 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
36
37
38
39
40
41
42
43
44
45
46
47
      json
Copy code

`{
    "nodes": [
        {
            "id": 1,
            "label": "Ohm's Law",
            "group": 1,
            "x": -300
        },
        {
            "id": 2,
            "label": "Voltage",
            "group": 1,
            "x": -300
        },
        {
            "id": 3,
            "label": "Current",
            "group": 1,
            "x": -300
        },
        {
            "id": 4,
            "label": "Resistance",
            "group": 1,
            "x": -300
        },
        {
            "id": 5,
            "label": "Power",
            "group": 1,
            "x": -300
        },
        {
            "id": 6,
            "label": "Energy",
            "group": 1,
            "x": -300
        }
    ],
    "edges": [
        // Your edge definitions here
    ]
}
`

Notes

  • Error Handling: The script includes basic error handling for missing files and invalid JSON formats.
  • Flexibility: You can specify different input and output files via command-line arguments.
  • Extensibility: You can easily modify the script to add more properties or handle different conditions as needed.

Feel free to customize the script further based on your specific requirements!

o1-mini

window.__oai_logHTML?window.__oai_logHTML():window.__oai_SSR_HTML=window.__oai_SSR_HTML||Date.now();requestAnimationFrame((function(){window.__oai_logTTI?window.__oai_logTTI():window.__oai_SSR_TTI=window.__oai_SSR_TTI||Date.now()}))

ChatGPT can make mistakes. Check imp