e  dZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZddlZddlmZddlmZe r5ddlmZmZe ej,gee ee eefdffZe e ee efZdZdd Zdd Zej8d Zej8d Zej8d Zd dZ d!dZ!dZ"dZ#dZ$de"e#ze$ze"e#e$zdZ%e"e#e$ddZ&GddZ'GddZ(d"dZ) d#dZ*d$dZ+ddZ,d%dZ-y)&z8Generic classes/functions for pyreverse core/extensions.) annotationsN) TYPE_CHECKINGAnyCallableOptionalTupleUnion)nodes)InferenceResult) ClassDiagramPackageDiagramz .pyreversercc>g}tjjdd}|r[tjj |t } t |d5}|jj}ddd|S|S#1swY|SxYw#t$rY|SwxYw)z,Read config file and return list of options.HOMEutf-8)encodingN) osenvirongetpathjoinRCFILEopenreadsplitOSError)optionshomercfile file_handles 8/usr/lib/python3/dist-packages/pylint/pyreverse/utils.pyget_default_optionsr"!sG ::>>&" %D dF+ fw/ 5;%**,224 5 N7N  5 N   N s0 BB6BB B B BBct}|j|D]"}tjj d|$y)z#Insert default options to sys.argv.N)r"reversesysargvinsert)rargs r!insert_default_optionsr*/s5!#G OO  3 z^__([^\W_]_*)+__$z^__(_*[^\W_])+_?$z^_\w*$ctj|rd}|Stj|rd}|Stj|rd}|Sd}|S)zIReturn the visibility from a name: public, protected, private or special.specialprivate protectedpublic)SPECIALmatchPRIVATE PROTECTED)name visibilitys r!get_visibilityr7=s^}}T   t       r+c |jdk(S)N exception)type)nodes r! is_exceptionr<Ks 99 ##r+)ALLPUB_ONLYr1OTHER)r-r/r.r0c eZdZdZddZddZy) FilterMixInz7Filter nodes according to a mode and nodes' visibility.cd}|jdD]} |t|z }||_y#t$r(}td|tj Yd}~Fd}~wwxYw)zInit filter modes.r+zUnknown filter mode )fileN)rMODESKeyErrorprintr&stderr_FilterMixIn__mode)selfmoderLnummodexs r!__init__zFilterMixIn.__init__fsjjjo DF D%-' D   D,RD1 CC Ds . AAAc\tt|d|}|jt|z S)z*Return true if the node should be treated.r5)r7getattrrLVIS_MOD)rMr;r6s r! show_attrzFilterMixIn.show_attrps,#GD&$$?@ ;;!4444r+N)rNstrreturnNone)r;znodes.NodeNG | strrWbool)__name__ __module__ __qualname____doc__rQrUr+r!rDrDcsA5r+rDc(eZdZdZddZddZddZy) LocalsVisitora Visit a project by traversing the locals dictionary. * visit_ on entering a node, where class name is the class of the node in lower case * leave_ on leaving a node, where class name is the class of the node in lower case c0i|_t|_yN)_cacheset_visited)rMs r!rQzLocalsVisitor.__init__sAC +.5 r+c 0|j}|jj|}|e|jj }t |d|t |dd}t |d|t |dd}||f|j|<||fS|\}}||fS)z0Get callbacks from handler for the visited node.Nvisit_ visit_defaultleave_ leave_default) __class__rcrrZlowerrS)rMr;klassmethodskide_methodl_methods r! get_callbackszLocalsVisitor.get_callbackss++//%( ?..&&(CsengdOT&JHsengdOT&JH#+H!5DKK !!") Hh!!r+c||jvry|jj||j|}|d |d|t|dr&|j D]}|j ||d |d|Sy)z.Launch the visit starting from the given node.Nrlocalsr$)readdrrhasattrvaluesvisit)rMr;rn local_nodes r!rxzLocalsVisitor.visits 4==  $$$T* 1: ! GAJt  4 ""kkm '  :& ' 1: !71:d# #r+NrWrX)r; nodes.NodeNGrW_CallbackTupleT)r;r{rWr)rZr[r\r]rQrrrxr^r+r!r`r`vs1""r+r`ct|tjr|j |jSt|tjr|j Sy)Nr) isinstancer Namer5NodeNG as_string)anns r!get_annotation_labelrs?#uzz"sxx';xx#u||$}} r+cd}t|jtjr|jj}nt|tj r{|jj} t t|j|jj}|j|jjj}n|S |j^}}t%|}|r^t'|ddQ|j)ds@t|tj*r t-d|j/Dsd|d}|r |r||_ |S#t$rYwxYw#t j"$rd}YwxYw)z!Return the annotation for `node`.Nrvaluerc3pK|].}t|tjxr|jdu0ywrb)r~r Constr).0childs r! z!get_annotation..s45%++.F5;;$3FFs46z Optional[])r~parentr AnnAssign annotation AssignAttrdictziprtargsrrrr5AttributeErrorinferastroidInferenceErrorrrS startswithBinOpany get_children)r;r init_methodrdefault_labels r!get_annotationrsV C$++u/kk$$ D%** +kk((  s;#5#5{7G7G7S7STUK!+//$++"3"3"8"89C jjl ! ! %E GWg . 6  ,3 , --/ UG1%  J9     ! !s%/A#E E/ E,+E,/FFcbt|} |r_t|tjs)t|tjr|j dk(r|hSt |jSt |jS#tj$r|r|hcYSt cYSwxYw)zReturn a set containing the node annotation if it exists otherwise return a set of the inferred types using the NodeNG.infer method. |) rr~r Subscriptroprdrrr)r;rs r! infer_noders  C ' #u/3 ,3u syy{# #4::<   ! !'u&&'s$ABB.BB.! B.-B.cptjd!tdtjdyy)zCheck if the ``dot`` command is available on the machine. This is needed if image output is desired and ``dot`` is used to convert from *.dot or *.gv into the final output format. dotNz?'Graphviz' needs to be installed for your chosen output format. )shutilwhichrJr&exitr^r+r!check_graphviz_availabilityrs, ||E" OP  #r+cHtjddgddd}tjd|jj }|s t d y |jd }||jvr't d |d |tjdy y )zCheck if the ``dot`` command supports the requested output format. This is needed if image output is desired and ``dot`` is used to convert from *.gv into the final output format. rz-T?TFr)capture_outputcheckrz$.*Use one of: (?P(\S*\s?)+))patternstringzUnable to determine Graphviz supported output formats. Pyreverse will continue, but subsequent error messages regarding the output format may come from Graphviz directly.NformatszFormat z, is not supported by Graphviz. It supports: r) subprocessrunrer2rKstriprJgrouprr&r) output_format dot_outputr2supported_formatss r!!check_if_graphviz_supports_formatrs  t57J HH7  &&( E   K  I.-3355 m_$PQbPc d   6r+)rWz list[str]rz)r5rVrWrV)r;znodes.ClassDefrWrY)rznodes.Name | nodes.NodeNGrWrV)r;#nodes.AssignAttr | nodes.AssignNamerWz#nodes.Name | nodes.Subscript | None)r;rrWzset[InferenceResult])rrVrWrX).r] __future__rrrrrr&typingrrrrrr rr astroid.typingr pylint.pyreverse.diagramsr r r _CallbackTr|rr"r*compiler1r3r4r7r<_SPECIAL _PROTECTED_PRIVATErHrTrDr`rrrrrr^r+r!rsY ?"  GG*F  eL!5)E#FLM OJHZ0(:2FFGO    "**) * "**) * BJJy !  $    :%0 ( "    55&--`( -(((V'$r+