B S\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'cCstt|_dS)N) collections defaultdictlist transforms)selfr 1lib/python3.7/site-packages/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 predicateZretr r r _transforms   zTransformVisitor._transformcCsLt|drBx6|jD],}t||}||}||krt|||qW||S)N_astroid_fields)hasattrrgetattr_visit_genericsetattrr)rrnamevalueZvisitedr r r _visit2s    zTransformVisitor._visitcsXt|trfdd|DSt|tr=sz3TransformVisitor._visit_generic..c3s|]}|VqdS)N)r)rr)rr r ?sz2TransformVisitor._visit_generic..) isinstancertuplestrr)rrr )rr r;s  zTransformVisitor._visit_genericNcCs|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|||fdS)zUnregister the given transform.N)rremove)rr"r#rr 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. csg|]}|qSr )r)rr)rr r rYsz*TransformVisitor.visit..)Zbodyr)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