o bg@s\dZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd lmZdd lmZed ZedZddZddZdZddddddddddddd d! Zd"d#eDZd$d#eDZd%d&d'dd(Zejejejej d(Z!d)d*Z"d+d,ej#_$d-d,ej%_$d.d,ej&_$d/d,ej'_$d0d,ej(_$d1d,d2d,d3d,d4d,d5d,d6d,d7d,d8d,d9d,d:d,d;d,dkrej+e)d?<e,e)D] \Z-Z.e.e)e-d@<qe j/dAdBZ0e0ej'_1dCdDZ2dEdFZ3e j/dGdHZ4e4ej#_1e4ej%_1e j/dIdJZ5e5e j6_1e5ej7_1 dKdLZ8e j9dedMdNZ:e:ej;_e>ej#_ej%_>@cCsi|] \}}|t|qSr)r.0keyvaluerrr Esr7cCsi|] \}}|dt|qS)=)rr3rrrr7Hs__pos____neg__ __invert__)r&r'~notcCst|}||}t|SN)_UNARY_OPERATORSr const_factory)objopfuncr6rrr_infer_unary_opZs rDcCtt|j|Sr>)rDtupleeltsselfrBrrr`rJcC t|j|Sr>)rDrGrHrrrrJa cCrEr>)rDsetrGrHrrrrJbrKcCrLr>)rDr6rHrrrrJcrMcCrEr>)rDdictitemsrHrrrrJdrKcCs||Sr>rabrrrrJicCs||Sr>rrQrrrrJjrTcCs||Sr>rrQrrrrJkrTcCs||Sr>rrQrrrrJlrTcCs||Sr>rrQrrrrJmrTcCs||Sr>rrQrrrrJnrTcCs||Sr>rrQrrrrJorTcCs||@Sr>rrQrrrrJprTcCs||BSr>rrQrrrrJqrTcCs||ASr>rrQrrrrJrrTcCs||>Sr>rrQrrrrJsrTcCs||?Sr>rrQrrrrJtrT) r&r'r(r)r*r+r,r-r.r/r0r1)r2r8ccstt}t|tjrLz2t|}zt||j|jVWWdSty,|VYn ty8t j VYnwWdSWdStyK|VYdSwt|jt r\|dkr\t j VdS|VdS)Nr,) r ConstNotImplemented isinstance BIN_OP_IMPLr@r6 TypeError Exceptionr Uninferablestr)rIopnodeoperatorotherr_not_implementedimplrrrconst_infer_binary_op~s(          recs4|j|d}fdd|jD}t||j|_|S)Nparentc3s,|]}|tjurt|ptjVqdSr>)r r]r Z safe_infer)r4eltrrr s z'_multiply_seq_by_int..) __class__rGlistr6)rIr_rarnodeZ filtered_eltsrrr_multiply_seq_by_ints  rmccsR|D]#}|tjurtVq||D]}|tjur |VqtVqqdSr>)r r]r Unknowninfer)rGrrhinferredrrr_filter_uninferable_nodess    rqc cstt}t||jr,|dkr,|j|d}ttt|j |t|j ||_ |VdSt|tjrK|dkrKt|j t sA|VdSt ||||VdSt|t jrl|dkrlt|}|sbtjVdSt ||||VdS|VdS)Nr&rfr*)r rWrXrYrjrk itertoolschainrqrGr6intrmrInstancer Zclass_instance_as_indexr r]) rIr_r`rarmethodrcrlZas_indexrrrtl_infer_binary_ops,         rwcCs |||Sr>)infer_call_result)rIr_r`rarrvrrrinstance_class_infer_binary_ops ryc cs|dd}|d}|D]e}|tjurqt|dsqz|}Wn ty+Yqw|D]D}t|}z|||}Wnt t j t j fyKYq.w|sR|Vq.|tjurYnzt ||||EdHWq.t jyrYnwqdS)z;recursive function to resolve multiple assignments on loopsNritered)popr r]hasattrrzr[r rWgetitemAttributeErrorrAstroidTypeErrorAstroidIndexError_resolve_looppartroInferenceError) parts assign_pathrindexpartrzstmt index_nodeassignedrrrrsJ          rccst|tjs t|ddrt||||dS|dur2|j|D]}t|tjtjfr0|j EdHqn t |j|||EdHt||||dS)Nis_asyncFrlunknownrr) rYr AsyncForgetattrrOiterroTupleListrGr)rIrlrrlstrrrfor_assigned_stmts s rc Csh|durg}z|j|}Wnty$}z tjd|||d|d}~ww|d||jj|||dS)Nz6Tried to retrieve a node {node!r} which does not exist)rlrrr)rlrr)rGr ValueErrorrrinsertrgassigned_stmts)rIrlrrrexcrrrsequence_assigned_stmtss& rcCs|jj||dS)N)rlr)rgrrIrlrrrrrassend_assigned_stmts6rrc cs^|js|js|jstjVdS|jrJt|jddd|krJ|jj}|jj}t |t j o2|jdk}|s9|dkr>|VdS|dkrJt |VdS|rd|jrdt|j|j}||j||EdHdS||jkrvt d}||_|VdS||jkrt i}||_|VdSzt|}|||EdHtjVWdStjytjVYdSw)Nrr metaclass classmethodrvr)argsvarargkwargr r]rrgtypescoperYr ClassDefrru callcontextrCallSiteZ extra_contextinfer_argumentr@ contextmod copy_context default_valuerorZ NoDefault) rIrrZfunctypeclsZ is_metaclassZ call_siterrrrr_arguments_infer_argname>sF         rcCsF|jr|j}t|}d|_t|}||j|j|St||j|Sr>) rrrrrrrgrr)rIrlrrrrrrrarguments_assigned_stmtsks  rccs<|s |jVdSt|j|||EdHt||||dSNr)r6_resolve_assignment_partsrorOrrrrassign_assigned_stmtsys rccs2t||||D]}|durtjVq|VqdSr>)rr r])rIrlrrrprrrassign_annassigned_stmtss  rc cs|dd}|d}|D]n}d}t|tjr-z |j|\}}Wn-ty,YdSwt|drOt|}z|||}Wnt j t j fyNYdSw|sTdS|sZ|Vq|t j urbdSzt||||EdHWqt jy|YdSwdS)z2recursive function to resolve multiple assignmentsNrr})r{rYr DictrP IndexErrorr|rWr}rrrr r]rror)rrrrrrrbrrrrrs@        rccs@t|jD]}t|tjrt|}|Vqt||||dSr) r Z unpack_inferrrYr rrZExceptionInstancerO)rIrlrrrrrrexcepthandler_assigned_stmtss   rc #s~t|j|d}t|tjrt|jjstjddjj D]}t||}t|t j r7| t kr7nq tjd t j}tfdd|Dd}|rr|jsft d}||_|j|_|VdS|jj|dEdHdSdSt|tjrz t|jd|d} Wntjtjfytj|dwt| tjstj| d|jstj|gd|_| ||EdHdStj|d)Nrz,No decorators found on inferred generator %srlc3s |] }|kr|VqdSr>)r)r4rlrCrrriz)_infer_context_manager.. __enter__)r)nextrorYr Generatorrgrrrr FunctionDefqname_CONTEXTLIB_MGRnodes_of_classYieldr6rWlinenoruZigetattrZAttributeInferenceErrorZ BoundMethodrrZ CallContextrx) rImgrrrpZdecorator_node decoratorZpossible_yield_pointsZ yield_pointconstZenterrrr_infer_context_managersN             rc #s ztfdd|jD}Wn tyYdSw|dur(t|||EdHnVt|||D]O}|}|D]E}t|dsEtjd|||dz|j|}Wq4tyc}z tjd|||d|d}~wt yy}z tjd|||d|d}~ww|Vq.t |||d S) aInfer names and other nodes from a *with* statement. This enables only inference for name binding in a *with* statement. For instance, in the following code, inferring `func` will return the `ContextManager` class, not whatever ``__enter__`` returns. We are doing this intentionally, because we consider that the context manager result is whatever __enter__ returns and what it is binded using the ``as`` keyword. class ContextManager(object): def __enter__(self): return 42 with ContextManager() as f: pass # ContextManager().infer() will return ContextManager # f.infer() will return 42. Arguments: self: nodes.With node: The target of the assignment, `as (a, b)` in `with foo as (a, b)`. context: Inference context used for caching already inferred objects assign_path: A list of indices, where each index specifies what item to fetch from the inference results. c3s |] \}}|kr|VqdSr>r)r4rvarsrrrrirz&with_assigned_stmts..NrGz0Wrong type ({targets!r}) for {node!r} assignment)rltargetsrrzCTried to infer a nonexistent target with index {index} in {node!r}.z1Tried to unpack a non-iterable value in {node!r}.r) rrP StopIterationrr|rrrGrr[rO) rIrlrrrresultrArrrrrwith_assigned_stmtss\  rc #sfdd|}t|tjtjfstjd||||d|dur&t}t|tjr|j }|j d}t dd| tj Dd krMtjd ||||d z t||}WntjyetjVYdSw|tjuspt|d svtjVdSz t|}WntytjVYdSwt|jD]G\} } t| tj s|sn:|qtt|j| d} | D]%} t| tj s|sn|qtjt||j|jd } | j |d| Vqt|tjrz t|j!j|d}WntjytjVYdSw|tjus t|d stjVdSz|}Wnty$tjVYdSw|j"}t|tj#s6tjd|dg}||||sHtjd|d|d\}}||d k}t$||r[dn||}||d<|D]]}d}|D]=}t|d svn4t|t$s|d}z |}||}Wnt%yYntytjVYdSw|}qltjt||j|jd }|j |pgd|VdStjVdSdS)aD Arguments: self: nodes.Starred node: a node related to the current underlying Node. context: Inference context used for caching already inferred objects assign_path: A list of indices, where each index specifies what item to fetch from the inference results. cs|}t|D]5\}}t|tjr&|jj|jjkr&||t|fdSt|tj r=||t|f|||qdSr>) rz enumeraterYr Starredr6rappendlenr)Zstarredtargetlookupsrzrelement$_determine_starred_iteration_lookupsrrrMs   zDstarred_assigned_stmts.._determine_starred_iteration_lookupszDStatement {stmt!r} enclosing {node!r} must be an Assign or For node.)rlrrrNrcss|]}dVqdS)Nr)r4rbrrrrilsz)starred_assigned_stmts..rz>Too many starred arguments in the assignment targets {lhs!r}.)rlrrrrz)ctxrgr col_offset)rGrz8Could not make sense of this, the target must be a tuplez5Could not make sense of this, needs at least a lookup)&Z statementrYr AssignForrrrZInferenceContextr6rsumrrrror r]r| collectionsdequerzr[rrGpopleftreversedr{rrrrZpostinitrrrslicer)rIrlrrrr6lhsrhsrGrZ left_nodeZlhs_eltsZ right_nodepackedZinferred_iterablerzrrZlast_element_indexZlast_element_lengthZis_starred_lastZ lookup_slicerZ found_elementlookupZitered_inner_elementunpackedrrrstarred_assigned_stmtsAs                   r)NNN)R__doc__rr`Z operator_modsysrrZastroidrrrrrrrr r r r Z lazy_importr rrrrZ BIN_OP_METHODrPZREFLECTED_BIN_OP_METHODZAUGMENTED_OP_METHODZUNARY_OP_METHODposneginvertnot_r?rDrZinfer_unary_oprSetrWrrZ version_infomatmulrkZ_KEYZ_IMPLZyes_if_nothing_inferredreZinfer_binary_oprmrqrwryrurrZraise_if_nothing_inferredrrrZ ComprehensionrrZ AssignNameZ AssignAttrrr Argumentsrrr AnnAssign AugAssignrr ExceptHandlerrrWithrrrrrrs                        (     -   (  1 E  2