yIfDddlmZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZer$ddlmZe d e Ze eegefZe d e d e d edfZe ee ee ed fedfZGddZdS)) annotationsN) defaultdict)Callable) TYPE_CHECKINGListOptionalTupleTypeVarUnioncastoverload)_invalidate_cache)SuccessfulInferenceResult TransformFn)nodes_SuccessfulInferenceResultT)bound nodes.NodeNG)r..ceZdZdZd!dZd"dZd#d Zed$d Zed%d Zed&dZed'dZed#dZd(dZ d)d*dZ d)d*dZ d#d Z dS)+TransformVisitoraPA visitor for handling transforms. The standard approach of using it is to call :meth:`~visit` with an *astroid* module and the class will take care of the rest, walking the tree and running the transforms for each encountered node. Based on its usage in AstroidManager.brain, it should not be reinstantiated. returnNonec8tt|_dSN)rlist transforms)selfs F/var/www/piapp/venv/lib/python3.11/site-packages/astroid/transforms.py__init__zTransformVisitor.__init__.s    noderc|j}|j|D]<\}}| ||r*||}|t|}|j|krn=|S)zdCall matching transforms for the given node if any and return the transformed node. ) __class__rr)rr!clstransform_func predicaterets r _transformzTransformVisitor._transform;sxn)-)=   %NI IIdOO $nT**?%'''D=C''E r rc|jD]S}t||}tt|}||}||krt |||T||Sr)_astroid_fieldsgetattrr _Vistables_visit_genericsetattrr()rr!namevaluevisiteds r_visitzTransformVisitor._visitNsu( - -DD$''EU++E))%00G%dG,,,t$$$r cdSrrr!s rr-zTransformVisitor._visit_genericWs25#r strcdSrr4r5s rr-zTransformVisitor._visit_genericZs03r list[nodes.NodeNG]list[SuccessfulInferenceResult]cdSrr4r5s rr-zTransformVisitor._visit_generic]s +.#r tuple[nodes.NodeNG, ...]%tuple[SuccessfulInferenceResult, ...]cdSrr4r5s rr-zTransformVisitor._visit_genericbs 14r cdSrr4r5s rr-zTransformVisitor._visit_genericgsORsr r, _VisitReturnsclt|trfd|DSt|trtfd|DS|rt|tr|S |S#t $r%t jd|dtd|cYSwxYw)Nc:g|]}|Sr4r-.0childrs r z3TransformVisitor._visit_generic..ls'AAA5D''..AAAr c3BK|]}|VdSrrBrCs r z2TransformVisitor._visit_generic..ns1FF,,U33FFFFFFr z Astroid was unable to transform z. Some functionality will be missing unless the system recursion limit is lifted. From pylint, try: --init-hook='import sys; sys.setrecursionlimit(2000)' or higher.r) stacklevel) isinstancertupler6r2RecursionErrorwarningswarn UserWarningr5s` rr-zTransformVisitor._visit_genericjs dD ! ! BAAAADAAA A dE " " GFFFFFFFFF F z$,, K ;;t$$ $    Me4eee     KKK s/B,B32B3N node_class!type[_SuccessfulInferenceResultT] transform(TransformFn[_SuccessfulInferenceResultT]r&._Predicate[_SuccessfulInferenceResultT] | NonecJ|j|||fdS)aJRegister `transform(node)` function to be applied on the given node. The transform will only be applied if `predicate` is None or returns true when called with the node as argument. The transform function may return a value which is then used to substitute the original node in the tree. N)rappendrrPrRr&s rregister_transformz#TransformVisitor.register_transforms*  #**Iy+ABBBBBr cJ|j|||fdS)zUnregister the given transform.N)rremoverWs runregister_transformz%TransformVisitor.unregister_transforms*  #**Iy+ABBBBBr c,||S)zWalk the given astroid *tree* and transform each encountered node. Only the nodes which have transforms registered will actually be replaced or changed. )r2r5s rvisitzTransformVisitor.visits {{4   r )rr)r!rrr)r!rrr)r!rrr)r!r6rr6)r!r8rr9)r!r;rr<)r!r,rr?r)rPrQrRrSr&rTrr) __name__ __module__ __qualname____doc__rr(r2r r-rXr[r]r4r rrr#sP    &%%%%555X5 333X3 ...X.444X4RRRXR2EI CCCCC(EI CCCCC!!!!!!r r) __future__rrM collectionsrcollections.abcrtypingrrrr r r r r astroid.contextrastroid.typingrrastroidrrbool _Predicater6r,r?rr4r rrks #"""""######$$$$$$WWWWWWWWWWWWWWWWWWWW------AAAAAAAAI")'%-F###($?#@$#FGHJ D(%0C*Dc4O  "# #S ()   {!{!{!{!{!{!{!{!{!{!r