ó '![c@s|dZddlmZddlmZddlmZdefd„ƒYZdefd„ƒYZ d efd „ƒYZ d S( s+Utilities for creating VCG and Dot diagramsiÿÿÿÿ(t is_exception(t VCGPrinter(t DotBackendt DiagramWritercBsVeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z RS( s,base class for writing project diagrams cCs4||_|\|_|_|_|_d|_dS(N(tconfigt pkg_edgest inh_edgest imp_edgest ass_edgestNonetprinter(tselfRtstyles((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyt__init__s cCsŒx…|D]}}|jjƒjddƒ}d||jjf}|j||ƒ|jdkrm|j|ƒn |j|ƒ|j ƒqWdS(s9write files for according to t t_s%s.%stclassN( ttitletstriptreplaceRt output_formatt set_printertTYPEt write_classestwrite_packagest close_graph(R tdiadefstdiagramtbasenamet file_name((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pytwrites  cCs£x]tt|jƒdd„ƒƒD]:\}}|jj|d|j|ƒddƒ||_q"Wx<|jdƒD]+}|jj|j j|j j|j qpWdS(swrite a package diagramtkeycSs|jS(N(R(tx((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyt*stlabeltshapetboxtdependsN( t enumeratetsortedtmodulesR t emit_nodet get_titletfig_idtget_relationshipst emit_edget from_objectt to_objectR(R Rtitobjtrel((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR's .% cCsxQtt|jdd„ƒƒD]1\}}|jj||j|ƒ||_qWx<|jdƒD]+}|jj|j j|j j|j qdWx<|jdƒD]+}|jj|j j|j j|j q£WxE|jdƒD]4}|jj|j j|j jd|j |jqâWdS(swrite a class diagramRcSs|jS(N(R(R ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR!5stspecializationt implementst associationR"N(R&R'tobjectsR R)t get_valuesR+R,R-R.R/RRtnameR(R RR0R1R2((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR2s+ cCs t‚dS(s set printerN(tNotImplementedError(R RR((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyREscCs t‚dS(sget project titleN(R9(R R1((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR*IscCs t‚dS(s get label and shape for classes.N(R9(R R1((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR7MscCs t‚dS(sfinalize the graphN(R9(R ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyRQs( t__name__t __module__t__doc__R RRRRR*R7R(((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyRs     t DotWritercBs;eZdZd„Zd„Zd„Zd„Zd„ZRS(s=write dot graphs from a diagram definition and a project c Cswtddddƒtddddƒtddddddƒtd d dddd dd ƒg}tj|||ƒdS( Nt arrowtailtnonet arrowheadtopentemptytnodetstyletdashedt fontcolortgreentdiamondtsolid(tdictRR (R RR ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR Zs cCs1tddƒ}t|d|ƒ|_||_dS(s9initialize DotWriter and add options for layout. trankdirtBTtadditional_paramN(RJRR R(R RRtlayout((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyRcscCs|jS(sget project title(R(R R1((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR*jscCs¼|j}|jdkr%d|}n|jjsd|dj|jƒf}x$|jD]}d||jf}qWWd|}nt|j ƒr©t ddd |d d ƒSt d |d d ƒS( s`get label and shape for classes. The label contains all attributes and methods t interfaceu«interface»\n%ss%s|%s\l|s\ls%s%s()\ls{%s}RFtredR"R#trecord( RR#Rtonly_classnamestjointattrstmethodsR8RRCRJ(R R1R"tfunc((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR7ns    cCs|jj|jƒdS(s$print the dot graph into N(R tgenerateR(R ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyRs(R:R;R<R RR*R7R(((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR=Vs    t VCGWritercBs;eZdZd„Zd„Zd„Zd„Zd„ZRS(s=write vcg graphs from a diagram definition and a project c Csƒtddddddƒtddddddƒtdddddd ddƒtddddd d ƒg}tj|||ƒdS( Nt arrowstyleRItbackarrowstyleR?t backarrowsizeii t linestyletdottedt textcolorRG(RJRR (R RR ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR ‡s  c Cswt|dƒ|_t|jƒ|_|jjd|ddddddd dƒ|jj|j_|jj|j_d S( s$initialize VCGWriter for a UML graphsw+Rtlayoutalgorithmtdfstlate_edge_labelstyest port_sharingtnotmanhattan_edgesN( RAt graph_fileRR t open_graphRCR)tedgeR-(R RR((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR“s cCs d|jS(sget project title in vcg formats\fb%s\fn(R(R R1((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR*sc Cs1t|jƒrd|j}n d|j}|jdkrDd}nd}|jjs|j}g|jD]}|j^qi}t d„|jg||Dƒƒ}d|d}d ||f}x|D]} d || f}qÇW|rúd ||f}nx!|D]}d ||f}qWnt d |d |ƒS(s`get label and shape for classes. The label contains all attributes and methods s \fb\f09%s\fns\fb%s\fnROtellipseR$css|]}t|ƒVqdS(N(tlen(t.0R8((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pys ²sRis%s\n\f%ss %s\n\f08%ss %s\n\f10%s()R"R#( RRCRR#RRRRTRUR8tmaxRJ( R R1R"R#RTRVRUtmaxlentlinetattr((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR7¡s&    $  cCs|jjƒ|jjƒdS(sclose graph and fileN(R RRftclose(R ((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyR½s (R:R;R<R RR*R7R(((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyRX„s   N( R<tpylint.pyreverse.utilsRtpylint.pyreverse.vcgutilsRt pylint.graphRtobjectRR=RX(((s6lib/python2.7/site-packages/pylint/pyreverse/writer.pyt s D.