ó î%![c@s2ddlZddlZdefd„ƒYZdS(iÿÿÿÿNtTransformVisitorcBsSeZdZd„Zd„Zd„Zd„Zdd„Zdd„Z d„Z RS( sþA 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. cCstjtƒ|_dS(N(t collectionst defaultdicttlistt transforms(tself((s1lib/python2.7/site-packages/astroid/transforms.pyt__init__scCs¦|j}||jkr|S|j|}|}xp|D]h\}}|dksZ||ƒr6||ƒ}|dk rž||k r’tjd|ƒn|}qžq6q6W|S(sdCall matching transforms for the given node if any and return the transformed node. s"node %s substituted multiple timesN(t __class__RtNonetwarningstwarn(RtnodetclsRt orig_nodettransform_funct predicatetret((s1lib/python2.7/site-packages/astroid/transforms.pyt _transforms     cCsat|dƒrTxB|jD]4}t||ƒ}|j|ƒ}t|||ƒqWn|j|ƒS(Nt_astroid_fields(thasattrRtgetattrt_visit_generictsetattrR(RR tfieldtvaluetvisited((s1lib/python2.7/site-packages/astroid/transforms.pyt_visit/s cset|tƒr/g|D]}ˆj|ƒ^qSt|tƒrXt‡fd†|DƒƒSˆj|ƒS(Nc3s|]}ˆj|ƒVqdS(N(R(t.0tchild(R(s1lib/python2.7/site-packages/astroid/transforms.pys ;s(t isinstanceRRttupleR(RR R((Rs1lib/python2.7/site-packages/astroid/transforms.pyR7s  cCs|j|j||fƒdS(s7Register `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(Rtappend(Rt node_classt transformR((s1lib/python2.7/site-packages/astroid/transforms.pytregister_transform?scCs|j|j||fƒdS(sUnregister the given transform.N(Rtremove(RR R!R((s1lib/python2.7/site-packages/astroid/transforms.pytunregister_transformIscCs5g|jD]}|j|ƒ^q |_|j|ƒS(s±Walk the given astroid *tree* and transform each encountered node Only the nodes which have transforms registered will actually be replaced or changed. (tbodyRR(RtmoduleR((s1lib/python2.7/site-packages/astroid/transforms.pytvisitMs(N( t__name__t __module__t__doc__RRRRRR"R$R'(((s1lib/python2.7/site-packages/astroid/transforms.pyR s     (RR tobjectR(((s1lib/python2.7/site-packages/astroid/transforms.pyts