yIf"dZddlmZddlZddlmZddlmZddlZddlm Z ddl m Z ddl m Z mZdd lmZmZdd lmZGd d ZGd deeZGddeZGddZdS)zHHandle diagram generation options for class diagram or default diagrams.) annotationsN) Generator)Any)nodes)is_stdlib_module) ClassDiagramPackageDiagram)LinkerProject) LocalsVisitorcbeZdZdZd dZd!d Zd"dZd#dZd$dZd%dZ d&dZ d'dZ d(dZ d)dZ dS)*DiaDefGeneratorz"Handle diagram generation options.linkerr handlerDiadefsHandlerreturnNonecf|j|_d|_|||_|dS)z&Common Diagram Handler initialization.FN)config module_names_set_default_optionsrselfrrs O/var/www/piapp/venv/lib/python3.11/site-packages/pylint/pyreverse/diadefslib.py__init__zDiaDefGenerator.__init__s8n "' !!###  '''nodenodes.ClassDefstrc^|j}|jr|jd|}|S)zGet title for objects..)namerroot)rrtitles r get_titlezDiaDefGenerator.get_title#s7    2yy{{'11%11E roption bool | Noneboolc<|t|jjS|S)z4Activate some options if not explicitly deactivated.)r(rclasses)rr&s r _set_optionzDiaDefGenerator._set_option*s! > +,, , rcb||jj|_||jj}||jj}d\}}|rd}|rd}|jj |jj}|jj |jj}||c|_|_dS)z7Set different default options with _default dictionary.)rrN) r+rr all_ancestorsall_associatedshow_ancestorsshow_associated anc_levelassociation_level)rr.r/r2r3s rrz$DiaDefGenerator._set_default_options2s ,,T[-EFF(()BCC ))$+*DEE'-$ $  I  # "  ; % 1 2I ; & 2 $ ; 1:>(++ NNNN  r klass_nodec#nK|dkrdSt|jt|jzD]\}|D]W}t |t jr|j}t |tj r| |sS|VX]dS)z(Return associated nodes of a class node.rN) listinstance_attrs_typevalues locals_type isinstanceastroidInstance_proxiedrClassDefr<)rrJrBassociation_nodesrs rget_associatedzDiaDefGenerator.get_associated`s A:: F!%j&D&K&K&M&M!N!NQU  " ) ) + +R R "    *  dG$455)=D"488T^^D=Q=Q    rr2r3c^|j|s||sdS|||||D]}|||dz ||||D]}||||dz dS)z2Extract recursively classes related to klass_node.N)r?has_noder<rArIextract_classesrV)rrJr2r3rHrs rrZzDiaDefGenerator.extract_classesps   % %j 1 1  9S9S  F z"""**:yAA M MH  9q=:K L L L L'' 4EFF I ID  y2Ca2G H H H H I IrNrr rrrr)rrrr)r&r'rr()rr)rr4)rrrr(rrrr)rrrBrCrrD)rJrrBrCrrD)rJrr2rCr3rCrr)__name__ __module__ __qualname____doc__rr%r+rr6r<rArIrVrZrrrrs,,((((NNNN 6666AAAA     I I I I I IrrcBeZdZdZddZdd ZddZddZddZddZ dS)DefaultDiadefGeneratorzGenerate minimum diagram definition for the project : * a package diagram including project's modules * a class diagram including project's classes rr rrrrcft|||tj|dSN)rrr rs rrzDefaultDiadefGenerator.__init__s1  vw777t$$$$$rrr c|jj}t|jdkrt d|j||_nd|_td|j||_dS)z`Visit a pyreverse.utils.Project node. create a diagram definition for packages rXz packages Nzclasses ) rmodelenmodulesr r" pkgdiagramrr?)rrrgs r visit_projectz$DefaultDiadefGenerator.visit_projectsq { t|  q 5C'DI''66DOO#DO()?DI)?)?FFr_rc<|jr|j|jfS|jfS)zaLeave the pyreverse.utils.Project node. return the generated diagram definition )rjr?)rrls r leave_projectz$DefaultDiadefGenerator.leave_projects* ? 6?D$55 5!##r nodes.Modulec|jr<|j||j|j|dSdS)z`Visit an astroid.Module node. add this class to the package diagram definition N)rjrr>r@r"r;s r visit_modulez#DefaultDiadefGenerator.visit_modulesN ? 8 K  d # # # O & &ty$ 7 7 7 7 7 8 8rrcb|\}}||||dS)z]Visit an astroid.Class node. add this class to the class diagram definition N)r6rZ)rrr2r3s rvisit_classdefz%DefaultDiadefGenerator.visit_classdefs: (,'7'7'9'9$ $ T9.?@@@@@rnodes.ImportFromcX|jr"|j||jdSdS)z@Visit astroid.ImportFrom and catch modules for package diagram.N)rjadd_from_dependmodnamer;s rvisit_importfromz'DefaultDiadefGenerator.visit_importfroms9 ? @ O + +D$, ? ? ? ? ? @ @rNr[)rr rr)rlr rr)rrorrr\)rrtrr) r]r^r_r`rrkrnrqrsrxrarrrcrcs %%%% G G G G$$$$8888AAAA@@@@@@rrcceZdZdZd dZd S) ClassDiadefGeneratorz\Generate a class diagram definition including all classes related to a given class. projectr klassrrrct||jj|_t |jdkr/|dd\}}||}n(|jd}|dd}t| |}| \}}| ||||jS)zDReturn a class diagram definition for the class and related classes.rXr!rr-) rrrgr?rhrirsplit get_modulesplitnextilookupr6rZ)rr{r|moduler2r3s r class_diagramz"ClassDiadefGenerator.class_diagrams( 0@AA w  ! # #!LLa00MFE''//FF_Q'FKK$$R(EV^^E**++'+'7'7'9'9$ $ UI/@AAA  rN)r{r r|rrr)r]r^r_r`rrarrrzrzs2 ! ! ! ! ! !rrzc"eZdZdZddZdd Zd S)rzDGet diagram definitions from user (i.e. xml files) or generate them.rargparse.Namespacerrc||_dSre)r)rrs rrzDiadefsHandler.__init__s  rr{r rr list[ClassDiagram]cg}t||}|jjD]+}||||,|s#t |||}|D]}||S)axGet 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`) )rzrr*appendrrcr>extract_relationships)rr{rdiagrams generatorr|diagrams r get_diadefszDiadefsHandler.get_diadefss(66 [( E EE OOI33GUCC D D D D K-fd;;AA'JJH , ,G  ) ) + + + +rN)rrrr)r{r rr rr)r]r^r_r`rrrarrrrsBNNrr)r` __future__rargparsecollections.abcrtypingrrQrastroid.modutilsrpylint.pyreverse.diagramsrr pylint.pyreverse.inspectorr r pylint.pyreverse.utilsr rrcrzrrarrrs ON""""""%%%%%%------BBBBBBBB66666666000000 dIdIdIdIdIdIdIdIN6@6@6@6@6@]O6@6@6@r!!!!!?!!!0r