e"dZddlmZddlZddlmZddlmZddlZddlm Z ddl m Z ddl m Z mZdd lmZmZdd lmZGd d ZGd deeZGddeZGddZy)zHHandle diagram generation options for class diagram or default diagrams.) annotationsN) Generator)Any)nodes)is_stdlib_module) ClassDiagramPackageDiagram)LinkerProject) LocalsVisitorceZdZdZd dZddZddZddZddZddZ ddZ dd Z dd Z dd Z y )DiaDefGeneratorz"Handle diagram generation options.cf|j|_d|_|j||_|y)z&Common Diagram Handler initialization.FN)config module_names_set_default_optionslinkerselfrhandlers =/usr/lib/python3/dist-packages/pylint/pyreverse/diadefslib.py__init__zDiaDefGenerator.__init__s+nn "' !!#  ct|j}|jr|jjd|}|S)zGet title for objects..)namerroot)rnodetitles r get_titlezDiaDefGenerator.get_title#s6    yy{''(%1E rcH|t|jjS|S)z4Activate some options if not explicitly deactivated.)boolrclasses)roptions r _set_optionzDiaDefGenerator._set_option*s# > ++, , rc|j|jj|_|j|jj}|j|jj}d\}}|rd}|rd}|jj |jj }|jj |jj }||c|_|_y)z7Set different default options with _default dictionary.)rrN) r%rr all_ancestorsall_associatedshow_ancestorsshow_associated anc_levelassociation_level)rr(r)r,r-s rrz$DiaDefGenerator._set_default_options2s ,,T[[-E-EF(()B)BC ))$++*D*DE'-$ $ I  "  ;; % % 1 22I ;; & & 2 $ ; ; 1:>(+N s57c#jK|dk(ryt|jjt|jjzD]_}|D]X}t |t j r |j}t |tjr|j|sU|Zayw)z(Return associated nodes of a class node.rN) listinstance_attrs_typevalues locals_type isinstanceastroidInstance_proxiedrClassDefr5)r klass_noder>association_nodesrs rget_associatedzDiaDefGenerator.get_associated`s A: !%j&D&D&K&K&M!NQU  " " ) ) +R "   * dG$4$45==D"48T^^D=Q   sB1B3c4|jj|s|j|sy|j||j ||D]}|j ||dz ||j ||D]}|j |||dz y)z2Extract recursively classes related to klass_node.N)r8has_noder5r:r@extract_classesrM)rrKr,r-r?rs rrQzDiaDefGenerator.extract_classesps    % %j 1 9S  z"**:yA MH  9q=:K L M'' 4EF ID  y2Ca2G H IrNrr rDiadefsHandlerreturnNone)rnodes.ClassDefrTstr)r$z bool | NonerTr")rTrU)rTztuple[int, int])rrVrTr"rrVrTrU)rrVr>intrT%Generator[nodes.ClassDef, None, None])rKrVr>rYrTrZ)rKrVr,rYr-rYrTrU)__name__ __module__ __qualname____doc__rr r%rr/r5r:r@rMrQrrrrs,(N 6A " +.  . (14 . I( I58 IMP I  Irrc@eZdZdZd dZd dZd dZd dZd dZddZ y)DefaultDiadefGeneratorzGenerate minimum diagram definition for the project : * a package diagram including project's modules * a class diagram including project's classes c\tj|||tj|yN)rrr rs rrzDefaultDiadefGenerator.__init__s"  vw7t$rc|jj}t|jdkDrt d|j ||_nd|_td|j ||_y)z`Visit a pyreverse.utils.Project node. create a diagram definition for packages rOz packages Nzclasses ) rmodelenmodulesr r pkgdiagramrr8)rrres r visit_projectz$DefaultDiadefGenerator.visit_projectsc {{ t|| q 5CDII;'6DO#DO(8DII;)?Frcd|jr|j|jfS|jfS)zaLeave the pyreverse.utils.Project node. return the generated diagram definition )rhr8)r_s r leave_projectz$DefaultDiadefGenerator.leave_projects/ ????D$5$55 5!!##rc|jrB|jj||jj|j|yy)z`Visit an astroid.Module node. add this class to the package diagram definition N)rhrr7r9rr4s r visit_modulez#DefaultDiadefGenerator.visit_modules: ?? KK  d # OO & &tyy$ 7 rcP|j\}}|j|||y)z]Visit an astroid.Class node. add this class to the class diagram definition N)r/rQ)rrr,r-s rvisit_classdefz%DefaultDiadefGenerator.visit_classdefs+ (,'7'7'9$ $ T9.?@rcj|jr'|jj||jyy)z@Visit astroid.ImportFrom and catch modules for package diagram.N)rhadd_from_dependmodnamer4s rvisit_importfromz'DefaultDiadefGenerator.visit_importfroms& ?? OO + +D$,, ? rNrR)rr rTrU)rkr rTr)rz nodes.ModulerTrUrX)rznodes.ImportFromrTrU) r[r\r]r^rrirlrnrprtr_rrraras( % G$8A@rraceZdZdZddZy)ClassDiadefGeneratorz\Generate a class diagram definition including all classes related to a given class. ct||jj|_t |j dkDr'|j dd\}}|j|}n#|j d}|jdd}t|j|}|j\}}|j||||jS)zDReturn a class diagram definition for the class and related classes.rOrrr') rrrer8rfrgrsplit get_modulesplitnextilookupr/rQ)rprojectklassmoduler,r-s r class_diagramz"ClassDiadefGenerator.class_diagrams( 0@0@A w ! #!LLa0MFE''/F__Q'FKK$R(EV^^E*+'+'7'7'9$ $ UI/@A   rN)r}r r~rVrTr)r[r\r]r^rr_rrrvrvs  !rrvc eZdZdZddZddZy)rSzDGet diagram definitions from user (i.e. xml files) or generate them.c||_yrc)r)rrs rrzDiadefsHandler.__init__s  rcg}t||}|jjD]#}|j|j ||%|st ||j |}|D]}|j|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`) )rvrr#appendrrar7extract_relationships)rr}rdiagrams generatorr~diagrams r get_diadefszDiadefsHandler.get_diadefss(6 [[(( EE OOI33GUC D E-fd;AA'JH ,G  ) ) + ,rN)rzargparse.NamespacerTrU)r}r rr rTzlist[ClassDiagram])r[r\r]r^rrr_rrrSrSsNrrS)r^ __future__rargparsecollections.abcrtypingrrGrastroid.modutilsrpylint.pyreverse.diagramsrr pylint.pyreverse.inspectorr r pylint.pyreverse.utilsr rrarvrSr_rrrs_ O"%-B60 dIdIN6@]O6@r!?!0r