yIfq
z 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 diZded< e
j ddde
j ddddde
j ddddde
j ddde
j dddd iZ 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 P/var/www/piapp/venv/lib/python3.11/site-packages/pylint/pyreverse/dot_printer.pyr r s )NNNr 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 j e Zd ZdZ d d! fd
Zd"d Z d#d$dZd%dZd&dZ d#d'd Z d(d Z
d"d Z xZ S ))
DotPrinterblackNtitlestrlayout
Layout | Noneuse_automatic_namespace bool | Nonec x |p t j }d| _ t ||| d S )Nr )r
BOTTOM_TO_TOPcharsetsuper__init__)selfr, r. r0 __class__s r r6 zDotPrinter.__init__9 s< /6/
(?@@@@@r returnNonec F | d| j d | j r"| d| j j | j rR| j t v sJ d| j | d| j d dS dS )zEmit the header lines.z digraph "z" {zrankdir=zunsupported charset z charset=""N)emitr, r. valuer4 lowerr r7 s r _open_graphzDotPrinter._open_graphC s .dj.../// ; 6 II4!244 5 5 5 < 3 ""$$(88884dl44 988 II1$,111 2 2 2 2 2 3 3r nametype_r
propertiesNodeProperties | Nonec @ |t | }t | }|j |j n| j }|| j k rdnd}| | }|rd| dnd}|j r d|j d nd} | d | d
| d | | d | d | d
dS )zVCreate a new node.
Nodes can be classes, packages, participants etc.
N)labelfilledr" z , label=<> z
, fontcolor="r< z
" [color="z , shape="z
", style="z"];)r r color
DEFAULT_COLOR_build_label_for_noder# r= )
r7 rB rC rD shaperK r$ rG
label_partfontcolor_parts
r emit_nodezDotPrinter.emit_nodeN s 'd333Ju
$.$4$@
dFX !T%7 7 7W**: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 |j sdS |j }|j |j |S |j pg }t j j d |D }d| d| t j j d}|j pg }|D ]}d | | dd }| r d|j
dn|j
}|| d | d
z
}|j r/t |j } |d |
| z z
}|t j j z
}|d z
}|S )
NrJ c 3 B K | ]}| d d V dS )|\|Nreplace).0attrs r z3DotPrinter._build_label_for_node..q sI B
B
)-DLLe $ $B
B
B
B
B
B
r {rT , rU zz()z: })rG attrsmethodsr r r> join_get_method_argumentsrW is_abstractrB returnsr _escape_annotation_label)
r7 rD rG r` attrs_stringra funcargs method_nameannotation_labels
r rM z DotPrinter._build_label_for_nodef s 2 % #
(:(BL &+1r%4:<AA B
B
16B
B
B
OeNNlNJ,E,KNNN ,6+=+C ; ;D99T 7 7 = =>>FFsERRD)-)9)9););O%di%%%%DI
..t... .E|
P#7 #E#E = =>N O OOO 28: :EE
r rk c 2 | dd }|S )NrT rU rV )r7 rk s r rf z#DotPrinter._escape_annotation_label s ,33C?? r from_nodeto_noder rG
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 )rX propr> s r
z(DotPrinter.emit_edge.. s- KKKuD$$E$$$KKKr zlabel="r< z" -> "z" [r\ z];N)r( itemsappendr= rb sorted)r7 rm rn rC rG
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)encodingr{ z-Tz-oT)check) _close_graphr, tempfilemkstemposcloser suffixlstripr&
writelineslines
subprocessrununlink) r7 ry graphviz_extensionsrB targetpdotdot_sourcepathppngoutfiles r generatezDotPrinter.generate s +z F#+#3E4#@#@ D. ' / = = D* HTNNN HTNNNN*%%,33C88F
7 '# - 6
000'/'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= r@ s r r zDotPrinter._close_graph s %r )NN)r, r- r. r/ r0 r1 )r9 r: )N)rB r- rC r
rD rE r9 r: )rD r r9 r- )rk r- r9 r- )
rm r- rn r- rC r rG ro r9 r: )ry r- r9 r: )
r r r rL r6 rA rQ rM rf rx r r
__classcell__)r8 s @r r* r* 6 s M
!%/3 A A A A A A A 3 3 3 3 -1
0 > ! 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__PACKAGECLASSr INHERITS ASSOCIATION AGGREGATIONUSESTYPE_DEPENDENCYr( 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
Mf== * .B B B B B B B B B Br