o ˷e^@s4ddlmZddlZddgZd ddZd ddZdS) )chainN tree_data tree_graphidchildrencs||dkrtd|stdt|std|dur9ddl}d}|j|tdd |d |d krBt d fd dt t |j | |fg}|||<|S)a5Returns 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 attrs : dict A dictionary that contains two keys 'id' and 'children'. The corresponding values provide the attribute names for storing NetworkX-internal graph data. The values should be unique. Default value: :samp:`dict(id='id', children='children')`. If some user-defined graph data use these attribute names as data keys, they may be silently dropped. .. deprecated:: 2.6 The `attrs` keyword argument is replaced by `ident` and `children` and will be removed in networkx 3.0 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.Nra The `attrs` keyword argument of tree_data is deprecated and will be removed in networkx 3.0. It is replaced with explicit `ident` and `children` keyword arguments. To make this warning go away and ensure usage is forward compatible, replace `attrs` with `ident` and `children, for example: >>> tree_data(G, root, attrs={'id': 'foo', 'children': 'bar'}) should instead be written as >>> tree_data(G, root, ident='foo', children='bar') The default values of 'id' and 'children' will not change. stacklevelrrz5The values for `id` and `children` must be different.csh||}t|dkr gSg}|D]!}tt|j||fg}||}|r,||<||q|S)Nr)lendictrnodesitemsappend)nGnbrs children_childdc add_childrenridentY/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/readwrite/json_graph/tree.pyrks   ztree_data..add_children)number_of_nodesnumber_of_edges TypeError is_directednxis_weakly_connectedwarningswarnDeprecationWarning NetworkXErrorr rr r)rrootattrsrrr"msgdatarrrr s&B    c st|durddl}d}|j|tdd|d|dfdd |}|g}fd d |D}j|fi|||S) aReturns graph from tree data format. Parameters ---------- data : dict Tree formatted graph data attrs : dict A dictionary that contains two keys 'id' and 'children'. The corresponding values provide the attribute names for storing NetworkX-internal graph data. The values should be unique. Default value: :samp:`dict(id='id', children='children')`. .. deprecated:: 2.6 The `attrs` keyword argument is replaced by `ident` and `children` and will be removed in networkx 3.0 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 Nra The `attrs` keyword argument of tree_graph is deprecated and will be removed in networkx 3.0. It is replaced with explicit `ident` and `children` keyword arguments. To make this warning go away and ensure usage is forward compatible, replace `attrs` with `ident` and `children, for example: >>> tree_graph(data, attrs={'id': 'foo', 'children': 'bar'}) should instead be written as >>> tree_graph(data, ident='foo', children='bar') The default values of 'id' and 'children' will not change.rr rrcsf|D].}|}|||g}|r||fdd|D}j|fi|qdS)Nc*i|]\}}|kr|krt||qSrstr.0kvrrrr s$z4tree_graph..add_children..)add_edgegetradd_node)parentrr)r grandchildrennodedatarrgraphrrrrs    z tree_graph..add_childrencr*rr+r-r1rrr2s*ztree_graph..)r DiGraphr"r#r$r4rr5) r)r'rrr"r(r&rr8rr9rr}s)   )Nrr) itertoolsrnetworkxr __all__rrrrrrs  t