o b1 @s&ddlZddlmZGdddZdS)N) lru_cachec@sZeZdZdZdZddZeedddZdd Zd d Z dd dZ dddZ ddZ d S)TransformVisitorzA 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. i'cCstt|_dSN) collections defaultdictlist transformsselfr Q/var/www/secure340b-portal/env/lib/python3.10/site-packages/astroid/transforms.py__init__szTransformVisitor.__init__)maxsizecCsd|j}||jvr |S|j|}|D]\}}|dus||r/||}|dur'|}|j|kr/|Sq|S)zdCall matching transforms for the given node if any and return the transformed node. N) __class__r)r nodeclsrZtransform_func predicateretr r r _transforms    zTransformVisitor._transformcCsHt|dr|jD]}t||}||}||krt|||q||S)N_astroid_fields)hasattrrgetattr_visit_genericsetattrr)r rnamevaluevisitedr r r _visit2s      zTransformVisitor._visitcsXt|trfdd|DSt|trtfdd|DS|r%t|tr'|S|S)Ncg|]}|qSr r.0childr r r =z3TransformVisitor._visit_generic..c3s|]}|VqdSrrr r r r ?sz2TransformVisitor._visit_generic..) isinstancertuplestrr)r rr r r r;s   zTransformVisitor._visit_genericNcC|j|||fdS)a7Register `transform(node)` function to be applied on the given astroid's `node_class` 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)rappendr Z node_classZ transformrr r r register_transformEsz#TransformVisitor.register_transformcCr))zUnregister the given transform.N)rremover+r r r unregister_transformOsz%TransformVisitor.unregister_transformcs fdd|jD|_|S)zWalk the given astroid *tree* and transform each encountered node Only the nodes which have transforms registered will actually be replaced or changed. crr )rr r r r r#Yr$z*TransformVisitor.visit..)bodyr)r moduler r r visitSs zTransformVisitor.visitr) __name__ __module__ __qualname____doc__ZTRANSFORM_MAX_CACHE_SIZEr rrrrr,r.r1r r r r r s  r)r functoolsrrr r r r s