o g^#@sdZddlZddlZddlZddlZddlmZmZddlZddlm Z dZ ddZ dd Z e d Ze d Ze d Zd dZe dZe dZddZddZddZddZdZdZdZdZdeeeeeedZeeeddZGdd d ZGd!d"d"ZGd#d$d$eZ d%ee j!e j"fd&e#fd'd(Z$d)ee j%e j&fd&eee j!e j"ffd*d+Z'd)ee j%e j&fd&e(fd,d-Z)d.d/Z*dS)0z9Generic classes/functions for pyreverse core/extensions. N)OptionalUnion)nodesz .pyreverserccCsg}tjdd}|r>tj|t}z!t|dd}|}WdW|S1s,wYW|St y=Y|Sw|S)z,Read config file and return list of options.HOMEzutf-8)encodingN) osenvirongetpathjoinRCFILEopenreadsplitOSError)optionshomercfile file_handlerT/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pylint/pyreverse/utils.pyget_default_options"s   rcCs*t}||D] }tjd|q dS)z"insert default options to sys.argvN)rreversesysargvinsert)rargrrrinsert_default_options0s rz^__([^\W_]_*)+__$z^__(_*[^\W_])+_?$z^_\w*$cCs>t|r d}|St|rd}|St|rd}|Sd}|S)zHreturn the visibility from a name: public, protected, private or specialspecialprivate protectedpublic)SPECIALmatchPRIVATE PROTECTED)name visibilityrrrget_visibility>s   r*z ^.*Abstract.*z ^[^\W\da-z]*$cC t|jS)zWreturn true if the given class node correspond to an abstract class definition )ABSTRACTr%r(noderrr is_abstractP r/cCr+)zTreturn true if the given class/function node correspond to final definition )FINALr%r(r-rrris_finalWr0r2cC |jdkS)N interfacetyper-rrr is_interface^ r7cCr3)N exceptionr5r-rrr is_exceptioncr8r:r)ALLPUB_ONLYr$OTHER)r r"r!r#c@ eZdZdZddZddZdS) FilterMixInz6filter nodes according to a mode and nodes' visibilityc Csfd}|dD]&}z|t|7}Wqty-}ztd|tjdWYd}~qd}~ww||_dS)zinit filter modesr+zUnknown filter mode )fileN)rMODESKeyErrorprintrstderr_FilterMixIn__mode)selfmoderInummodexrrr__init__s  zFilterMixIn.__init__cCs tt|d|}|jt|@ S)z)return true if the node should be treatedr()r*getattrrIVIS_MOD)rJr.r)rrr show_attrszFilterMixIn.show_attrN)__name__ __module__ __qualname____doc__rNrQrrrrrB|s rBc@s:eZdZdZddZd ddZddZd d Zd d ZdS) ASTWalkeraa walker visiting a tree in preorder, calling on the handler: * 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 cCs||_i|_dSN)handler_cache)rJrXrrrrNs zASTWalker.__init__NcCs|durt}||vrtt|||jf|||||D]}||us+J|||q#|||j|us>JdS), getting callbacks from handlerN) setAssertionErroridparentaddvisit get_childrenwalkleave)rJr._done child_noderrrrbs     zASTWalker.walkcCs|j}|j|}|dur:|j}|j}t|d|t|dd}t|d|t|dd}||f|j|<||fS|\}}||fS)z/get callbacks from handler for the visited nodeNvisit_ visit_defaultleave_ leave_default) __class__rYr rXrRlowerrO)rJr.klassmethodsrXkide_methodl_methodrrr get_callbackss  zASTWalker.get_callbackscC&||d}|dur||dSdS)rZrNrqrJr.methodrrrr` zASTWalker.visitcCrr)rZrNrsrtrrrrcrvzASTWalker.leaverW) rRrSrTrUrNrbrqr`rcrrrrrVs  rVc@rA) LocalsVisitorz3visit a project by traversing the locals dictionarycCst||t|_dSrW)rVrNr[_visited)rJrrrrNs  zLocalsVisitor.__init__cCsz||jvrdS|j|||}|ddur|d|t|dr/|D]}||q'|ddur;|d|SdS)z-launch the visit starting from the given nodeNrlocalsr)rxr_rqhasattrvaluesr`)rJr.rm local_noderrrr`s          zLocalsVisitor.visitN)rRrSrTrUrNr`rrrrrws rwannreturncCs2d}t|tjr|}|St|tjr|j}|S)Nr) isinstancer Subscript as_stringNamer()r}labelrrrget_annotation_labels  rr.cCsd}t|jtjr|jj}n*t|tjr6|jj}ztt|j|j j }| |jj j }Wn ty5Ynw|Sz|^}}Wn tjyLd}Ynwt|}|rht|dddurf|dsfd|dn|}|rm||_ |S)z return the annotation for `node`Nrvaluerz Optional[])rr^r AnnAssign annotation AssignAttrdictzipryargs annotationsr rr(AttributeErrorinferastroidInferenceErrorrrO startswith)r.r} init_methodrdefault_rrrrget_annotations8    rcCsft|}z|rt|tjr|hWSt|WSt|WStjy2|r-|hYStYSw)zReturn a set containing the node annotation if it exists otherwise return a set of the inferred types using the NodeNG.infer method)rrrrr[rrr)r.r}rrr infer_nodes rcCs(tddurtdtddSdS)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.dotNzThe requested output format is currently not available. Please install 'Graphviz' to have other output formats than 'dot' or 'vcg'. )shutilwhichrGrexitrrrrcheck_graphviz_availabilitys r)+rUrrerrtypingrrrrr rrcompiler$r&r'r*r,r1r/r2r7r: _CONSTRUCTOR_SPECIAL _PROTECTED_PRIVATErErPrBrVrwrrstrrr AssignNamerr[rrrrrrs\      ;  #