\Mh bddlmZddlZddgZd dZejdd d d ZdS) )chainN tree_data tree_graphidchildrenc||dzkrtd|stdt j|stdkrt jdfdi|j||||iS)aReturns data in tree format that is suitable for JSON serialization and use in JavaScript documents. Parameters ---------- G : NetworkX graph G must be an oriented tree root : node The root of the tree ident : string Attribute name for storing NetworkX-internal graph data. `ident` must have a different value than `children`. The default is 'id'. children : string Attribute name for storing NetworkX-internal graph data. `children` must have a different value than `ident`. The default is 'children'. Returns ------- data : dict A dictionary with node-link formatted data. Raises ------ NetworkXError If `children` and `ident` attributes are identical. Examples -------- >>> from networkx.readwrite import json_graph >>> G = nx.DiGraph([(1, 2)]) >>> data = json_graph.tree_data(G, root=1) To serialize with json >>> import json >>> s = json.dumps(data) Notes ----- Node attributes are stored in this format but keys for attributes must be strings if you want to serialize with JSON. Graph and edge attributes are not stored. See Also -------- tree_graph, node_link_data, adjacency_data zG is not a tree.zG is not directed.zG is not weakly connected.z5The values for `id` and `children` must be different.c||}t|dkrgSg}|D]=}i|j| |i}||}|r||<||>|S)Nr)lennodesappend) nGnbrs children_childdc add_childrenridents b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/networkx/readwrite/json_graph/tree.pyrztree_data..add_childrenFst t99>>I   E0175>05%00A UA&&A (   Q    )number_of_nodesnumber_of_edges TypeError is_directednxis_weakly_connected NetworkXErrorr )rrootrrrs ``@rrrsh a//11A555*+++ ==??.,--- !! $ $64555 VWWW        Kagdm JUD(LLq4I4I J JJrT)graphs returns_graphctjfd|}|g}fd|D}j|fi|||S)aReturns graph from tree data format. Parameters ---------- data : dict Tree formatted graph data ident : string Attribute name for storing NetworkX-internal graph data. `ident` must have a different value than `children`. The default is 'id'. children : string Attribute name for storing NetworkX-internal graph data. `children` must have a different value than `ident`. The default is 'children'. Returns ------- G : NetworkX DiGraph Examples -------- >>> from networkx.readwrite import json_graph >>> G = nx.DiGraph([(1, 2)]) >>> data = json_graph.tree_data(G, root=1) >>> H = json_graph.tree_graph(data) See Also -------- tree_data, node_link_data, adjacency_data c|D]s}| }|||g}|r || fd|D}j|fi|tdS)NcNi|]!\}}|k |kt||"Sstr.0kvrrs r z4tree_graph..add_children..s9"aa5jjQ(]]A]]]r)add_edgegetitemsadd_node) parentrdatar grandchildrennodedatarrgraphrs rrz tree_graph..add_childrenxs . .DKE NN65 ) ) ) HHXr22M 3 UM222&*jjllH EN5 - -H - - - - . .rcNi|]!\}}|k |kt||"Sr&r'r)s rr-ztree_graph..s2SSSdaa5jjQ(]]A]]]r)rDiGraphr/r0r1)r3rrr rr5rr6s `` @@rrrVs@ JLLE . . . . . . . . ;D2&&ISSSSSdjjllSSSHEN4$$8$$$Ly!!! Lr)rr) itertoolsrnetworkxr__all__r _dispatchablerr&rrr=s  %KKKKKKKK\T22222232222r