Warning

This documents an unmaintained version of NetworkX. Please upgrade to a maintained version and see the current NetworkX documentation.

Source code for networkx.readwrite.json_graph.jit

#    Copyright (C) 2011-2019 by
#    Aric Hagberg <hagberg@lanl.gov>
#    Dan Schult <dschult@colgate.edu>
#    Pieter Swart <swart@lanl.gov>
#    All rights reserved.
#    BSD license.

"""
Read and write NetworkX graphs as JavaScript InfoVis Toolkit (JIT) format JSON.

See the `JIT documentation`_ for more examples.

Format
------
var json = [
  {
    "id": "aUniqueIdentifier",
    "name": "usually a nodes name",
    "data": {
      "some key": "some value",
      "some other key": "some other value"
     },
    "adjacencies": [
    {
      nodeTo:"aNodeId",
      data: {} //put whatever you want here
    },
    'other adjacencies go here...'
  },

  'other nodes go here...'
];
.. _JIT documentation: http://thejit.org
"""

import json
import networkx as nx
from networkx.utils.decorators import not_implemented_for

__all__ = ['jit_graph', 'jit_data']


[docs]def jit_graph(data, create_using=None): """Read a graph from JIT JSON. Parameters ---------- data : JSON Graph Object create_using : Networkx Graph, optional (default: Graph()) Return graph of this type. The provided instance will be cleared. Returns ------- G : NetworkX Graph built from create_using if provided. """ if create_using is None: G = nx.Graph() else: G = create_using G.clear() if nx.utils.is_string_like(data): data = json.loads(data) for node in data: G.add_node(node['id'], **node['data']) if node.get('adjacencies') is not None: for adj in node['adjacencies']: G.add_edge(node['id'], adj['nodeTo'], **adj['data']) return G
[docs]@not_implemented_for('multigraph') def jit_data(G, indent=None): """Returns data in JIT JSON format. Parameters ---------- G : NetworkX Graph indent: optional, default=None If indent is a non-negative integer, then JSON array elements and object members will be pretty-printed with that indent level. An indent level of 0, or negative, will only insert newlines. None (the default) selects the most compact representation. Returns ------- data: JIT JSON string """ json_graph = [] for node in G.nodes(): json_node = { "id": node, "name": node } # node data json_node["data"] = G.nodes[node] # adjacencies if G[node]: json_node["adjacencies"] = [] for neighbour in G[node]: adjacency = { "nodeTo": neighbour, } # adjacency data adjacency["data"] = G.edges[node, neighbour] json_node["adjacencies"].append(adjacency) json_graph.append(json_node) return json.dumps(json_graph, indent=indent)