a h77@sdZddlZddlmZddlZddlmZgdZedddd d Z edd dej d dd ddZ ej ddddZ ddZ ddZdddZdddZdS)a ***** Pydot ***** Import and export NetworkX graphs in Graphviz dot format using pydot. Either this module or nx_agraph can be used to interface with graphviz. Examples -------- >>> G = nx.complete_graph(5) >>> PG = nx.nx_pydot.to_pydot(G) >>> H = nx.nx_pydot.from_pydot(PG) See Also -------- - pydot: https://github.com/erocarrera/pydot - Graphviz: https://www.graphviz.org - DOT Language: http://www.graphviz.org/doc/info/lang.html N)getpreferredencoding) open_file) write_dotread_dotgraphviz_layout pydot_layoutto_pydot from_pydotw)modecCs.d}tj|tddt|}||dS)zgWrite NetworkX graph G to Graphviz dot format on path. Path can be a string or a file handle. znx.nx_pydot.write_dot depends on the pydot package, which has known issues and is not actively maintained. Consider using nx.nx_agraph.write_dot instead. See https://github.com/networkx/networkx/issues/5723 stacklevelN)warningswarnDeprecationWarningrwriteZ to_string)GpathmsgPrG/var/www/auris/lib/python3.9/site-packages/networkx/drawing/nx_pydot.pyr&s rrZpydot_read_dot)namegraphscCs:ddl}d}tj|tdd|}||}t|dS)aOReturns a NetworkX :class:`MultiGraph` or :class:`MultiDiGraph` from the dot file with the passed path. If this file contains multiple graphs, only the first such graph is returned. All graphs _except_ the first are silently ignored. Parameters ---------- path : str or file Filename or file handle. Returns ------- G : MultiGraph or MultiDiGraph A :class:`MultiGraph` or :class:`MultiDiGraph`. Notes ----- Use `G = nx.Graph(nx.nx_pydot.read_dot(path))` to return a :class:`Graph` instead of a :class:`MultiGraph`. rNznx.nx_pydot.read_dot depends on the pydot package, which has known issues and is not actively maintained. Consider using nx.nx_agraph.read_dot instead. See https://github.com/networkx/networkx/issues/5723r r)pydotrrrreadgraph_from_dot_datar )rrrdataZP_listrrrr8s r)rc Cs d}tj|tdd|dr$d}nd}|dkrL|rBt}qbt}n|rZt}nt }| d}|d kr~||_ | D]2}| d}|d vrq|j|fi|q|D]}|}|} |} g} g} t|tr| | dn |d D]} | | dq t| trB| | dn | d D]}| | dqJ| D]&}| D]}|j||fi| qnqfq|}|r||jd<z|d |jd <WnttfyYn0z|d |jd<WnttfyYn0|S)aReturns a NetworkX graph from a Pydot graph. Parameters ---------- P : Pydot graph A graph created with Pydot Returns ------- G : NetworkX multigraph A MultiGraph or MultiDiGraph. Examples -------- >>> K5 = nx.complete_graph(5) >>> A = nx.nx_pydot.to_pydot(K5) >>> G = nx.nx_pydot.from_pydot(A) # return MultiGraph # make a Graph instead of MultiGraph >>> G = nx.Graph(nx.nx_pydot.from_pydot(A)) znx.nx_pydot.from_pydot depends on the pydot package, which has known issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723r rNFTgraph")noder!edgenodesrr$r%)rrrZ get_strictget_typenxZ MultiGraphZGraphZ MultiDiGraphZDiGraphget_namestriprZ get_node_listadd_nodeZget_attributesZ get_edge_list get_sourceZget_destination isinstancestrappendadd_edger!Zget_node_defaults IndexError TypeErrorZget_edge_defaults)rrZ multiedgesNrpneuvattrsdZunodesZvnodesZ source_nodeZdestination_nodeZpattrrrrr csd            r cCs d|vo|ddkp|ddkS)N:rr"r)r:rrr_check_colon_quotessr>cCsddl}d}tj|tdd|r*d}nd}t|dkoD| }|j}|j di}|dkrz|j d||d |}n |j d |d f||d |}z|j fi|j d Wnt yYn0z|jfi|j d Wnt yYn0|jd dD]n\}} dd| D} t|}t|pBtdd| D} | rRtd|j|fi| } || q|r|jd d dD]\} }}}dd|D}t| t|} }t| pt|ptdd|D} | rtd|j| |fdt|i|}||qn|jd dD]\} }}dd|D}t| t|} }t| pt|ptdd|D} | rtd|j| |fi|}||q*|S)aReturns a pydot graph from a NetworkX graph N. Parameters ---------- N : NetworkX graph A graph created with NetworkX Examples -------- >>> K5 = nx.complete_graph(5) >>> P = nx.nx_pydot.to_pydot(K5) Notes ----- rNznx.nx_pydot.to_pydot depends on the pydot package, which has known issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723r rZdigraphr!r#) graph_typestrictr"r$r%T)r cSsi|]\}}t|t|qSrr..0kr8rrr zto_pydot..css"|]\}}t|pt|VqdSNr>rBrrr szto_pydot..zNode names and attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258)r keyscSs&i|]\}}|dkrt|t|qS)keyrArBrrrrErFcss"|]\}}t|pt|VqdSrGrHrCrDvalrrrrIsaNode names and attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258rKcSsi|]\}}t|t|qSrrArBrrrrE*rFcss"|]\}}t|pt|VqdSrGrHrLrrrrI0s)r#)rrrrZ is_directedr(Znumber_of_selfloopsZ is_multigraphrr!getZDotZset_node_defaultsKeyErrorZset_edge_defaultsr&itemsr.r>any ValueErrorNoder+edgesZEdger0)r3rrr?r@rZgraph_defaultsrr5ZnodedataZ str_nodedataZ raise_errorr4r7r8rKZedgedataZ str_edgedatar%rrrrs         rneatocCs"d}tj|tddt|||dS)a=Create node positions using Pydot and Graphviz. Returns a dictionary of positions keyed by node. Parameters ---------- G : NetworkX Graph The graph for which the layout is computed. prog : string (default: 'neato') The name of the GraphViz program to use for layout. Options depend on GraphViz version but may include: 'dot', 'twopi', 'fdp', 'sfdp', 'circo' root : Node from G or None (default: None) The node of G from which to start some layout algorithms. Returns ------- Dictionary of (x, y) positions keyed by node. Examples -------- >>> G = nx.complete_graph(4) >>> pos = nx.nx_pydot.graphviz_layout(G) >>> pos = nx.nx_pydot.graphviz_layout(G, prog="dot") Notes ----- This is a wrapper for pydot_layout. znx.nx_pydot.graphviz_layout depends on the pydot package, which has known issues and is not actively maintained. Consider using nx.nx_agraph.graphviz_layout instead. See https://github.com/networkx/networkx/issues/5723r r)rprogroot)rrrr)rrVrWrrrrrAsrcCsVddl}d}tj|tddt|}|dur<|dt||j|d}t|td}|d krt d |d t t d t d t dt d|ddS| |}t |dksJ|d} i} | D]} t| } t | rtd|| } | | }t|tr|d}|dd}|dur|d\}}t|t|f| | <q| S)aCreate node positions using :mod:`pydot` and Graphviz. Parameters ---------- G : Graph NetworkX graph to be laid out. prog : string (default: 'neato') Name of the GraphViz command to use for layout. Options depend on GraphViz version but may include: 'dot', 'twopi', 'fdp', 'sfdp', 'circo' root : Node from G or None (default: None) The node of G from which to start some layout algorithms. Returns ------- dict Dictionary of positions keyed by node. Examples -------- >>> G = nx.complete_graph(4) >>> pos = nx.nx_pydot.pydot_layout(G) >>> pos = nx.nx_pydot.pydot_layout(G, prog="dot") Notes ----- If you use complex node objects, they may have the same string representation and GraphViz could treat them as the same node. The layout may assign both nodes a single location. See Issue #1568 If this occurs in your case, consider relabeling the nodes just for the layout computation using something similar to:: H = nx.convert_node_labels_to_integers(G, label_attribute='node_label') H_layout = nx.nx_pydot.pydot_layout(G, prog='dot') G_layout = {H.nodes[n]['node_label']: p for n, p in H_layout.items()} rNznx.nx_pydot.pydot_layout depends on the pydot package, which has known issues and is not actively maintained. See https://github.com/networkx/networkx/issues/5723r rrW)rV)encodingr#zGraphviz layout with z failedzTo debug what happened try:zP = nx.nx_pydot.to_pydot(G)zP.write_dot("file.dot")z And then run z on file.dotr zNode names and node attributes should not contain ":" unless they are quoted with "". For example the string 'attribute:data1' should be written as '"attribute:data1"'. Please refer https://github.com/pydot/pydot/issues/258r=,)rrrrrsetr.Z create_dotrprintrlenr&r>rRrSr)Zget_noder-listZget_possplitfloat)rrVrWrrrZD_bytesDZQ_listQZnode_posr5Zstr_nZ pydot_noder$posxxyyrrrrjsH&     r)rUN)rUN)__doc__rlocalerZnetworkxr(Znetworkx.utilsr__all__r _dispatchrr r>rrrrrrrs        )  au )