B S\g@sbdZddlZddlZddlZddlZddlmZddlmZddlm 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?<x&e,e)D]\Z-Z.e.e)e-d@<qWe j/dAdBZ0e0ej'_1dCdDZ2dEdFZ3e j/dGdHZ4e4ej#_1e4ej%_1e j/dIdJZ5e5e j6_1e5ej7_1dKdLZ8e j9dedMdNZ:e:ej;_e>ej#_ej%_>@cCsi|]\}}t||qSr)r).0keyvaluerrr Esr-cCsi|]\}}t||dqS)=)r)r*r+r,rrrr-Hs__pos____neg__ __invert__)r!r"~notcCst|}||}t|S)N)_UNARY_OPERATORSr const_factory)objopfuncr,rrr_infer_unary_opZsr9cCstt|j|S)N)r9tupleelts)selfr7rrr`sr=cCs t|j|S)N)r9r;)r<r7rrrr=ascCstt|j|S)N)r9setr;)r<r7rrrr=bscCs t|j|S)N)r9r,)r<r7rrrr=cscCstt|j|S)N)r9dictitems)r<r7rrrr=dscCs||S)Nr)abrrrr=iscCs||S)Nr)rArBrrrr=jscCs||S)Nr)rArBrrrr=kscCs||S)Nr)rArBrrrr=lscCs||S)Nr)rArBrrrr=mscCs||S)Nr)rArBrrrr=nscCs||S)Nr)rArBrrrr=oscCs||@S)Nr)rArBrrrr=pscCs||BS)Nr)rArBrrrr=qscCs||AS)Nr)rArBrrrr=rscCs||>S)Nr)rArBrrrr=sscCs||?S)Nr)rArBrrrr=ts) r!r"r#z//r$z**r%r&r'r(z<>)r)r.ccstt}t|tjry\t|}yt||j|jVWn4tk rT|VYntk rnt j VYnXWqtk r|VYqXn$t|jt r|dkrt j Vn|VdS)Nr%) r ConstNotImplemented isinstance BIN_OP_IMPLr5r, TypeError Exceptionr Uninferablestr)r<opnodeoperatorotherr_not_implementedimplrrrconst_infer_binary_op~s    rScs4|j|d}fdd|jD}t||j|_|S)N)parentc3s*|]"}|tjk rt|p tjVqdS)N)r rKr Z safe_infer)r*elt)rrr sz'_multiply_seq_by_int..) __class__r;listr,)r<rMrOrnodeZ filtered_eltsr)rr_multiply_seq_by_ints    rZccsXxR|D]J}|tjkr tVqx.||D] }|tjk rB|Vq,tVq,WqWdS)N)r rKr ZUnknowninfer)r;rrUinferredrrr_filter_uninferable_nodess    r]c cstt}t||jrT|dkrT|j|d}ttt|j |t|j ||_ |Vnt|tjr|dkrt|j t s~|VdSt ||||VnDt|t jr|dkrt|}|stjVqt ||||Vn|VdS)Nr!)rTr$)r rErFrGrWrX itertoolschainr]r;r,intrZrInstancer Zclass_instance_as_indexr rK) r<rMrNrOrmethodrQrYZas_indexrrrtl_infer_binary_ops&      rccCs |||S)N)infer_call_result)r<rMrNrOrrbrrrinstance_class_infer_binary_opsrec cs|dd}|d}x|D]}|tjkr,qt|ds8qy |}Wntk rZwYnXx|D]}t|}y|||}Wn t t j t j fk rwbYnX|s|Vqb|tjkrPqbyt ||||EdHWqbt jk rPYqbXqbWqWdS)z;recursive function to resolve multiple assignments on loopsNritered)popr rKhasattrrfrIr rEgetitemAttributeErrorrAstroidTypeErrorAstroidIndexError_resolve_looppartr[InferenceError) parts assign_pathrindexpartrfstmt index_nodeassignedrrrrms:         rmccst|tjst|ddr(t||||dS|dkrfxN|j|D]"}t|tjtjfr>|j EdHq>Wnt |j|||EdHt||||dS)NZis_asyncF)rYunknownrpr) rGr ZAsyncForgetattrr?iterr[TupleListr;rm)r<rYrrpZlstrrrfor_assigned_stmts sr{c Csr|dkr g}y|j|}Wn6tk rR}ztjd|||d|Wdd}~XYnX|d||jj|||dS)Nz6Tried to retrieve a node {node!r} which does not exist)rYrprr)rYrrp)r;rq ValueErrorrrninsertrTassigned_stmts)r<rYrrprqexcrrrsequence_assigned_stmtss rcCs|jj||dS)N)rYr)rTr~)r<rYrrprrrassend_assigned_stmts6src csb|js|js|jstjVdS|jrt|jddd|kr|jj}|jj}t |t j ob|jdk}|sp|dkrz|VdS|dkrt |VdS|r|jrt|j|j}||j||EdHdS||jkrt d}||_|VdS||jkrt i}||_|VdSy,t|}|||EdHtjVWn tjk r\tjVYnXdS)Nrr metaclass classmethodrbr)argsvarargkwargr rKrwrTtypescoperGr ClassDefrra callcontextrCallSiteZ extra_contextinfer_argumentr5 contextmod copy_contextZ default_valuer[rZ NoDefault) r<rrZfunctypeclsZ is_metaclassZ call_siterrrrr_arguments_infer_argname>sB          rcCsF|jr8|j}t|}d|_t|}||j|j|St||j|S)N) rrrrrrrTrr)r<rYrrprrrrrarguments_assigned_stmtsks  rccs:|s|jVdSt|j|||EdHt||||dS)N)rYrvrpr)r,_resolve_assignment_partsr[r?)r<rYrrprrrassign_assigned_stmtsys rccs4x.t||||D]}|dkr&tjVq|VqWdS)N)rr rK)r<rYrrpr\rrrassign_annassigned_stmtss rc cs|dd}|d}x|D]}d}t|tjrZy|j|\}}Wqtk rVdSXnBt|drt|}y|||}Wnt j t j fk rdSX|sdS|s|Vq|t j krdSyt||||EdHWqt jk rdSXqWdS)z2recursive function to resolve multiple assignmentsNrri)rgrGr Dictr@ IndexErrorrhrErirrkrlr rKrr[rn)rorprrqrrrurPrtrrrrs4       rccsBx0t|jD] }t|tjr(t|}|VqWt||||dS)N)rYrvrpr) rZ unpack_inferrrGr rr ZExceptionInstancer?)r<rYrrprurrrexcepthandler_assigned_stmtss    rc #st|j|d}t|tjr|jjs6tjddxDjj D],}t||}t|t j r@| t kr@Pq@Wtjd t j}tfdd|Dd}|r|jst d}||_|j|_|Vn|jj|dEdHnt|tjrvyt|jd|d} Wn*tjtjfk r.tj|dYnXt| tjsJtj| d|jsbtj|gd|_| ||EdHn tj|ddS)N)rz,No decorators found on inferred generator %s)rYc3s|]}|kr|VqdS)N)r)r*rY)r8rrrVsz)_infer_context_manager.. __enter__)r)nextr[rGr GeneratorrTrrrnr Z FunctionDefZqname_CONTEXTLIB_MGRnodes_of_classZYieldr,rElinenoraZigetattrZAttributeInferenceErrorZ BoundMethodrrZ CallContextrd) r<mgrrr\Zdecorator_nodeZ decoratorZpossible_yield_pointsZ yield_pointZconstZenterr)r8r_infer_context_managersB        rc #s*ytfdd|jD}Wntk r0dSX|dkrNt|||EdHnxt|||D]}|}x|D]}t|dstjd|||dy|j|}Wqjtk r}ztjd|||d|Wdd}~XYqjt k r }ztjd|||d|Wdd}~XYqjXqjW|Vq\Wt |||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|VqdS)Nr)r*rvars)rYrrrVsz&with_assigned_stmts..Nr;z0Wrong type ({targets!r}) for {node!r} assignment)rYtargetsrprzCTried to infer a nonexistent target with index {index} in {node!r}.z1Tried to unpack a non-iterable value in {node!r}.)rYrvrpr) rr@ StopIterationrrhrrnr;rrIr?) r<rYrrprresultr6rqrr)rYrwith_assigned_stmtssH   rc #sfdd|}t|tjtjfs:tjd||||d|dkrJt}t|tjr|j }|j d}t dd| tj Dd krtjd ||||d yt||}Wntjk rtjVdSX|tjkst|d stjVdSyt|}Wntk rtjVdSXxt|jD]\} } t| tj sT|sHP|q*tt|j| d} xZ| D]R} t| tj s|sP|qrtjt||j|jd } | j |d| VPqrWq*Wt|tjryt|j!j|d}Wn tjk rtjVdSX|tjks,t|d s8tjVdSy |}Wntk rbtjVdSX|j"}t|tj#stjd|dg}||||stjd|d|d\}}||d k}t$||rdn||}||d<x|D]}d}x~|D]v}t|d sPt|t$s|d}y|}||}Wn4t%k rHPYn"tk rdtjVdSX|}qWtjt||j|jd }|j |pgd|VdSWtjVdS)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|}xrt|D]f\}}t|tjrJ|jj|jjkrJ||t|fPt|tj r||t|f|||qWdS)N) rf enumeraterGr Starredr,rappendlenry)Zstarredtargetlookupsrfrqelement)$_determine_starred_iteration_lookupsrrrMs  zDstarred_assigned_stmts.._determine_starred_iteration_lookupszDStatement {stmt!r} enclosing {node!r} must be an Assign or For node.)rYrsrvrNrcss|] }dVqdS)Nr)r*rPrrrrVlsz)starred_assigned_stmts..rz>Too many starred arguments in the assignment targets {lhs!r}.)rYrrvrrf)ZctxrTr col_offset)r;)rz8Could not make sense of this, the target must be a tuplez5Could not make sense of this, needs at least a lookup)&Z statementrGr AssignForrrnrZInferenceContextr,rsumrrrr[r rKrh collectionsdequerfrIrr;popleftreversedrgrzrrrZpostinitrxrryslicer)r<rYrrprsr,ZlhsZrhsr;rqZ left_nodeZlhs_eltsZ right_nodeZpackedZinferred_iterablerfrrZlast_element_indexZlast_element_lengthZis_starred_lastZ lookup_slicerZ found_elementlookupZitered_inner_elementZunpackedr)rrstarred_assigned_stmtsAs                 r)NNN)NNN)NNN)NNN)NNN)NNN)NNN)NNN)NNN)R__doc__rrNZ operator_modsysr^Zastroidrrrrrrrrr r r Z lazy_importr r rrrZ BIN_OP_METHODr@ZREFLECTED_BIN_OP_METHODZAUGMENTED_OP_METHODZUNARY_OP_METHODposneginvertnot_r4r9ryZinfer_unary_oprzSetrErrH version_infomatmulrXZ_KEYZ_IMPLZyes_if_nothing_inferredrSZinfer_binary_oprZr]rcrerarrmZraise_if_nothing_inferredr{rr~Z ComprehensionrrZ AssignNameZ AssignAttrrrZ ArgumentsrrrZ AnnAssignZ AugAssignrrZ ExceptHandlerrrZWithrrrrrrs                         (   -  ( 1 E 2