a =a"@svdZddlZddlmZddlmZmZddlmZGdddZ Gdd d ee Z Gd d d e Z Gd d d Z dS)zHhandle diagram generation options for class diagram or default diagrams N)BUILTINS) ClassDiagramPackageDiagram) LocalsVisitorc@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 Zhandlerr :lib/python3.9/site-packages/pylint/pyreverse/diadefslib.py__init__#szDiaDefGenerator.__init__cCs$|j}|jr |jd|}|S)zget title for objects.)name module_namesroot)r nodetitler r r get_title*szDiaDefGenerator.get_titlecCs|durt|jjS|S)z3activate some options if not explicitly deactivatedN)boolrclasses)r Zoptionr r r _set_option1s zDiaDefGenerator._set_optioncCs~||jj|_||jj}||jj}d\}}|r}|D]4}t|tjrF|j}t|tjr0| |s^q0|Vq0q(dS)z'return associated nodes of a class noderN) listZinstance_attrs_typevaluesZ locals_type isinstanceastroidZInstanceZ_proxiedZClassDefr!)r klass_noder%Zassociation_nodesrr r rget_associatedas  zDiaDefGenerator.get_associatedcCsp|j|s||sdS|||||D]}|||d|q0|||D]}||||dqTdS)z1extract recursively classes related to klass_nodeN)r Zhas_noder!r$r'extract_classesr-)r r,rrr&rr r rr/os zDiaDefGenerator.extract_classesN)__name__ __module__ __qualname____doc__rrrrrr!r$r'r-r/r r r rr s 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 r r rrszDefaultDiadefGenerator.__init__cCsF|jj}t|jdkr*td|j||_nd|_td|j||_dS)z_visit a pyreverse.utils.Project node create a diagram definition for packages r.z packages %sNz classes %s) rmodelenmodulesrr pkgdiagramrr )r rr6r r r visit_projects z$DefaultDiadefGenerator.visit_projectcCs|jr|j|jfS|jfS)z`leave the pyreverse.utils.Project node return the generated diagram definition )r9r r r r r leave_projects z$DefaultDiadefGenerator.leave_projectcCs&|jr"|j||j|j|dS)z_visit an astroid.Module node add this class to the package diagram definition N)r9r r"r#rr r r r visit_modules z#DefaultDiadefGenerator.visit_modulecCs|\}}||||dS)z\visit an astroid.Class node add this class to the class diagram definition N)rr/)r rrrr r rvisit_classdefs z%DefaultDiadefGenerator.visit_classdefcCs|jr|j||jdS)z?visit astroid.ImportFrom and catch modules for package diagramN)r9Zadd_from_dependmodnamer r r rvisit_importfromsz'DefaultDiadefGenerator.visit_importfromN) r0r1r2r3rr:r;r<r=r?r r r rr4|s   r4c@s eZdZdZddZddZdS)ClassDiadefGeneratorz[generate a class diagram definition including all classes related to a given class cCst|||dSr5)rrr r r rrszClassDiadefGenerator.__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 r.rrr) rrr6r r7r8rsplitZ get_modulesplitnextZilookuprr/)r projectklassmodulerrr r r class_diagrams   z"ClassDiadefGenerator.class_diagramN)r0r1r2r3rrGr r r rr@sr@c@s eZdZdZddZddZdS)DiadefsHandlerzYhandle diagram definitions : get it from user (i.e. xml files) or generate them cCs ||_dSr5)r)r rr r rrszDiadefsHandler.__init__cCsXg}t||}|jjD]}||||q|sBt|||}|D] }|qF|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`) )r@rrappendrGr4r"Zextract_relationships)r rDr Zdiagrams generatorrEZdiagramr r r get_diadefss    zDiadefsHandler.get_diadefsN)r0r1r2r3rrKr r r rrHsrH) r3r+Zpylint.constantsrZpylint.pyreverse.diagramsrrZpylint.pyreverse.utilsrrr4r@rHr r r rs  \7