o gW @s"UdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZddlmZmZedZe eed <ejd ejd ejd iZeeefed <eje d ddej!e ddddej"e dd dddej#e d ddiZ$eeefed<GdddeZ%dS)zP Class to generate files in dot format and image formats supported by Graphviz. N)Path)Dict FrozenSetListOptional)nodes)EdgeTypeLayoutNodePropertiesNodeTypePrinter)check_graphviz_availabilityget_annotation_label)utf-8z iso-8859-1latin1ALLOWED_CHARSETSboxrecordSHAPESnoneempty) arrowtail arrowheadnodedashed)rrstylegreendiamondsolid) fontcolorrrropenARROWSc seZdZdZ  ddedeedeeffdd Zdd d Z d d ed e d ee ddfddZ d!d e deedefddZ d deded edeeddf ddZdeddfddZdddZZS)" DotPrinterblackNtitlelayoutuse_automatic_namespacecs$|ptj}d|_t|||dS)Nr)r BOTTOM_TO_TOPcharsetsuper__init__)selfr$r%r& __class__Z/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pylint/pyreverse/dot_printer.pyr*,s zDotPrinter.__init__returncCsn|d|jd|jr|d|jj|jr5|jtvs)Jd|j|d|jddSdS)zEmit the header linesz digraph "z" {zrankdir=zunsupported charset z charset=""N)emitr$r%valuer(lowerrr+r.r.r/ _open_graph6s zDotPrinter._open_graphnametype_ propertiesc Cs|dur t|d}t|}|jdur|jn|j}||jkrdnd}||}|r.d|dnd}|jr:d|jdnd} |d|d |d| |d |d |d dS) zDCreate a new node. Nodes can be classes, packages, participants etc.N)labelfilledrz , label="r1z , fontcolor="z " [color="z , shape="z ", style="z"];)r rcolor DEFAULT_COLOR_build_label_for_noderr2) r+r7r8r9shaper=rr: label_partfontcolor_partr.r.r/ emit_nodeAs  &zDotPrinter.emit_nodeF is_interfacecCs|jsdS|j}|rd|}|jdur|jdur|S|jpg}d|dd|d}|jp0g}|D]%}||}||jdd|d 7}|jrT|d t|j7}|d7}q3|d 7}|S) Nr<z<>\n{|z\lz\l|(, )z: })r:attrsmethodsjoin_get_method_argumentsr7returnsr)r+r9rDr:rKrLfuncargsr.r.r/r?Vs$    z DotPrinter._build_label_for_node from_nodeto_noder:c Cs\t|}dd|D}|r|d|d|d|d|ddt|dd S) zACreate an edge from one node to another to display relationships.cSs g|] \}}|d|dqS)z="r1r.).0propr3r.r.r/ }s z(DotPrinter.emit_edge..zlabel="r1z" -> "z" [rHz];N)r!itemsappendr2rMsorted)r+rRrSr8r: arrowstylerKr.r.r/ emit_edgets ,zDotPrinter.emit_edge outputfilec Cs|d}|j}|dur*d}td|\}}td|\}}t|t|n&t|jd}|sstrrr boolr*r6r r rCr?rr[r{rf __classcell__r.r.r,r/r")sX   # r")&__doc__rirqrorgpathlibrtypingrrrrastroidrpylint.pyreverse.printerrr r r r pylint.pyreverse.utilsr r frozensetrr__annotations__PACKAGE INTERFACECLASSrINHERITSdict IMPLEMENTS ASSOCIATIONUSESr!r"r.r.r.r/s.