o g#@sdZddlmZmZddlZddlmZddlmZmZddl m Z ddl m Z Gdd d Z Gd d d e e ZGd d d e ZGdddZdS)zHhandle diagram generation options for class diagram or default diagrams )AnyOptionalN)nodes) ClassDiagramPackageDiagram)Project) 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 handlerrY/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pylint/pyreverse/diadefslib.py__init__'s zDiaDefGenerator.__init__cCs$|j}|jr|jd|}|S)zget title for objects.)name module_namesroot)rnodetitlerrr get_title.szDiaDefGenerator.get_titlecCs|dur t|jjS|S)z3activate some options if not explicitly deactivatedN)boolr classes)roptionrrr _set_option5s zDiaDefGenerator._set_optioncCs~||jj|_||jj}||jj}d\}}|rd}|r"d}|jjdur,|jj}|jjdur6|jj}|||_|_dS)z6set different default options with _default dictionary)rrN) rr r all_ancestorsall_associatedshow_ancestorsshow_associated anc_levelassociation_level)rr r!r$r%rrrr =s  z$DiaDefGenerator._set_default_optionscCs |j|jfS)zhelp function for search levels)r$r%)rrrr _get_levelsMs zDiaDefGenerator._get_levelscCs|jjrdS|jdkS)z&true if builtins and not show_builtinsTbuiltins)r show_builtinrrrrrrr show_nodeQszDiaDefGenerator.show_nodecCs$|j||j|||dS)z%visit one class and add it to diagramN)r visitr add_objectrr)rrr add_classWs zDiaDefGenerator.add_classccs6|dkrdS|jddD] }||sq |Vq dS)z%return ancestor nodes of a class noderNF)recurs) ancestorsr*)rrlevelancestorrrr get_ancestors\s zDiaDefGenerator.get_ancestorsccsn|dkrdSt|jt|jD]}|D]}t|tjr$|j}t|tj r/| |s0q|VqqdS)z'return associated nodes of a class noderN) listinstance_attrs_typevalues locals_type isinstanceastroidInstance_proxiedrClassDefr*)r klass_noder0association_nodesrrrrget_associatedes  zDiaDefGenerator.get_associatedcCsp|j|s ||s dS|||||D] }|||d|q|||D] }||||dq*dS)z1extract recursively classes related to klass_nodeN)r has_noder*r-r2extract_classesr>)rr<r$r%r1rrrrrAss zDiaDefGenerator.extract_classesN)__name__ __module__ __qualname____doc__rrrr r&r*r-r2r>rArrrrr $s r c@sxeZdZdZddZdeddfddZd edefd d Zde j ddfd d Z de j ddfddZ de jddfddZdS)DefaultDiadefGeneratorzgenerate minimum diagram definition for the project : * a package diagram including project's modules * a class diagram including project's classes cCst|||t|dSN)r rrrrrrrszDefaultDiadefGenerator.__init__rreturnNcCsJ|jj}t|jdkrtd|j||_nd|_td|j||_dS)z_visit a pyreverse.utils.Project node create a diagram definition for packages r?z packages Nzclasses ) r modelenmodulesrr pkgdiagramrr )rrrIrrr visit_projects z$DefaultDiadefGenerator.visit_project_cCs|jr |j|jfS|jfS)z`leave the pyreverse.utils.Project node return the generated diagram definition )rLr )rrNrrr 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)rLr 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&rA)rrr$r%rrrvisit_classdefs z%DefaultDiadefGenerator.visit_classdefcCs|jr |j||jdSdS)z?visit astroid.ImportFrom and catch modules for package diagramN)rLadd_from_dependmodnamer)rrrvisit_importfromsz'DefaultDiadefGenerator.visit_importfrom)rBrCrDrErrrMrrOrModulerPr;rQ ImportFromrTrrrrrFs  rFc@ eZdZdZddZddZdS)ClassDiadefGeneratorz[generate a class diagram definition including all classes related to a given class cCst|||dSrG)r rrrrrrszClassDiadefGenerator.__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) rr rIr rJrKrsplit get_modulesplitnextilookupr&rA)rprojectklassmoduler$r%rrr class_diagrams   z"ClassDiadefGenerator.class_diagramN)rBrCrDrErrarrrrrXs rXc@rW)DiadefsHandlerzYhandle diagram definitions : get it from user (i.e. xml files) or generate them cCs ||_dSrG)r )rr rrrrs 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`) )rXr rappendrarFr+extract_relationships)rr^r diagrams generatorr_diagramrrr get_diadefss   zDiadefsHandler.get_diadefsN)rBrCrDrErrhrrrrrbs rb)rEtypingrrr8rpylint.pyreverse.diagramsrrpylint.pyreverse.inspectorrpylint.pyreverse.utilsrr rFrXrbrrrrs   \9