eq
d d Z ddlmZ ddlmZmZmZmZ ddlm Z G d de Z
G d de
Z y )
z,Class to generate files in mermaidjs format. ) annotations)EdgeTypeNodePropertiesNodeTypePrinter)get_annotation_labelc
" e Zd ZU dZdZej dej diZde d< e
j de
j de
j de
j d e
j d
iZd e d < dd
Z d ddZ d ddZddZy)MermaidJSPrinterz Printer for MermaidJS diagrams.blackclasszdict[NodeType, str]NODESz--|>z--*z--oz-->z..>zdict[EdgeType, str]ARROWSc F | j d | j y)zEmit the header lines. classDiagramN)emit _inc_indentselfs D/usr/lib/python3/dist-packages/pylint/pyreverse/mermaidjs_printer.py _open_graphz MermaidJSPrinter._open_graph s .! Nc | t | }| j | }g }|j r|j |j |j r|j D ] }| j | }|j ddj | d}||j rdndz
}|j r|dt |j z
}|j | |j d d
}| j | d| d | j |D ] }| j | | j | j d y)
zVCreate a new node.
Nodes can be classes, packages, participants etc.
N)label(z, )* .z {})r r
attrsextendmethods_get_method_argumentsnamejoin is_abstractreturnsr appendsplitr r _dec_indent) r r& type_
propertiesnodetypebodyfuncargslines r emit_nodezMermaidJSPrinter.emit_node# s? 'd3J::e$ KK
(( ) "**
"11$7 ))Adiio%6a8t / / 1r9<<a 4T\\ B C DDD D!
" zz#r" XJavS)* D IIdO #r c |j d d }|j d d }| d| j | d| }|r|d| z
}| j | y)zACreate an edge from one node to another to display relationships.r r r z : N)r+ r r )r from_nodeto_noder- r edges r emit_edgezMermaidJSPrinter.emit_edgeD sh OOC(, --$R(Adkk% 0 17)< c%M !D $r c $ | j y)z2Emit the lines needed to properly close the graph.N)r, r s r _close_graphz MermaidJSPrinter._close_graphS s r returnNoneN)r& strr- r r. zNodeProperties | Noner= r> )
r6 r@ r7 r@ r- r r z
str | Noner= r> )__name__
__module__ __qualname____doc__
DEFAULT_COLORr CLASSPACKAGEr
__annotations__r INHERITS ASSOCIATION AGGREGATIONUSESTYPE_DEPENDENCYr r r4 r9 r; r r r
r
s )M '"E
6 e e
u % #F -1 *
L !
r r
c 8 e Zd ZdZdZdZdZd fd ZddZ xZ S )HTMLMermaidJSPrinterz=Printer for MermaidJS diagrams wrapped in a html boilerplate.z
z!
c | j | j t | j D ] }| j t
| y r? )r HTML_OPEN_BOILERPLATErangeGRAPH_INDENT_LEVELr superr )r _ __class__s r r z HTMLMermaidJSPrinter._open_graphg sF $,,-t../ A
r c t | j D ] }| j | j | j y r? )rT rU r, r HTML_CLOSE_BOILERPLATE)r rW s r r; z!HTMLMermaidJSPrinter._close_graphm s: t../ A $--.r r< )
rA rB rC rD rS rZ rU r r;
__classcell__)rX s @r rP rP X s( G
/r rP N) rD
__future__r pylint.pyreverse.printerr r r r pylint.pyreverse.utilsr r
rP rN r r r_ s2
3 " P P 7Hw HV/+ /r