eq
U d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
mZmZmZmZ ddlmZ G d d
e Z ed Zd ed
< ej0 dej2 diZded< e
j6 ddde
j8 ddddde
j: ddddde
j< ddde
j> dddd iZ d ed < G d d e Z!y)!zNClass to generate files in dot format and image formats supported by Graphviz. ) annotationsN)Enum)Path)nodes)EdgeTypeLayoutNodePropertiesNodeTypePrinter)get_annotation_labelc e Zd ZdZy)
HTMLLabelsz
N)__name__
__module__ __qualname__LINEBREAK_LEFT >/usr/lib/python3/dist-packages/pylint/pyreverse/dot_printer.pyr r s )Nr r )utf-8z
iso-8859-1latin1zfrozenset[str]ALLOWED_CHARSETSboxrecordzdict[NodeType, str]SHAPESnoneempty) arrowtail arrowheadgreendiamondsolid) fontcolorr r styleodiamondopendashed)r r r$ z dict[EdgeType, dict[str, str]]ARROWSc e Zd ZdZ d
d fd
Zd dZ d
ddZddZddZ d
ddZ ddZ
d d Z xZ S )
DotPrinterblackc ^ |xs t j }d| _ t | ||| y )Nr )r
BOTTOM_TO_TOPcharsetsuper__init__)selftitlelayoutuse_automatic_namespace __class__s r r0 zDotPrinter.__init__9 s. /6//
(?@r c j | j d| j d | j r(| j d| j j | j rT| j j t v sJ d| j | j d| j d yy)zEmit the header lines.z digraph "z" {zrankdir=zunsupported charset z charset=""N)emitr2 r3 valuer. lowerr r1 s r _open_graphzDotPrinter._open_graphC s Idjj\./ ;; II!2!2 34 5 << ""$(88
5%dll^4
58 II $,, q1 2 r c X | t | }t | }|j |j n | j }|| j k7 rdnd}| j | }|rd| dnd}|j
rd|j
d nd} | j
d | d
| d | | d | d | d
y)zVCreate a new node.
Nodes can be classes, packages, participants etc.
N)labelfilledr" z , label=<> z
, fontcolor="r7 z
" [color="z , shape="z
", style="z"];)r r color
DEFAULT_COLOR_build_label_for_noder# r8 )
r1 nametype_
propertiesshaperB r$ r>
label_partfontcolor_parts
r emit_nodezDotPrinter.emit_nodeN s 'd3Ju
$.$4$4$@
dFXFX !T%7%7 7W**:6-2yq)
7A7K7KmJ 0 01 3QS
vZwa'7
|9UGS]^c]ddg h
r c |j sy|j }|j |j |S |j xs g }t j j
j
d |D }d| d| t j j
d}|j xs g }|D ] }dj
| j | j dd }|j rd|j dn |j }|| d | d
z
}|j r,t |j } |d | j | z z
}|t j j
z
} |d z
}|S )
NrA c 3 @ K | ] }|j d d yw)|\|Nreplace).0attrs r z3DotPrinter._build_label_for_node..q s" B
)-DLLe $B
s {rN , rO zz()z: })r> attrsmethodsr r r9 join_get_method_argumentsrQ is_abstractrE returnsr _escape_annotation_label)
r1 rG r> rZ attrs_stringr[ funcargs method_nameannotation_labels
r rD z DotPrinter._build_label_for_nodef s~ %% #
(:(:(BL &++1r%44::;AA B
16B
eWAl^J,E,E,K,K+LAN ,6+=+=+C ;D99T 7 7 =>FFsERD)-)9)9);#dii[%DII;
QtfA. .E||#7 #E = =>N OOO 22889 :E ; r c * |j dd }|S )NrN rO rP )r1 re s r r` z#DotPrinter._escape_annotation_label s ,33C? r c t | }|j D cg c]
\ }}| d| d }}}|r|j d| d | j d| d| ddj t | d yc c}}w ) zACreate an edge from one node to another to display relationships.z="r7 zlabel="z" -> "z" [rV z];N)r( itemsappendr8 r\ sorted) r1 from_nodeto_noderF r>
arrowstylepropr9 rZ s r emit_edgezDotPrinter.emit_edge s E]
8B8H8H8JKuD6E7!$KK LL75'+ , Ai[wis499VE]3K2LBOP Ls A=c | j d}| j }|_d}t j d| \ }}t j d| \ }}t j
| t j
| net
| j j d }|s
d}|dz |z }||vr/t j d| \ }}t j
| n|}t |dd 5 }|j | j d d d ||vr3t j d d
||d |gd
t j | y y # 1 sw Y AxY w)N)dotgvpngz.gvz.png.wutf8)encodingrq z-Tz-oT)check) _close_graphr2 tempfilemkstemposcloser suffixlstripr&
writelineslines
subprocessrununlink) r1
outputfilegraphviz_extensionsrE targetpdotdot_sourcepathppngoutfiles r generatezDotPrinter.generate s= +zz F#+#3#3E4#@ D. ' / / = D* HHTN HHTN*%,,33C8F '# - 6
00'/'7'7t'D$n!+
.#
7 +7 tzz * + , , NNfndJGt
IIn % - + +s 3 EEc & | j d y)z2Emit the lines needed to properly close the graph.z}
N)r8 r; s r ry zDotPrinter._close_graph s %r )NN)r2 strr3 z
Layout | Noner4 z bool | None)returnNone)N)rE r rF r
rG zNodeProperties | Noner r )rG r r r )re r r r )
rk r rl r rF r r> z
str | Noner r )r r r r )
r r r rC r0 r< rK rD r` ro r ry
__classcell__)r5 s @r r* r* 6 s M
!%/3 AA A "- A 3 -1
*
0 > ! Q Q Q Q
Q
Q &8r r* )"__doc__
__future__r r| r rz enumr pathlibr astroidr pylint.pyreverse.printerr r r r
r pylint.pyreverse.utilsr r frozensetr __annotations__PACKAGECLASSr INHERITS ASSOCIATION AGGREGATIONUSESTYPE_DEPENDENCYr( r* r r r r s
U " X X 7* * $--N#O . O e NNH
V'B
MMf= *& .B Br