o b!@sndZddlZddlmZmZddlmZdZGdddZGdd d eeZ Gd d d eZ Gd d d Z dS)zHhandle diagram generation options for class diagram or default diagrams N)PackageDiagram ClassDiagram) LocalsVisitorbuiltinsc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)DiaDefGeneratorz!handle diagram generation optionscCs |j|_|||_d|_dS)z%common Diagram Handler initializationN)config_set_default_optionslinker classdiagramselfr handlerrZ/var/www/secure340b-portal/env/lib/python3.10/site-packages/pylint/pyreverse/diadefslib.py__init__!s zDiaDefGenerator.__init__cCs"|j}|jrd|j|f}|S)zget title for objectsz%s.%s)name module_namesroot)r nodetitlerrr get_title(szDiaDefGenerator.get_titlecCs|dur t|jjS|S)z3activate some options if not explicitly deactivatedN)boolrclasses)r optionrrr _set_option/s zDiaDefGenerator._set_optioncCs~||jj|_||jj}||jj}d\}}|rd}|r"d}|jjdur,|jj}|jjdur6|jj}|||_|_dS)z6set different default options with _default dictionary)rrN) rrr all_ancestorsall_associatedZshow_ancestorsZshow_associated anc_levelassociation_level)r rrrrrrrr7s  z$DiaDefGenerator._set_default_optionscCs |j|jfS)zhelp function for search levels)rr)r rrr _get_levelsGs zDiaDefGenerator._get_levelscCs|jjrdS|jtkS)z&true if builtins and not show_builtinsT)rZ show_builtinrr BUILTINS_NAMEr rrrr show_nodeKszDiaDefGenerator.show_nodecCs$|j||j|||dS)z%visit one class and add it to diagramN)r visitr add_objectrr"rrr add_classQs zDiaDefGenerator.add_classccs6|dkrdS|jddD] }||sq |Vq dS)z%return ancestor nodes of a class noderNF)Zrecurs)Z ancestorsr#)r rlevelancestorrrr get_ancestorsVs zDiaDefGenerator.get_ancestorsccsn|dkrdSt|jt|jD]}|D]}t|tjr$|j}t|tjr/| |s0q|VqqdS)z'return associated nodes of a class noderN) listZinstance_attrs_typevaluesZ locals_type isinstanceastroidInstanceZ_proxiedClassDefr#)r klass_noder'Zassociation_nodesrrrrget_associated_s  zDiaDefGenerator.get_associatedcCsp|j|s ||s dS|||||D] }|||d|q|||D] }||||dq*dS)z1extract recursively classes related to klass_nodeN)r Zhas_noder#r&r)extract_classesr1)r r0rrr(rrrrr3ms zDiaDefGenerator.extract_classesN)__name__ __module__ __qualname____doc__rrrrr r#r&r)r1r3rrrrrs rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)DefaultDiadefGeneratorzgenerate minimum diagram definition for the project : * a package diagram including project's modules * a class diagram including project's classes cCst|||t|dSN)rrrr rrrrszDefaultDiadefGenerator.__init__cCsF|jj}t|jdkrtd|j||_nd|_td|j||_dS)z_visit a pyreverse.utils.Project node create a diagram definition for packages r2z packages %sNz classes %s) rmodelenmodulesrr pkgdiagramrr )r rr:rrr visit_projects z$DefaultDiadefGenerator.visit_projectcCs|jr |j|jfS|jfS)z`leave the pyreverse.utils.Project node return the generated diagram definition )r=r r"rrr leave_projects z$DefaultDiadefGenerator.leave_projectcCs*|jr|j||j|j|dSdS)z_visit an astroid.Module node add this class to the package diagram definition N)r=r r$r%rr"rrr visit_modules z#DefaultDiadefGenerator.visit_modulecCs|\}}||||dS)z\visit an astroid.Class node add this class to the class diagram definition N)r r3)r rrrrrrvisit_classdefs z%DefaultDiadefGenerator.visit_classdefcCs|jr |j||jdSdS)zHvisit astroid.ImportFrom and catch modules for package diagram N)r=Zadd_from_dependmodnamer"rrrvisit_importfromsz'DefaultDiadefGenerator.visit_importfromN) r4r5r6r7rr>r?r@rArCrrrrr8zs   r8c@ eZdZdZddZddZdS)ClassDiadefGeneratorz[generate a class diagram definition including all classes related to a given class cCst|||dSr9)rrr rrrrszClassDiadefGenerator.__init__cCst||jj|_t|jdkr|dd\}}||}n |jd}|dd}t | |}| \}}| ||||jS)z^return a class diagram definition for the given klass and its related klasses r2.rr) rrr:r r;r<rsplitZ get_modulesplitnextZilookupr r3)r projectklassmodulerrrrr class_diagrams   z"ClassDiadefGenerator.class_diagramN)r4r5r6r7rrMrrrrrEs rEc@rD)DiadefsHandlerzYhandle diagram definitions : get it from user (i.e. xml files) or generate them cCs ||_dSr9)r)r rrrrrs zDiadefsHandler.__init__cCsXg}t||}|jjD] }||||q |s!t|||}|D]}|q#|S)avGet the diagrams 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`) )rErrappendrMr8r$Zextract_relationships)r rJr Zdiagrams generatorrKdiagramrrr get_diadefss   zDiadefsHandler.get_diadefsN)r4r5r6r7rrRrrrrrNs rN) r7r-Zpylint.pyreverse.diagramsrrZpylint.pyreverse.utilsrr!rr8rErNrrrrs \8