;:l d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
Z
ddl m Z m
Z
mZmZmZ ddlmZmZmZmZ ddlmZ dd lmZ G d
d ZdS ) z,Utilities for creating VCG and Dot diagrams. ) annotationsN)Iterable)modutilsnodes) ClassDiagram ClassEntity
DiagramEntityPackageDiagram
PackageEntity)EdgeTypeNodePropertiesNodeTypePrinter)get_printer_for_filetype) is_exceptionc Z e Zd ZdZd dZd d Zd d Zd dZd!dZd"dZ d#dZ
d$dZ d%dZ d S )&
DiagramWriterz(Base class for writing project diagrams.configargparse.NamespacereturnNonec || _ t | j j | _ | d| _ | j j | _ t j | j j | _
i | _ d S )N ) r r
output_format
printer_class file_namemax_color_depthdepth itertoolscycle
color_paletteavailable_colors used_colors)selfr s i/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/pylint/pyreverse/writer.py__init__zDiagramWriter.__init__ s\ 5dk6O P P [0
) 0I J J +- diadefs'Iterable[ClassDiagram | PackageDiagram]c 2 |D ]}|j dd dd }| d| j j }t
j | j j r*t
j | j j | }|
|| t |t r|
| n| | | dS )z1Write files for according to ./_ .N)titlestripreplacer r ospathexistsoutput_directoryjoin set_printer
isinstancer
write_packages
write_classessave)r$ r( diagrambasenamer s r% writezDiagramWriter.write) s
G }**,,44S#>>FFsCPPH#AAdk&?AAIw~~dk :;;
R GLL)Ey Q Q Y 1 1 1'>22
,##G,,,,""7+++ IIKKKK
r' r< r
c t | d D ]_}|j |_ | j |j t j | | `|
d D ]=}| j |j j |j
j t j >dS )zWrite a package diagram.c | j S Nr/ xs r% z.DiagramWriter.write_packages..: s ag r' keytype_
propertiesdependsrI N)sortedmodulesnodeqnamefig_idprinter emit_noder PACKAGEget_package_propertiesget_relationships emit_edge from_object to_objectr USES)r$ r< modulerels r% r9 z DiagramWriter.write_packages7 s W__ . .4E4EFFF F "K - - / /FM L " "
& 66v>>
#
,,Y77 C L " "&
$ m
#
r' r c t |j d D ]x}|j |_ |j dk r t j n t j }| j
|j || | y| d D ]=}| j
|j j |j j t j >| d D ]=}| j
|j j |j j t j >| d D ]C}| j
|j j |j j |j t j D| d
D ]C}| j
|j j |j j |j t j Dd S ) zWrite a class diagram.c | j S rA rB rC s r% rE z-DiagramWriter.write_classes..L s r' rF interfacerH specializationrL
implements association)labelrI aggregationN)rM objectsrO rP rQ shaper INTERFACECLASSrR rS get_class_propertiesrV rW rX rY r INHERITS
IMPLEMENTSname ASSOCIATION AGGREGATION)r$ r< objrI r\ s r% r: zDiagramWriter.write_classesI s '//@/@AAA C ))CJ*-){*B*BH&&E L " "
%D4M4Mc4R4R
#
,,-=>> C L " "&
$ '
#
,,\:: C L " "&
$ )
#
,,];; C L " "&
$h *
#
,,];; C L " "&
$h *
#
r' r strr= c H | | | _ || _ dS )z Set printer.N)r rR r )r$ r r= s r% r7 zDiagramWriter.set_printerq s" ))(33 "r' ro r r
c p t |j | j j r| | nd S )z!Get label and shape for packages.black)rc color)r
r/ r colorizedget_shape_color)r$ ro s r% rU z$DiagramWriter.get_package_propertiesv s> )/3{/DQ$&&s+++'
r' r c t |j | j j s|j nd| j j s|j ndt
|j rdnd| j j r| | nd }|S )z Get label and shape for classes.Nredrs )rc attrsmethods fontcolorrt )
r
r/ r only_classnamesry rz r rO ru rv )r$ ro rJ s r% ri z"DiagramWriter.get_class_properties} s #)#';#>H#))D'+{'BLCKK +CH 5 5Bee7/3{/DQ$&&s+++'
r' r c B |j }t j | dd d rdS t |j t j r | dd d }n,|j j r|}n | dd d }d
| d| j d| j }|| j vr t | j | j |< | j | S )zGet shape color.r. )maxsplitr grey N)rO rP r is_stdlib_modulesplitr8 r ClassDefrsplitpackager6 r r# nextr" )r$ ro qualified_namer base_names r% rv z DiagramWriter.get_shape_color s )) $^%9%9#%9%J%J1%M N N 6 ch / / @$++C!+< r9 r: r7 rU ri rv r; r' r% r r s 22. . . . $& & & &P# # # #
+ + + + . . . . . .r' r )r
__future__r argparser r2 collections.abcr astroidr r pylint.pyreverse.diagramsr r r r
r pylint.pyreverse.printerr r
r r pylint.pyreverse.printer_factoryr pylint.pyreverse.utilsr r r r' r% r s4
3 2 " " " " " " $ $ $ $ $ $ # # # # # # # # Q P P P P P P P P P P P E E E E E E / / / / / /~. ~. ~. ~. ~. ~. ~. ~. ~. ~.r'