B ,["@sdZddlZddlZddlmZmZmZddddgZddZ ed d d dd dZ eddd dddZ ddZ ddZ ddZdS)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)is_string_like open_filemake_str read_pajek parse_pajekgenerate_pajek write_pajekc cs|jdkrd}n|j}d|Vt|}tt|tdt|d}x|D]}|j|i }| dd}| dd}t | d||}|||<| d d } d t t||||| f} xb|D]V\} } t| r| dkr| d t| t| f7} qtd | t| rdndfqW| VqNW|r>dVndVx|jddD]\} } }| }| dd}d t t|| || |f} xh|D]\\} } t| r| dkr| d t| t| f7} n td| t| rdndfqW| VqRWdS)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 %sxgyidshapeZellipse z %s %sz'Node attribute %s is not processed. %s.zEmpty attributezNon-string attributez*arcsz*edgesT)dataweightg?z'Edge attribute %s is not processed. %s.N)nameorderlistdictziprangelennodesgetcopypopintjoinmap make_qstritemsrstripwarningswarnZ is_directedZedges)GrrZ nodenumbernZnar r r rskvuZedgedatadvaluer-7lib/python3.7/site-packages/networkx/readwrite/pajek.pyr!sJ        r wb)modeUTF-8cCs.x(t|D]}|d7}|||q WdS)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%pathencodingliner-r-r.rdsrbcsfdd|D}t|S)aTRead 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)N)decode).0r7)r6r-r. szread_pajek..)r)r5r6linesr-)r6r.rsc svddl}t|rt|d}tdd|D}t}gx2|rpy t|}WnPYnX|dry|dd\}}Wnt k rYn X||j d<q@|d ri}|\}}xt t |D]}t|}y"d d|t |d D} Wn$tk r*|t|} YnX| dd \} }|||||| <| |j|d <y6| d d\} } } |j|t| t| | dWn YnXt| ddd | ddd }|j||qWq@|ds|dr6|drt|}|dr0|}x:|D]}y"dd|t |d D} Wn$tk r|t|} YnXt| d krq8| dd \}}|||}|||}i}y&| d d}|dt|diWn YnXt| ddd | ddd }|||j||f|q8Wq@|dr@t|}fddt|D}||q@W|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() rNr2cSsg|]}|dqS)r2)rstrip)r:r7r-r-r. szparse_pajek..z*networkr rz *verticescSsg|]}|dqS)zutf-8)r9)r:r r-r-r.r>szutf-8r )r r rz*edgesz*arcsz*edgecSsg|]}|dqS)zutf-8)r9)r:r r-r-r.r>srz*matrixc3sN|]F\}}t|D]0\}}t|dkr||dt|ifVqqdS)rrN) enumeratesplitr)r:rowr7colr)labelsr-r.r;szparse_pajek..)shlexriterrEnxZ MultiDiGraphnextlower startswith ValueErrorZgraphrrrr4AttributeErrorstrappendZadd_noderupdatefloatrZ MultiGraphZ to_directedrrZadd_edgeZDiGraphrDZadd_edges_from)r<rIr%lZlabelrZ nodelabelsZnnodesiZ splitliner r r rZ extra_attrZuiZvir*r)Z edge_datawZadj_listr-)rHr.rs                cCs$t|st|}d|kr d|}|S)zbReturn the string representation of t. Add outer double-quotes if the string has a space. rz"%s")rrQ)tr-r-r.r s r cCsddl}|ddS)Nrztest.net)osunlink)modulerYr-r-r.teardown_modulesr\)r1)r1)__doc__r#ZnetworkxrKZnetworkx.utilsrrr__all__rrrrr r\r-r-r-r.s C   j