B S\,@s\dZddlZddlZddlmZGdddZGdddZeeedd d Zd d ZdS) zIVarious context related utilities, including inference and call contexts.N)Optionalc@sLeZdZdZdZdddZddZdd Zd d Ze j d d Z ddZ dS)InferenceContextzProvide context for inference Store already inferred nodes to save time Account for already visited nodes to infinite stop infinite recursion )path lookupname callcontext boundnodeinferred extra_contextNcCs2|pt|_d|_d|_d|_|p$i|_i|_dS)N)setrrrrrr )selfrrr .lib/python3.7/site-packages/astroid/context.py__init__s   zInferenceContext.__init__cCs,|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)r nodenamer r r pushPs zInferenceContext.pushcCs,t|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)rrrrrr )r cloner r r r_s zInferenceContext.cloneccs4g}x|D]}|||Vq Wt||j|<dS)zRCache result of generator into dictionary Used to cache inference resultsN)appendtupler)r key generatorZresultsresultr r r cache_generatorls    z InferenceContext.cache_generatorccst|j}dV||_dS)N)r r)r rr r r restore_pathws zInferenceContext.restore_pathcs,fddjD}dtjd|fS)Nc3s2|]*}d|tjt|dt|dfVqdS)z%s=%sP)widthN)pprintZpformatgetattrlen).0Zfield)r r r sz+InferenceContext.__str__..z%s(%s)z, ) __slots__type__name__join)r stater )r r __str__}s  zInferenceContext.__str__)NN) r$ __module__ __qualname____doc__r"rrrr contextlibcontextmanagerrr'r r r r rs 1  rc@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 )argvalue)r r0r r r sz(CallContext.__init__..N)r.r/)r r.r/r r r rs zCallContext.__init__)N)r$r(r)r*r"rr r r r r-sr-)contextreturncCs|dk r|StS)z4Clone a context if given, or return a fresh contexxtN)rr)r3r r r copy_contextsr5cCst|}||_|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 )r5r)r3rr r r bind_context_to_nodesr6) r*r+rtypingrrr-r5r6r r r r  s w