8
| d Z ddlmZ ddlZddlmZ ddlmZ ddlZddlm Z ddl
m Z m Z ddl
mZmZ dd lmZ G d
d Z G d d
ee Z G d de Z G d d ZdS )zHHandle diagram generation options for class diagram or default diagrams. ) annotationsN) Generator)Any)nodes) ClassDiagramPackageDiagram)LinkerProject)
LocalsVisitorc b e Zd ZdZd dZd!d Zd"dZd#dZd$dZd%dZ d&dZ
d'dZ d(dZ d)d Z
d S )*DiaDefGeneratorz"Handle diagram generation options.linkerr handlerDiadefsHandlerreturnNonec f |j | _ d| _ | || _ | dS )z&Common Diagram Handler initialization.FN)config module_names_set_default_optionsr selfr r s m/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/pylint/pyreverse/diadefslib.py__init__zDiaDefGenerator.__init__ s8 n "' !!### ''' nodenodes.ClassDefstrc ^ |j }| j r | j d| }|S )zGet title for objects..)namer root)r r titles r get_titlezDiaDefGenerator.get_title" s7 2yy{{'11%11E r option bool | Noneboolc < |t | j j S |S )z4Activate some options if not explicitly deactivated.)r' r classes)r r% s r _set_optionzDiaDefGenerator._set_option) s! > +,, ,
r c b | | j j | _ | | j j }| | j j }d\ }}|rd}|rd}| j j | j j }| j j | j j }||c| _ | _ dS )z7Set different default options with _default dictionary.)r r N) r* r r
all_ancestorsall_associatedshow_ancestorsshow_associated anc_levelassociation_level)r r- r. r1 r2 s r r z$DiaDefGenerator._set_default_options1 s , ,T[-E F F (()BCC
))$+*DEE'-$ $ I # " ; % 1 2I ; & 2 $ ; 1:>(++
NNNN r
klass_nodec # n K |dk rdS t |j t |j z D ]\}|D ]W}t |t
j r|j }t |t j r|
| sS|V X]dS )z(Return associated nodes of a class node.r N) listinstance_attrs_typevalues locals_type
isinstanceastroidInstance_proxiedr ClassDefr: )r rH r@ association_nodesr s r get_associatedz DiaDefGenerator.get_associated[ s A:: F!%j&D&K&K&M&M!N!NQU " ) ) + +R
R
"
*
dG$455 ) =D"4 88 T^^D=Q=Q
r r1 r2 c ^ | j | s| | sdS | | | || D ] }| ||dz
| | || D ] }| |||dz
dS )z2Extract recursively classes related to klass_node.N )r= has_noder: r? rG extract_classesrT )r rH r1 r2 rF r s r rX z DiaDefGenerator.extract_classesk s % %j 1 1
9S9S F z""" **:yAA M MH 9q=:K L L L L''
4EFF I ID y2Ca2G H H H H I Ir Nr r r r r r )r r r r )r% r& r r' )r r )r r3 )r r r r' r r r r )r r r@ rA r rB )rH r r@ rA r rB )rH r r1 rA r2 rA r r )__name__
__module__ __qualname____doc__r r$ r* r r5 r: r? rG rT rX r r r
r
s ,,( ( ( ( N N N N 6 6 6 6. . . . A A A A
I I I I I Ir r
c B e Zd ZdZddZdd ZddZddZddZddZ dS ) DefaultDiadefGeneratorzGenerate minimum diagram definition for the project :
* a package diagram including project's modules
* a class diagram including project's classes
r r r r r r c f t | || t j | d S N)r
r r r s r r z DefaultDiadefGenerator.__init__ s1 vw777 t$$$$$r r r
c | j j }t |j dk r t d|j | | _ nd| _ t d|j | | _ dS )z`Visit a pyreverse.utils.Project node.
create a diagram definition for packages
rV z packages Nzclasses ) r modelenmodulesr r!
pkgdiagramr r= )r r re s r
visit_projectz$DefaultDiadefGenerator.visit_project sq
{ t| q 5C'DI''6 6DOO #DO ()?DI)?)? F Fr _r c < | j r| j | j fS | j fS )zaLeave the pyreverse.utils.Project node.
return the generated diagram definition
)rh r= )r rj s r
leave_projectz$DefaultDiadefGenerator.leave_project s*
? 6?D$55 5!##r nodes.Modulec | j r<| j | | j |j | dS dS )z`Visit an astroid.Module node.
add this class to the package diagram definition
N)rh r r< r> r! r9 s r visit_modulez#DefaultDiadefGenerator.visit_module sN
? 8 K d # # # O & &ty$ 7 7 7 7 7 8 8r r c b | \ }}| ||| dS )z]Visit an astroid.Class node.
add this class to the class diagram definition
N)r5 rX )r r r1 r2 s r visit_classdefz%DefaultDiadefGenerator.visit_classdef s:
(,'7'7'9'9$ $ T9.?@@@@@r nodes.ImportFromc X | j r"| j ||j dS dS )z@Visit astroid.ImportFrom and catch modules for package diagram.N)rh add_from_dependmodnamer9 s r visit_importfromz'DefaultDiadefGenerator.visit_importfrom s9 ? @ O + +D$, ? ? ? ? ? @ @r NrY )r r
r r )rj r
r r )r rm r r rZ )r rr r r )
r[ r\ r] r^ r ri rl ro rq rv r_ r r ra ra z s % % % % G G G G $ $ $ $8 8 8 8A A A A@ @ @ @ @ @r ra c e Zd ZdZd
dZd S ) ClassDiadefGeneratorz\Generate a class diagram definition including all classes related to a
given class.
projectr
klassr r r c t || j j | _ t |j dk r/| dd \ }}| | }n(|j d }| d d }t |
| }| \ }}| ||| | j S )zDReturn a class diagram definition for the class and related classes.rV r r r, )
r r re r= rf rg rsplit
get_modulesplitnextilookupr5 rX )r ry rz moduler1 r2 s r
class_diagramz"ClassDiadefGenerator.class_diagram s ( 0@ A A w ! # # !LLa 0 0MFE ''//FF _Q'FKK$$R(EV^^E**++'+'7'7'9'9$ $ UI/@AAA r N)ry r
rz r r r )r[ r\ r] r^ r r_ r r rx rx s2
!
!
!
!
!
!r rx c " e Zd ZdZddZdd Zd
S )r zDGet diagram definitions from user (i.e. xml files) or generate them.r argparse.Namespacer r c || _ d S rc )r )r r s r r zDiadefsHandler.__init__ s
r ry r
r r list[ClassDiagram]c g }t || }| j j D ]+}| | || ,|s#t || | }|D ]}| |S )ax Get the diagram's configuration data.
:param project:The pyreverse project
:type project: pyreverse.utils.Project
:param linker: The linker
:type linker: pyreverse.inspector.Linker(IdGeneratorMixIn, LocalsVisitor)
:returns: The list of diagram definitions
:rtype: list(:class:`pylint.pyreverse.diagrams.ClassDiagram`)
)rx r r) appendr ra r< extract_relationships)r ry r diagrams generatorrz diagrams r get_diadefszDiadefsHandler.get_diadefs s (66 [( E EE OOI 3 3GU C C D D D D K-fd;;AA'JJH , ,G ) ) + + + +r N)r r r r )ry r
r r r r )r[ r\ r] r^ r r r_ r r r r sB NN r r )r^
__future__r argparsecollections.abcr typingr rO r pylint.pyreverse.diagramsr r pylint.pyreverse.inspectorr r
pylint.pyreverse.utilsr r
ra rx r r_ r r r st
O N " " " " " " % % % % % % B B B B B B B B 6 6 6 6 6 6 6 6 0 0 0 0 0 0
`I `I `I `I `I `I `I `IF6@ 6@ 6@ 6@ 6@]O 6@ 6@ 6@r! ! ! ! !? ! ! !0 r