a h!@sdZddlZddlZddlmZgdZddZeddd dd d Zedd d ej dddddZ ej ddddZ ddZ dS)a  ***** Pajek ***** Read graphs in Pajek format. This implementation handles directed and undirected graphs including those with self loops and parallel edges. Format ------ See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. N) open_file) read_pajek parse_pajekgenerate_pajek write_pajekc csX|jdkrd}n|j}d|Vt|}tt|tdt|d}|D]$}|j|i }| dd}| dd}zt | d||}Wn4t y} z| j d 7_ WYd } ~ n d } ~ 00|||<| d d } d tt||||| f} |D]j\} } t| trB| dkrB| d t| d t| 7} n(td| dt| tr^dnddq| VqN|rdVndV|jddD]\}} }| }| dd}d tt|||| |f} |D]j\} } t| tr| dkr| d t| d t| 7} n(td| dt| tr:dnddq| Vqd S)zGenerate lines in Pajek graph format. Parameters ---------- G : graph A Networkx graph References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. ZNetworkXz *vertices xgyid)zSPajek format requires 'id' to be an int(). Refer to the 'Relabeling nodes' section.NshapeZellipse zNode attribute z is not processed. zEmpty attributezNon-string attribute.*arcs*edgesT)dataweightg?zEdge attribute )nameorderlistdictziprangelennodesgetcopypopint ValueErrorargsjoinmap make_qstritems isinstancestrstripwarningswarnZ is_directededges)GrrZ nodenumbernnar r r errr skvuZedgedatadvaluer5F/var/www/auris/lib/python3.9/site-packages/networkx/readwrite/pajek.pyrsN         rrwb)modeUTF-8cCs*t|D]}|d7}|||qdS)aWrite graph in Pajek format to path. Parameters ---------- G : graph A Networkx graph path : file or string File or filename to write. Filenames ending in .gz or .bz2 will be compressed. Examples -------- >>> G = nx.path_graph(4) >>> nx.write_pajek(G, "test.net") Warnings -------- Optional node attributes and edge attributes must be non-empty strings. Otherwise it will not be written into the file. You will need to convert those attributes to strings if you want to keep them. References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information.  N)rwriteencode)r+pathencodingliner5r5r6rcs rrb)Zgraphscsfdd|D}t|S)aZRead graph in Pajek format from path. Parameters ---------- path : file or string File or filename to write. Filenames ending in .gz or .bz2 will be uncompressed. Returns ------- G : NetworkX MultiGraph or MultiDiGraph. Examples -------- >>> G = nx.path_graph(4) >>> nx.write_pajek(G, "test.net") >>> G = nx.read_pajek("test.net") To create a Graph instead of a MultiGraph use >>> G1 = nx.Graph(G) References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. c3s|]}|VqdS)Ndecode.0r?r>r5r6 zread_pajek..)r)r=r>linesr5rEr6rsrc sfddl}t|tr t|d}tdd|D}t}g|rbz t|}WnYqbYn0| drz|dd\}}Wnt yYn 0||j d<q>| d ri}|\}}t t |D]}t|}z"d d|t|d D} Wn"ty"|t|} Yn0| dd \} }|||||| <| |j|d <z6| d d\} } } |j|t| t| | dWn Yn0t| ddd | ddd }|j||qq>| ds| dr(| drt|}| dr&|}|D]}z"dd|t|d D} Wn"tyr|t|} Yn0t| d krq*| dd \}}|||}|||}i}z&| d d}|dt|diWn Yn0t| ddd | ddd }|||j||fi|q*q>| dr>t|}fddt|D}||q>|S)zParse Pajek format graph from string or iterable. Parameters ---------- lines : string or iterable Data in Pajek format. Returns ------- G : NetworkX graph See Also -------- read_pajek rNr:cSsg|]}|dqS)r:)rstriprCr5r5r6 rGzparse_pajek..z*networkrrz *verticescSsg|]}|dqSutf-8rArDr r5r5r6rJsrLr )r r r rrz*edgecSsg|]}|dqSrKrArMr5r5r6rJsrz*matrixc3sN|]F\}}t|D]0\}}t|dkr||dt|ifVqqdS)rrN) enumeratesplitr)rDrowr?colrlabelsr5r6rF s zparse_pajek..)shlexr%r&iterrTnxZ MultiDiGraphnextlower startswithrgraphrrr<AttributeErrorappendadd_noderupdatefloatrZ MultiGraphZ to_directedrrZadd_edgeZDiGraphrSZadd_edges_from)rHrYr+llabelrZ nodelabelsZnnodesiZ splitliner r r r Z extra_attrZuivir2r1Z edge_datawZadj_listr5rWr6rs                    rcCs*t|tst|}d|vr&d|d}|S)zcReturns the string representation of t. Add outer double-quotes if the string has a space. r ")r%r&)tr5r5r6r#s   r#)r9)r9) __doc__r(Znetworkxr[Znetworkx.utilsr__all__rr _dispatchrrr#r5r5r5r6s J     o