B a@sdZddlZddlZddlmZmZmZmZmZer@ddl m Z iZ ddZ GdddZ Gd d d Zee e d d d ZddZdS)zIVarious context related utilities, including inference and call contexts.N) TYPE_CHECKINGMutableMappingOptionalSequenceTuple)NodeNGcCs tdS)N)_INFERENCE_CACHEclearr r 3/tmp/pip-unpacked-wheel-mm06h1t3/astroid/context.py_invalidate_cachesr c@seZdZdZdZdZdddZeddZej d dZee e d e e e e e e fed fd d d ZddZddZejddZddZdS)InferenceContextzProvide context for inference Store already inferred nodes to save time Account for already visited nodes to stop infinite recursion )path lookupname callcontext boundnode extra_context_nodes_inferreddNcCs@|dkrdg|_n||_|p t|_d|_d|_d|_i|_dS)Nr)rsetrrrrr)selfrnodes_inferredr r r __init__0s   zInferenceContext.__init__cCs |jdS)z Number of nodes inferred in this context and all its clones/decendents Wrap inner value in a mutable cell to allow for mutating a class variable in the presence of __slots__ r)r)rr r r r]szInferenceContext.nodes_inferredcCs||jd<dS)Nr)r)rvaluer r r rgsr)returncCstS)z Inferred node contexts to their mapped results Currently the key is ``(node, lookupname, callcontext, boundnode)`` and the value is tuple of the inferred results )r)rr r r inferredks zInferenceContext.inferredcCs,|j}||f|jkrdS|j||fdS)zPush node into inference path :return: True if node is already in context path else False :rtype: bool Allows one to see if the given node has already been looked at for this inference contextTF)rradd)rnodenamer r r pushys zInferenceContext.pushcCs0t|j|jd}|j|_|j|_|j|_|S)zClone inference path For example, each side of a binary operation (BinOp) starts with the same context but diverge as each side is inferred so the InferenceContext will need be cloned)r)r rcopyrrrr)rcloner r r r!s zInferenceContext.cloneccst|j}dV||_dS)N)rr)rrr r r restore_paths zInferenceContext.restore_pathcs,fddjD}dtjd|S)Nc3s2|]*}d|tjt|dt|dfVqdS)z%s=%sP)widthN)pprintpformatgetattrlen).0field)rr r sz+InferenceContext.__str__..z{}({})z, ) __slots__formattype__name__join)rstater )rr __str__s  zInferenceContext.__str__)NN)r/ __module__ __qualname____doc__r,Z max_inferredrpropertyrsetterrrrstrrrrr! contextlibcontextmanagerr"r2r r r r r s - 2  r c@seZdZdZdZdddZdS) CallContextz"Holds information for a call site.)argskeywordsNcCs(||_|rdd|D}ng}||_dS)z :param List[NodeNG] args: Call positional arguments :param Union[List[nodes.Keyword], None] keywords: Call keywords cSsg|]}|j|jfqSr )argr)r)r>r r r sz(CallContext.__init__..N)r<r=)rr<r=r r r rs zCallContext.__init__)N)r/r3r4r5r,rr r r r r;sr;)contextrcCs|dk r|StS)z4Clone a context if given, or return a fresh contexxtN)r!r )r@r r r copy_contextsrAcCst|}||_|S)aGive a context a boundnode to retrieve the correct function name or attribute value with from further inference. Do not use an existing context since the boundnode could then be incorrectly propagated higher up in the call stack. :param context: Context to use :type context: Optional(context) :param node: Node to do name lookups from :type node NodeNG: :returns: A new context :rtype: InferenceContext )rAr)r@rr r r bind_context_to_nodesrB)r5r9r%typingrrrrrZastroid.node_classesrrr r r;rArBr r r r s