3 %^1 @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'cCstjt|_dS)N) collections defaultdictlist transforms)selfr 5/tmp/pip-build-8app2_gc/astroid/astroid/transforms.py__init__szTransformVisitor.__init__)maxsizecCsd|j}||jkr|S|j|}x@|D]8\}}|dks<||r$||}|dk rP|}|j|kr$Pq$W|S)zdCall matching transforms for the given node if any and return the transformed node. N) __class__r)rnodeclsrZtransform_func predicateretr r r _transforms   zTransformVisitor._transformcCsLt|drBx6|jD],}t||}|j|}||krt|||qW|j|S)N_astroid_fields)hasattrrgetattr_visit_genericsetattrr)rrnamevaluevisitedr r r _visit2s    zTransformVisitor._visitcsZt|trfdd|DSt|tr=sz3TransformVisitor._visit_generic..c3s|]}j|VqdS)N)r)rr)rr r ?sz2TransformVisitor._visit_generic..) isinstancertuplestrr)rrr )rr r;s  zTransformVisitor._visit_genericNcCs|j|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)rappend)r node_class transformrr r r register_transformEsz#TransformVisitor.register_transformcCs|j|j||fdS)zUnregister the given transform.N)rremove)rr$r%rr r r unregister_transformOsz%TransformVisitor.unregister_transformcs fdd|jD|_j|S)zWalk the given astroid *tree* and transform each encountered node Only the nodes which have transforms registered will actually be replaced or changed. csg|]}j|qSr )r)rr)rr r rYsz*TransformVisitor.visit..)bodyr)rmoduler )rr visitSszTransformVisitor.visit)N)N) __name__ __module__ __qualname____doc__ZTRANSFORM_MAX_CACHE_SIZEr rrrrr&r(r+r r r r r s  r)r functoolsrrr r r r  s