XylotrechusZ
�
� h� � �\ � d dl mZ d dlZd dlmZ d dlmZ erd dlmZ G d� d� � Z dS )� )� annotationsN)�
TYPE_CHECKING)�_invalidate_cache)�NodeNGc �F � e Zd ZdZd� Zdd�Zd� Zd� Zddd �Zddd �Z d
� Z
d S )�TransformVisitoraP 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.
Based on its usage in AstroidManager.brain, it should not be reinstantiated.
c �B � t j t � � | _ d S �N)� collections� defaultdict�list�
transforms)�selfs �c/builddir/build/BUILD/cloudlinux-venv-1.0.8/venv/lib/python3.11/site-packages/astroid/transforms.py�__init__zTransformVisitor.__init__ s � �%�1�$�7�7����� �noder �returnc � � |j }| j | }|D ]<\ }}|� ||� � r* ||� � }|�t � � |}|j |k r n�=|S )zdCall matching transforms for the given node if any and return the
transformed node.
)� __class__r r )r r �clsr �transform_func� predicate�rets r �
_transformzTransformVisitor._transform s} � � �n���_�S�)�
�)3�
�
� %�N�I�� �I�I�d�O�O� �$�n�T�*�*�� �?�%�'�'�'� �D��=�C�'�'��E��� r c �� � t |d� � rF|j D ]>}t ||� � }| � |� � }||k rt |||� � �?| � |� � S )N�_astroid_fields)�hasattrr �getattr�_visit_generic�setattrr )r r �name�value�visiteds r �_visitzTransformVisitor._visit2 sy � � �4�*� +� +� 1� �,�
1�
1�� ��d�+�+�� �-�-�e�4�4���e�#�#��D�$��0�0�0�����t�$�$�$r c � � � t |t � � r� fd�|D � � S t |t � � rt � fd�|D � � � � S |rt |t � � r|S � � |� � S )Nc �: �� g | ]}�� |� � ��S � �r ��.0�childr s �r �
<listcomp>z3TransformVisitor._visit_generic.<locals>.<listcomp>= s' �� �A�A�A�5�D�'�'��.�.�A�A�Ar c 3 �B �K � | ]}�� |� � V � �d S r
r) r* s �r � <genexpr>z2TransformVisitor._visit_generic.<locals>.<genexpr>? s1 �� � � �F�F���,�,�U�3�3�F�F�F�F�F�Fr )�
isinstancer
�tuple�strr% )r r s ` r r z TransformVisitor._visit_generic; s� �� � �d�D� !� !� B�A�A�A�A�D�A�A�A� A� �d�E� "� "� G��F�F�F�F��F�F�F�F�F� F�� �z�$��,�,� ��K��{�{�4� � � r N�Nonec �J � | j | � ||f� � dS )a7 Register `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)r �append�r �
node_class� transformr s r �register_transformz#TransformVisitor.register_transformE s* � �
��
�#�*�*�I�y�+A�B�B�B�B�Br c �J � | j | � ||f� � dS )z Unregister the given transform.N)r �remover6 s r �unregister_transformz%TransformVisitor.unregister_transformO s( � � ��
�#�*�*�I�y�+A�B�B�B�B�Br c �, � | � |� � S )z�Walk the given astroid *tree* and transform each encountered node.
Only the nodes which have transforms registered will actually
be replaced or changed.
)r% )r �modules r �visitzTransformVisitor.visitS s � � �{�{�6�"�"�"r )r r r r r
)r r3 ) �__name__�
__module__� __qualname__�__doc__r r r% r r9 r< r? r( r r r r s� � � � � � �� �8� 8� 8�� � � �(%� %� %�!� !� !�C� C� C� C� C�C� C� C� C� C�#� #� #� #� #r r )
�
__future__r r �typingr �astroid.contextr �astroidr r r( r r �<module>rH s� ��
#� "� "� "� "� "� � � � � � � � � � � -� -� -� -� -� -�� � � � � � � �I#� I#� I#� I#� I#� I#� I#� I#� I#� I#r