ó '![c@s°dZddlmZddlZddlmZmZddlmZej Z de fd„ƒYZ dee fd „ƒYZ d e fd „ƒYZd e fd „ƒYZdS(sHhandle diagram generation options for class diagram or default diagrams iÿÿÿÿ(tbuiltinsN(tPackageDiagramt ClassDiagram(t LocalsVisitortDiaDefGeneratorcBsheZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z RS( s!handle diagram generation optionscCs,|j|_|jƒ||_d|_dS(s%common Diagram Handler initializationN(tconfigt_set_default_optionstlinkertNonet classdiagram(tselfRthandler((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt__init__s   cCs2|j}|jr.d|jƒj|f}n|S(sget title for objectss%s.%s(tnamet module_namestroot(R tnodettitle((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt get_title&s  cCs |dkrt|jjƒS|S(s3activate some options if not explicitly deactivatedN(RtboolRtclasses(R toption((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt _set_option-s cCsÅ|j|jjƒ|_|j|jjƒ}|j|jjƒ}d\}}|r]d}n|rld}n|jjdk r|jj}n|jjdk r®|jj}n|||_|_ dS(s6set different default options with _default dictionaryiiÿÿÿÿN(ii( RRRt all_ancestorstall_associatedtshow_ancestorsRtshow_associatedt anc_levelt ass_level(R RRRR((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR5s   cCs|j|jfS(shelp function for search levels(RR(R ((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt _get_levelsEscCs#|jjrtS|jƒjtkS(s&true if builtins and not show_builtins(Rt show_builtintTrueRR t BUILTINS_NAME(R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt show_nodeIs cCs0|jj|ƒ|jj|j|ƒ|ƒdS(s%visit one class and add it to diagramN(RtvisitR t add_objectR(R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt add_classOsccsK|dkrdSx4|jdtƒD] }|j|ƒs>q#n|Vq#WdS(s%return ancestor nodes of a class nodeiNtrecurs(t ancestorstFalseR!(R Rtleveltancestor((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt get_ancestorsTs  ccs¢|dkrdSx‹t|jjƒƒt|jjƒƒD]a}xX|D]P}t|tjƒrj|j}nt|tjƒoˆ|j |ƒs‘qFn|VqFWq9WdS(s'return associated nodes of a class nodeiN( tlisttinstance_attrs_typetvaluest locals_typet isinstancetastroidtInstancet_proxiedtClassDefR!(R t klass_nodeR(t ass_nodestass_node((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pytget_associated]s   cCsŸ|jj|ƒs"|j|ƒ r&dS|j|ƒx1|j||ƒD]}|j||d|ƒqFWx1|j||ƒD]}|j|||dƒqzWdS(s1extract recursively classes related to klass_nodeNi(R thas_nodeR!R$R*textract_classesR7(R R4RRR)R6((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR9ks" ( t__name__t __module__t__doc__R RRRRR!R$R*R7R9(((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyRs        tDefaultDiadefGeneratorcBsDeZdZd„Zd„Zd„Zd„Zd„Zd„ZRS(s¡generate minimum diagram definition for the project : * a package diagram including project's modules * a class diagram including project's classes cCs$tj|||ƒtj|ƒdS(N(RR R(R RR ((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR scCsc|jj}t|jƒdkr=td|j|ƒ|_n d|_td|j|ƒ|_ dS(s`visit an pyreverse.utils.Project node create a diagram definition for packages is packages %ss classes %sN( RtmodetlentmodulesRR t pkgdiagramRRR (R RR>((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt visit_projectƒs   cCs#|jr|j|jfS|jfS(s`leave the pyreverse.utils.Project node return the generated diagram definition (RAR (R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt leave_projects cCs6|jr2|jj|ƒ|jj|j|ƒndS(s_visit an astroid.Module node add this class to the package diagram definition N(RARR"R#R (R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt visit_module˜s cCs)|jƒ\}}|j|||ƒdS(s\visit an astroid.Class node add this class to the class diagram definition N(RR9(R RRR((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pytvisit_classdef¡scCs&|jr"|jj||jƒndS(sHvisit astroid.ImportFrom and catch modules for package diagram N(RAtadd_from_dependtmodname(R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pytvisit_importfrom©s ( R:R;R<R RBRCRDRERH(((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR=xs  tClassDiadefGeneratorcBs eZdZd„Zd„ZRS(s[generate a class diagram definition including all classes related to a given class cCstj|||ƒdS(N(RR (R RR ((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR µscCs¸t||jjƒ|_t|jƒdkrW|jddƒ\}}|j|ƒ}n |jd}|jdƒd}t |j |ƒƒ}|j ƒ\}}|j |||ƒ|jS(s^return a class diagram definition for the given klass and its related klasses it.iiÿÿÿÿ( RRR>R R?R@trsplitt get_moduletsplittnexttilookupRR9(R tprojecttklasstmoduleRR((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt class_diagram¸s (R:R;R<R RS(((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyRI°s tDiadefsHandlercBs eZdZd„Zd„ZRS(sYhandle diagram definitions : get it from user (i.e. xml files) or generate them cCs ||_dS(N(R(R R((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyR ÒscCs…g}t||ƒ}x-|jjD]}|j|j||ƒƒq"W|sft||ƒj|ƒ}nx|D]}|jƒqmW|S(svGet 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`) (RIRRtappendRSR=R"textract_relationships(R RPRtdiagramst generatorRQtdiagram((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyt get_diadefsÕs  (R:R;R<R RZ(((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyRTÌs (R<t six.movesRR0tpylint.pyreverse.diagramsRRtpylint.pyreverse.utilsRR:R tobjectRR=RIRT(((s:lib/python2.7/site-packages/pylint/pyreverse/diadefslib.pyts  \8