D^(h ^ddlmZddlZddgZd dZej ddd dZy) )chainN tree_data tree_graphcZ|j|jdzk7r td|js tdt j |s tdk(rt j dfdi|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|dk(rgSg}|D]8}i|j| |i}||}|r||<|j|:|S)Nr)lennodesappend) nGnbrs children_childdc add_childrenchildrenidents `/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/networkx/readwrite/json_graph/tree.pyrztree_data..add_childrenFswt t9>I  E01775>05%0AUA&A(   Q   )number_of_nodesnumber_of_edges TypeError is_directednxis_weakly_connected NetworkXErrorr )r rootrrrs ``@rrrsh a//1A55*++ ==?,-- ! !! $455 VWW  Kaggdm JUD(Lq4I JJrT)graphs returns_graphc( tj  fd|}|jg}|jDcic]\}}|k7s |k7st ||}}} j |fi||| Scc}}w)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]}| } j|||j g}|r |||jDcic]\}}| k7s | k7st||}}} j|fi|ycc}}w)N)add_edgegetitemsstradd_node) parentrdatar grandchildrenkvnodedatarrgraphrs rrz tree_graph..add_childrenxs .DKE NN65 ) HHXr2MUM2&*jjl"aa5jQ(]A H ENN5 -H - . s B B "B )rDiGraphr%r&r'r() r*rrrrr,r-r.rr/s `` @@rrrVs@ JJLE . ;D2&I&*jjlSdaa5jQ(]A SHSENN4$8$y! LTs BBB)idr) itertoolsrnetworkxr__all__r _dispatchablerrrr7s@  %KK\T2232r