B aCq@sDdZddlZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z ddl mZmZmZmZmZddlmZddlmZmZmZmZmZejd krdd lmZn dd lmZed Zed Z d dZ!ddZ"dZ#dddddddddddddd Z$d d!e$%DZ&d"d!e$%DZ'd#d$d%dd&Z(ej)ej*ej+ej,d&Z-d'd(Z.d)d*ej/_0d+d*ej1_0d,d*ej2_0d-d*ej3_0d.d*ej4_0d/d*d0d*d1d*d2d*d3d*d4d*d5d*d6d*d7d*d8d*d9d*d:d*ej5d Z6x&e7e6%D]\Z8Z9e9e6e8d;<qWej:dd?Z=d@dAZ>ej:dBdCZ?e?ej/_srzcontextlib.contextmanager__add____sub__ __truediv__ __floordiv____mul____pow____mod____and____or____xor__ __lshift__ __rshift__ __matmul__) +-/z//*z**%&|^z<>@cCsi|]\}}t||qSr)r).0keyvaluerrr Ssr6cCsi|]\}}t||dqS)=)r)r3r4r5rrrr6Vs__pos____neg__ __invert__)r*r+~notcCst|}||}t|S)N)_UNARY_OPERATORSr const_factory)objopfuncr5rrr_infer_unary_ophsrBcCstt|j|S)N)rBtupleelts)selfr@rrrnrFcCs t|j|S)N)rBrD)rEr@rrrrForGcCstt|j|S)N)rBsetrD)rEr@rrrrFprGcCs t|j|S)N)rBr5)rEr@rrrrFqrGcCstt|j|S)N)rBdictitems)rEr@rrrrFrrGcCs||S)Nr)abrrrrFwrGcCs||S)Nr)rKrLrrrrFxrGcCs||S)Nr)rKrLrrrrFyrGcCs||S)Nr)rKrLrrrrFzrGcCs||S)Nr)rKrLrrrrF{rGcCs||S)Nr)rKrLrrrrF|rGcCs||S)Nr)rKrLrrrrF}rGcCs||@S)Nr)rKrLrrrrF~rGcCs||BS)Nr)rKrLrrrrFrGcCs||AS)Nr)rKrLrrrrFrGcCs||>S)Nr)rKrLrrrrFrGcCs||?S)Nr)rKrLrrrrFrGr7ccstt}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_IMPLr>r5 TypeError Exceptionr Uninferablestr)rEopnodeoperatorotherr_not_implementedimplrrrconst_infer_binary_ops    r[cs4|j|d}fdd|jD}t||j|_|S)N)parentc3s*|]"}|tjk rt|p tjVqdS)N)r rSrZ safe_infer)r3elt)rrr sz'_multiply_seq_by_int..) __class__rDlistr5)rErUrWrnodeZ filtered_eltsr)rr_multiply_seq_by_ints    rbccsXxR|D]J}|tjkr tVqx.||D] }|tjk rB|Vq,tVq,WqWdS)N)r rSr Unknowninfer)rDrr]inferredrrr_filter_uninferable_nodess    rfc 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*)r\r-)r rMrNrOr_r` itertoolschainrfrDr5intrbrInstancerZclass_instance_as_indexr rS) rErUrVrWrmethodrYraZas_indexrrrtl_infer_binary_ops&      rlcCs |||S)N)infer_call_result)rErUrVrWrrkrrrinstance_class_infer_binary_opsrnc cs|dd}|d}x|D]}|tjkr,qt|ds8qy |}Wntk rZwYnXx|D]}t|}y|||}Wnt t t fk rwbYnX|s|Vqb|tjkrPqbyt | |||EdHWqbtk rPYqbXqbWqWdS)z;recursive function to resolve multiple assignments on loopsNritered)popr rShasattrrorQr rMgetitemAttributeErrorrr _resolve_looppartrdr) parts assign_pathrindexpartrostmt index_nodeassignedrrrrts4         rtccst|tjst|ddr(t||||dS|dkrfxN|j|D]"}t|tjtjfr>|j EdHq>Wnt |j|||EdHt||||dS)NZis_asyncF)raunknownrvr) rOr ZAsyncForgetattrrIiterrdTupleListrDrt)rErarrvlstrrrfor_assigned_stmtssrc Csp|dkr g}y|j|}Wn4tk rP}ztd|||d|Wdd}~XYnX|d||jj|||dS)Nz6Tried to retrieve a node {node!r} which does not exist)rarvrr)rarrv)rDrw ValueErrorrinsertr\assigned_stmts)rErarrvrwexcrrrsequence_assigned_stmts&s rcCs|jj||dS)N)rar)r\r)rErarrvrrrassend_assigned_stmts=src cs|js|js|jstjVdS|jj}|jrt|jddd|kr|dkr|jj}t |t j oj|jdk}|j rt |j t jr|j j}|s|dkr|VdS|dkr|VdS|r|jrt|j|j}||j||EdHdS||jkrrrD contextmod copy_context default_valuerdr) rErrZfunctypeclsZ is_metaclassZ call_siterrrrr_arguments_infer_argnameEsP            rcCsJ|jr<|j}t|}d|_tj||d}||j|j|St||j|S)N)r) rrrrrrr\rr)rErarrvrargsrrrarguments_assigned_stmts|s rccs:|s|jVdSt|j|||EdHt||||dS)N)rar|rvr)r5_resolve_assignment_partsrdrI)rErarrvrrrassign_assigned_stmtss rccs4x.t||||D]}|dkr&tjVq|VqWdS)N)rr rS)rErarrvrerrrassign_annassigned_stmtss rc cs|dd}|d}x|D]}d}t|tjrZy|j|\}}Wqtk rVdSXn>t|drt|}y|||}Wnt t fk rdSX|sdS|s|Vq|t j krdSyt ||||EdHWqtk rdSXqWdS)z2recursive function to resolve multiple assignmentsNrrr)rprOr DictrJ IndexErrorrqrMrrrr r rSrrdr)rurvrrwrxr{rXrzrrrrs4       rccsBx0t|jD] }t|tjr(t|}|VqWt||||dS)N)rar|rvr) r Z unpack_inferrrOr rrZExceptionInstancerI)rErarrvr{rrrexcepthandler_assigned_stmtss    rc csyt|j|d}Wn.tk rB}zt|d|Wdd}~XYnXt|tjr|j}|jshtd|dxF|jj D]0}t|j|dd}t|t j rr| t krrPqrWt|dyt| VWn.tk r}zt|d|Wdd}~XYnXnt|tjr|yt|jd|d}Wn6tttfk rN} zt|d| Wdd} ~ XYnXt|tjsht|d|||EdHn t|ddS)N)r)raz,No decorators found on inferred generator %s __enter__)nextrd StopIterationrrOrrr\rr Z FunctionDefqname_CONTEXTLIB_MGRZinfer_yield_typesrjZigetattrrZ BoundMethodrm) rEmgrrreerAZdecorator_node decoratorZenterrrrr_infer_context_managers8       rc #s$ytfdd|jD}Wntk r0dSX|dkrNt|||EdHnxt|||D]}|}x|D]}t|dstd|||dy|j|}Wqjtk r}ztd|||d|Wdd}~XYqjtk r}ztd|||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)r3rvars)rarrr^sz&with_assigned_stmts..NrDz0Wrong type ({targets!r}) for {node!r} assignment)ratargetsrvrzCTried to infer a nonexistent target with index {index} in {node!r}.z1Tried to unpack a non-iterable value in {node!r}.)rar|rvr) rrJrrrqrrDrrQrI) rErarrvrresultr?rwrr)rarwith_assigned_stmtssH   rccs4|j|kr |jj|dEdHntd|||ddS)z9Infer names and other nodes from an assignment expression)rNz$Cannot infer NamedExpr node {node!r})rarvr)targetr5rdr)rErarrvrrrnamed_expr_assigned_stmtsGs rc #sfdd|}t|tjtjfs8td||||d|dkrHt}t|tjr|j}|j d}t dd| tj Dd krtd ||||d yt ||}Wn ttfk rtjVdSX|tjkst|d stjVdSyt|}Wntk rtjVdSXxt|jD]\} } t| tj sR|sFP|q(tt|j| d} x`| D]X} t| tj s|sP|qptjtj||j|j d } | j!t"|d| VPqpWq(Wt|tjryt |j#j|d}Wn"ttfk rtjVdSX|tjks2t|d s>tjVdSy |}Wntk rhtjVdSX|j$}t|tj%std|dg}||||std|d|d\}}||d k}t&||rdn||}||d<x|D]}d}x~|D]v}t|d s Pt|t&s|d}y|}||}Wn4t'k rJPYn"tk rftjVdSX|}qWtjtj||j|j d }|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) ro enumeraterOr Starredr5rappendlenr)Zstarredrlookupsrorwelement)$_determine_starred_iteration_lookupsrrrds  zDstarred_assigned_stmts.._determine_starred_iteration_lookupszDStatement {stmt!r} enclosing {node!r} must be an Assign or For node.)raryr|rNrcss|] }dVqdS)Nr)r3rXrrrr^sz)starred_assigned_stmts..rz>Too many starred arguments in the assignment targets {lhs!r}.)rarr|rro)ctxr\lineno col_offset)rD)rz8Could not make sense of this, the target must be a tuplez5Could not make sense of this, needs at least a lookup)(Z statementrOr AssignForrrInferenceContextr5rsumZnodes_of_classrrrdrr rSrq collectionsdequerorQrrDpopleftreversedrprr ZStorerrZpostinitr`r~rrslicer)rErarrvryr5lhsrhsrDrwZ left_nodeZlhs_eltsZ right_nodepackedZinferred_iterablerorrZlast_element_indexZlast_element_lengthZis_starred_lastZ lookup_slicerZ found_elementlookupZitered_inner_elementunpackedr)rrstarred_assigned_stmtsXs                  r)rErarrvreturnccsdS)zbReturn empty generator (return -> raises StopIteration) so inferred value is Uninferable. Nr)rErarrvrrrmatch_mapping_assigned_stmtss rccsdS)zbReturn empty generator (return -> raises StopIteration) so inferred value is Uninferable. Nr)rErarrvrrrmatch_star_assigned_stmts%s rccs8t|jtjr4t|jjtjr4|jdkr4|jjjVdS)z}Infer MatchAs as the Match subject if it's the only MatchCase pattern else raise StopIteration to yield Uninferable. N)rOr\r Z MatchCaseMatchpatternsubject)rErarrvrrrmatch_as_assigned_stmts6s  r)NNN)NNN)NNN)NNN)NNN)NNN)NNN)NNN)NN)NNN)NN)NN)NN)g__doc__rrgrVZ operator_modsystypingrrZastroidrrrrrrr r r Z astroid.constr Zastroid.exceptionsr rrrr version_infortyping_extensionsZ lazy_importrrrrrZ BIN_OP_METHODrJZREFLECTED_BIN_OP_METHODZAUGMENTED_OP_METHODZUNARY_OP_METHODposneginvertnot_r=rBrZinfer_unary_oprSetrMrmatmulrPr`Z_KEYZ_IMPLZyes_if_nothing_inferredr[Zinfer_binary_oprbrfrlrnrjrrtZraise_if_nothing_inferredrrrZ ComprehensionrrZ AssignNameZ AssignAttrrr ArgumentsrrrZ AnnAssignZ AugAssignrrZ ExceptHandlerrrZWithrZ NamedExprrrZ MatchMappingrZNodeNGrZ MatchStarrZMatchAsrrrrrs                 $   7  ( & E  9