c U d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
mZmZmZmZ ddlmZ G d d
e Z ed Zd ed
< ej dej dej diZded< e
j ddde
j dddde
j ddddde
j ddd dde
j dd diZ!d ed < G d d!e Z"dS )"zNClass to generate files in dot format and image formats supported by Graphviz. ) annotationsN)Enum)Path)nodes)EdgeTypeLayoutNodePropertiesNodeTypePrinter)get_annotation_labelc e Zd ZdZdS )
HTMLLabelsz
N)__name__
__module__ __qualname__LINEBREAK_LEFT z
, fontcolor="r= z
" [color="z , shape="z
", style="z"];)r r color
DEFAULT_COLOR_build_label_for_noder& r> )
r8 rC rD rE shaperL r" rH
label_partfontcolor_parts
r emit_nodezDotPrinter.emit_nodeK s 4'd333Ju
$.$4X
dFX !T%7 7DW**:66-2:)))))
7A7K S 3J 0 3 3 3 3QS
h h h h h h
h hU h h^c h h h
r r c 0 |j sdS |j }|j |j |S |j pg }t j j d |D }d| d| t j j d}|j pg }|D ]}| | }| r d|j dn|j }|| dd | d z
}|j
r|d
t |j
z z
}|t j j z
}|d z
}|S ) NrK c 3 B K | ]}| d d V dS )|z\|N)replace).0attrs r z3DotPrinter._build_label_for_node..n sI B
B
)-DLLe $ $B
B
B
B
B
B
r {rU zz(, )z: }) rH attrsmethodsr r r? join_get_method_arguments is_abstractrC returnsr ) r8 rE rH r_ attrs_stringr` funcargs method_names r rN z DotPrinter._build_label_for_nodec sx 2 %
(: L &+1r%4:<AA B
B
16B
B
B
OeNNlNJ,E,KNNN ,6+=+C ; ;D--d33D)-)9)9););O%di%%%%DI
99tyy999 9E|
C 4T\ B BBB 28: :EE
r from_nodeto_noder rH
str | Nonec t | }d | D }|r| d| d | d| d| dd t | d dS ) zACreate an edge from one node to another to display relationships.c $ g | ]
\ }}| d | dS )z="r= r )rW propr? s r
z(DotPrinter.emit_edge.. s- KKKuD$$E$$$KKKr zlabel="r= z" -> "z" [r\ z];N)r) itemsappendr> ra sorted)r8 ri rj rD rH
arrowstyler_ s r emit_edgezDotPrinter.emit_edge s E]
KK
8H8H8J8JKKK - LL+5+++ , , , OiOOwOO499VE]]3K3KOOOPPPPPr
outputfilec | d}| j }|[d}t j d| \ }}t j d| \ }}t j | t j | nft
| j d }|s
d}|dz |z }||vr-t j d| \ }}t j | n|}t |dd 5 }|
| j d d d n # 1 swxY w Y ||vr2t j
d d
||d |gd
t j | d S d S )N)dotgvpngz.gvz.png.wutf8)encodingrw z-Tz-oT)check) _close_graphr- tempfilemkstemposcloser suffixlstripr(
writelineslines
subprocessrununlink) r8 ru graphviz_extensionsrC targetpdotdot_sourcepathppngoutfiles r generatezDotPrinter.generate s +z ,F#+#3E4#@#@ D. ' / = = D* HTNNN HTNNNN*%%,33C88F
7 '# - 6
00
,'/'7t'D'D$n!+
.#
7
7
7 +7 tz * * * + + + + + + + + + + + + + + + , , & NfndJGt
In % % % % % & &s 3DD !D c 0 | d dS )z2Emit the lines needed to properly close the graph.z}
N)r> rA s r r zDotPrinter._close_graph s %r )NN)r- r. r/ r0 r1 r2 )r: r; )N)rC r. rD r
rE rF r: r; )rE r r: r. )
ri r. rj r. rD r rH rk r: r; )ru r. r: r; ) r r r rM r7 rB rR rN rt r r
__classcell__)r9 s @r r+ r+ 3 s M
!%/3 A A A A A A A 3 3 3 3 -1
0 F ! Q Q Q Q Q & & & &8 r r+ )#__doc__
__future__r r r r enumr pathlibr astroidr pylint.pyreverse.printerr r r r
r pylint.pyreverse.utilsr r frozensetr __annotations__PACKAGE INTERFACECLASSr INHERITS
IMPLEMENTS ASSOCIATION AGGREGATIONUSESr) r+ r r r r s
U T T " " " " " " X X X X X X X X X X X X X X 7 7 7 7 7 7* * * * * * * * $-9-N#O#O O O O O e NH
V'BB vGhWW
Mf== * &z z z z z z z z z zr