%![c@sdZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd l mZdd lmZdd lmZe jZdd Zee j_ee j_ee j_ee j_ee j_ee j_ddZddZee j_ee j_ee j _ddZ!dZ"e!e j#_dZ$ddZ%e j&e%e j'_e%e j(_)e j*e j&ddZ+e+e j,_e j&de-dZ.e.e j/_dZ0e0e j/_0e j&de-dZ1e1e j2_e j*ddZ3e j&e3e j4_e3e j5_)e j&ddZ6e6e j7_e8Z9e j*ddZ:e j&e:e j;_e:e j;_)e j*e j&ddZ<e<e j=_dZ>ddZ?e j*e j&ddZ@e?e jA_?e@e jA_dZBd ZCeDd!ZEeDd"ZFd#ZGd$ZHd%ZId&ZJd'ZKd(ZLe jMe j&dd)ZNeLe jO_LeNe jO_dd*ZPe j&dd+ZQePe jR_PeQe jR_dd,ZSeSe jT_e j&dd-ZUeUe j(_eUe j5_e j&dd.ZVeVe jW_dd/ZXeXe jY_dd0ZZeZej[_\dS(1sMthis module contains a set of functions to handle inference on astroid trees iN(tbases(tcontext(t exceptions(t decorators(thelpers(tmanager(tnodes(t dunder_lookup(t protocols(tutilccs |VdS(sRinference's end for node such as Module, ClassDef, FunctionDef, Const... N((tselfR((s0lib/python2.7/site-packages/astroid/inference.pyt infer_end"sccsgtd|jDs!|VnBt||}t||j|j|j}|j||VdS(Ncss!|]}t|tjVqdS(N(t isinstanceRtStarred(t.0te((s0lib/python2.7/site-packages/astroid/inference.pys 1s(tanyteltst _infer_seqttypetlinenot col_offsettparenttpostinit(R Rtvaluestnew_seq((s0lib/python2.7/site-packages/astroid/inference.pyt infer_seq0s ! cCsg}x|jD]}t|tjrtj|j|}|dtj fkrmt j d|d|nt |dst j d|d|n|j t|q|j|qW|S(s-Infer all values based on _BaseContainer.eltstnodeRRN(RR RR Rt safe_infertvaluetNoneR t UninferableRtInferenceErrorthasattrtextendRtappend(RRRtelttstarred((s0lib/python2.7/site-packages/astroid/inference.pyR:s  ccsstd|jDs!|VnNt||}t||j|j|j}|jt|j|VdS(Ncss'|]\}}t|tjVqdS(N(R Rt DictUnpack(Rtkt_((s0lib/python2.7/site-packages/astroid/inference.pys Ss( Rtitemst _infer_mapRRRRRtlist(R RR)R((s0lib/python2.7/site-packages/astroid/inference.pyt infer_mapRs !cCsi}x |jD]\}}t|tjrtj||}|dtjfkrat j nt|tj st j d|d|n|j t ||qtj|d|}tj|d|}td||fDrt j d|d|n|||ksN(R)R RR&RRRR RRR tDicttupdateR*R(RRRtnameRtdouble_starredtkey((s0lib/python2.7/site-packages/astroid/inference.pyR*\s"   cCsO|}x,|jr4t|jtj r4|j}q W|rK|jrK|jSdS(s Search for the first function which encloses the given scope. This can be used for looking up in that function's scope, in case looking up in a lower scope for a particular name fails. :param node: A scope node. :returns: ``None``, if no parent function scope was found, otherwise an instance of :class:`astroid.scoped_nodes.Function`, which encloses the given node. N(RR Rt FunctionDefR(Rtcurrent((s0lib/python2.7/site-packages/astroid/inference.pyt_higher_function_scopeus " cCs|j|j\}}|st|j}|rQ|j|j\}}n|stjd|jd|jd|qn|j}|j|_tj |||S(s#infer a Name: use name lookup rulesR0tscopeR( tlookupR0R5R6RtNameInferenceErrortclonet lookupnameRt _infer_stmts(R Rtframetstmtstparent_functionR(((s0lib/python2.7/site-packages/astroid/inference.pyt infer_names   ccs|j}tjd|jd|j|_d|_x|jj |D]r}|t j kri|VqIny8t |drx"|j ||D] }|VqWnWqItjk rqIqIXqIWttd|d|dS(s>infer a Call node by trying to guess what the function returnstargstkeywordstinfer_call_resultRRN(R9t contextmodt CallContextR@RAt callcontextRt boundnodetfunctinferR RR!RBRR t StopIterationtdict(R RREtcalleetinferred((s0lib/python2.7/site-packages/astroid/inference.pyt infer_calls   ccs|j}|dkr0tjd|d|ny2|rS|j|j|Vn|j|VWn;tjk r}tjtjd|d|d|nXdS(s7infer an Import node: return the imported module/objectRRterrorN( R:RRR tdo_import_modulet real_nametAstroidBuildingErrorR treraise(R RtasnameR0texc((s0lib/python2.7/site-packages/astroid/inference.pyt infer_imports  cCs(tj}||_|j|dtS(NRS(RCtInferenceContextR:RHtFalse(R R0R((s0lib/python2.7/site-packages/astroid/inference.pytinfer_name_modules  c Cs+|j}|dkr0tjd|d|n|rH|j|}ny|j}Wn;tjk r}tjtjd|d|d|nXyMt j |}||_|j |d||j k}t j||SWnAtjk r&}tjtj|jd|d|d|nXdS(s:infer a ImportFrom node: return the imported module/objectRRRNt ignore_localsttargett attributeN(R:RRR RPRORQR RRRCt copy_contexttgetattrtrootRR;tAttributeInferenceErrortmessage(R RRSR0tmoduleRTR=RN((s0lib/python2.7/site-packages/astroid/inference.pytinfer_import_froms$   ! ccs]x;|jj|D]'}|tjkr3|Vqn|r|jrt|tjrt|jtjry7tj tj |jtj |r|j}nWqt j k rqXqny;||_x"|j |j|D] }|VqWd|_Wqt jt jfk r d|_qtk r9d|_qXqWttd|d|dS(sAinfer an Attribute node by using getattr on the associated objectRRN(texprRHR RRFR RtInstanceRt is_subtypet object_typeRt_NonDeducibleTypeHierarchytigetattrtattrnameRR_R tAttributeErrorRIRJ(R Rtownertobj((s0lib/python2.7/site-packages/astroid/inference.pytinfer_attributes.      c Cs|jdkr*tjd|d|ny&tj|jj|j|SWnDtjk r}t j tj|j d|d|jd|nXdS(NRRRZR[( R:RRR RR;R^R]R_R RRR`(R RRN((s0lib/python2.7/site-packages/astroid/inference.pyt infer_globals  c cst|jj|}|tjkr3tjVdSt|jj|}|tjkrftjVdSt}|jtj kr|}n9|jtj krt j |}|r|}qn|}|tkrt j d|d|ny|j||}WnMt jt jt jtfk rL}tjt j d|d|d|nX||ksh|tjkrttjVdSx|j|D] }|VqWttd|d|dS(sInference for subscripts We're understanding if the index is a Const or a slice, passing the result of inference to the value's `getitem` method, which should handle each supported index type accordingly. NRRRN(tnextRRHR Rtslicet_SUBSCRIPT_SENTINELt __class__RRdRtclass_instance_as_indexRR tgetitemtAstroidTypeErrortAstroidIndexErrorR_RjRRRIRJ( R RRtindext index_valuetinstance_as_indextassignedRTRL((s0lib/python2.7/site-packages/astroid/inference.pytinfer_subscript#s@     c csY|j}|jdkr$tj}n tj}y)g|D]}|jd|^q7}Wntjk rutj VdSXxt j |D]}t d|Drtj Vqng|D]}|j ^q}t d|Drtj Vqntj }x7t||D]!\}}||r |VPq q W|VqWttd|d|dS(sInfer a boolean operation (and / or / not). The function will calculate the boolean operation for all pairs generated through inference for each component node. torRNcss|]}|tjkVqdS(N(R R(Rtitem((s0lib/python2.7/site-packages/astroid/inference.pys tscss|]}|tjkVqdS(N(R R(RR}((s0lib/python2.7/site-packages/astroid/inference.pys zsR(Rtoptoperatorttruthtnot_RHRR R Rt itertoolstproductRt bool_valuetzipRIRJ( R RRt predicateRtpairR}t bool_valuesR((s0lib/python2.7/site-packages/astroid/inference.pyt _infer_boolop^s0   )   ccs=x6|||D]%}t||r0tjVq|VqWdS(N(R R R(R tinfer_callableRRNtresult((s0lib/python2.7/site-packages/astroid/inference.pyt_filter_operation_errorss c cs?x8|jj|D]$}y|j|jVWqtk rZ}tj||j|Vqtk r6}tj |j}|dkr|j }|tj k rt j| Vq3tj Vq7t|tjt jfstj||j|Vqnyytj||}Wn.tjk r>tj||j|VwnX|d}t|jd|}|tj ks}|j rwntj|}tjd|g|_|j|d|}t|d} | dkr|Vn| VWq7tjk r}tj||j|Vq7tjk r2tj Vq7XqXqWdS(s3Infer what an UnaryOp should return when evaluated.iRR@N(toperandRHtinfer_unary_opR~t TypeErrorR tBadUnaryOperationMessageRjRtUNARY_OP_METHODRRRRt const_factoryR RRdtClassDefRR7RR_RotcallableRCR\RDRERBR ( R RRRTtmethRtmethodsRLt call_resultsR((s0lib/python2.7/site-packages/astroid/inference.pyt_infer_unaryopsH      ccsGx%t|t|tjD] }|VqWttd|d|dS(s3Infer what an UnaryOp should return when evaluated.RRN(RRR RRIRJ(R RRL((s0lib/python2.7/site-packages/astroid/inference.pyt infer_unaryops cCst|tjo|jtkS(s0Check if the given const node is NotImplemented.(R RtConstRtNotImplemented(tconst((s0lib/python2.7/site-packages/astroid/inference.pyt_is_not_implementedsc Cshtj||}|d}t|jd|}|tjkrOtjn|j|||||S(s8Invoke binary operation inference on the given instance.iR( RR7RoRHR RRR tinfer_binary_op( tinstancetopnodeR~totherRt method_nameRtmethodRL((s0lib/python2.7/site-packages/astroid/inference.pyt_invoke_binop_inferences   cCs>tj|}tjtd|d|d|d|d|d|S(s<Get an inference callable for an augmented binary operation.RR~RRRR(RtAUGMENTED_OP_METHODt functoolstpartialR(RRR~RRtreverseR((s0lib/python2.7/site-packages/astroid/inference.pyt_aug_ops   cCsT|rtj|}n tj|}tjtd|d|d|d|d|d|S(sGet an inference callable for a normal binary operation. If *reverse* is True, then the reflected method will be used instead. RR~RRRR(RtREFLECTED_BIN_OP_METHODt BIN_OP_METHODRRR(RRR~RRRR((s0lib/python2.7/site-packages/astroid/inference.pyt_bin_ops  ccsMxF||fD]8}|j}tjd|g|_d|_|Vq WdS(sGet contexts for binary operations. This will return two inferrence contexts, the first one for x.__op__(y), the other one for y.__rop__(x), where only the arguments are inversed. R@N(R9RCRDRERRF(Rtlefttrighttargt new_context((s0lib/python2.7/site-packages/astroid/inference.pyt_get_binop_contextss   cCs|j|jkS(s$Check if type1 is the same as type2.(tqname(ttype1ttype2((s0lib/python2.7/site-packages/astroid/inference.pyt _same_type"sc Cs|j}t||r6t|||||g}ntj||rft|||||g}ntj||rt|||||dtt|||||g}n6t|||||t|||||dtg}|S(sVGet the flow for binary operations. The rules are a bit messy: * if left and right have the same type, then only one method will be called, left.__op__(right) * if left and right are unrelated typewise, then first left.__op__(right) is tried and if this does not exist or returns NotImplemented, then right.__rop__(left) is tried. * if left is a subtype of right, then only left.__op__(right) is tried. * if left is a supertype of right, then right.__rop__(left) is first tried and then left.__op__(right) R(R~RRRRet is_supertypetTrue( Rt left_typet binary_opnodeRt right_typeRtreverse_contextR~R((s0lib/python2.7/site-packages/astroid/inference.pyt_get_binop_flow's !c CsQ|jjd}|j}t||r]t|||||t|||||g} ntj||rt|||||t|||||g} ntj||rt|||||t|||||dtt|||||g} nKt|||||t|||||t|||||dtg} | S(sGet the flow for augmented binary operations. The rules are a bit messy: * if left and right have the same type, then left.__augop__(right) is first tried and then left.__op__(right). * if left and right are unrelated typewise, then left.__augop__(right) is tried, then left.__op__(right) is tried and then right.__rop__(left) is tried. * if left is a subtype of right, then left.__augop__(right) is tried and then left.__op__(right). * if left is a supertype of right, then left.__augop__(right) is tried, then right.__rop__(left) and then left.__op__(right) t=R( R~tstripRRRRReRR( RRt aug_opnodeRRRRtbin_optaug_opR((s0lib/python2.7/site-packages/astroid/inference.pyt _get_aug_flowEs  !c csmt|||\}}tj|}tj|}||||||||}x|D]} yt| } WnFtk rq[q[tjk rq[q[tjk rtj VdSXt d| Drtj VdSt t t | rq[ntd| D} | r4| t| kr4tj VdSx| D] } | Vq;WdSq[Wtj||j|VdS(sInfer a binary operation between a left operand and a right operand This is used by both normal binary operations and augmented binary operations, the only difference is the flow factory used. Ncss|]}|tjkVqdS(N(R R(RR((s0lib/python2.7/site-packages/astroid/inference.pys scss!|]}t|rdVqdS(iN(R(RR((s0lib/python2.7/site-packages/astroid/inference.pys s(RRRfR+RjRR_R R RRtalltmapRtsumtlentBadBinaryOperationMessageR~( RRRRt flow_factoryRRRRRtresultstnot_implementedR((s0lib/python2.7/site-packages/astroid/inference.pyt_infer_binary_operationis8     c cs|dkrtj}n|j}|j}|j}|j}x|jd|D]}|tjkrytjVdSx|jd|D]l}|tjkrtjVdSy,x%t ||||t D] }|VqWWqt j k rtjVqXqWqXWdS(s"Binary operation inferrence logic.RN( RRCRVRRR9RHR RRRRRg( R RRRt lhs_contextt rhs_contexttlhstrhsR((s0lib/python2.7/site-packages/astroid/inference.pyt _infer_binops(      cCst|t|tjS(N(RRR R(R R((s0lib/python2.7/site-packages/astroid/inference.pyt infer_binops ccs|dkrtj}nx|jjd|D]}|tjkrRtjVdS|j}x|jj d|D]l}|tjkrtjVdSy,x%t ||||t D] }|VqWWqtt j k rtjVqtXqtWq1WdS(s0Inference logic for augmented binary operations.RN(RRCRVRZt infer_lhsR RR9RRHRRRRg(R RRRRR((s0lib/python2.7/site-packages/astroid/inference.pyt_infer_augassigns    cCst|t|tjS(N(RRR R(R R((s0lib/python2.7/site-packages/astroid/inference.pytinfer_augassigns cCsC|j}|dkr0tjd|d|ntj|||S(NRR(R:RRR Rt_arguments_infer_argname(R RR0((s0lib/python2.7/site-packages/astroid/inference.pytinfer_argumentss  cCsS|j}t|tjr+|j|St|jd|}tj||S(sWinfer a AssignName/AssignAttr: need to inspect the RHS part of the assign node R( t statementR Rt AugAssignRHR+tassigned_stmtsRR;(R RtstmtR=((s0lib/python2.7/site-packages/astroid/inference.pyt infer_assigns   ccsf|jstjVnKy,x%tj|jd|D] }|Vq3WWntjk ratjVnXdS(NR(thas_underlying_objectR RtMANAGERtinfer_ast_from_somethingtobjectRt AstroidError(R RRL((s0lib/python2.7/site-packages/astroid/inference.pytinfer_empty_nodes    cCs|jj|S(N(RRH(R R((s0lib/python2.7/site-packages/astroid/inference.pyt infer_indexsc Cs|r|j}ntj}}tjd|g|_||_t|jdd|}t|t j st j dd|d|nyt|j ||SWn<tk rtjt j ddd|d|d|nXdS( NR@t __getitem__Rs(Could not find __getitem__ for {node!r}.RR`s)Inference for {node!r}[{index!s}] failed.Rw(R9RCRVRDRERFRoRhR Rt BoundMethodRR RBRIR RR(R RwRRR((s0lib/python2.7/site-packages/astroid/inference.pytinstance_getitems   (]t__doc__RRRtastroidRRRCRRRRRtastroid.interpreterRRR tAstroidManagerRRR tModulet_inferRR3tLambdaRtSliceRRtListtTupletSetR,R*R.R5R?t path_wrappertNamet AssignNameRtraise_if_nothing_inferredRMtCallRRUtImportRXRbt ImportFromRmt Attributet AssignAttrRntGlobalRRqR{t SubscriptRtBoolOpRRRtUnaryOpRRRWRRRRRRRRtyes_if_nothing_inferredRtBinOpRRRRt ArgumentsRRt EmptyNodeRtIndexRRdRt(((s0lib/python2.7/site-packages/astroid/inference.pyt s                         "   6 4  6        $ / !