�
c��j� � �� � d Z dZddlZddlmZmZmZmZmZm Z ddl
m Z ddl
m Z G d� d e
� Zd
� Z G d � d e� Zej$ ej& ej( dd
�Zd� Zd� Zd� Zy)z�Pattern compiler.
The grammar is taken from PatternGrammar.txt.
The compiler compiles a pattern to a pytree.*Pattern instance.
z#Guido van Rossum � N� )�driver�literals�token�tokenize�parse�grammar)�pytree)�pygramc � � e Zd Zy)�PatternSyntaxErrorN)�__name__�
__module__� __qualname__� � �(/usr/lib64/python3.12/lib2to3/patcomp.pyr
r
s � �r r
c # �� K � t j t j t j h}t j
t
j | � j � }|D ] }|\ }}}}}||vs�|�� � y�w)z6Tokenizes a string suppressing significant whitespace.N) r �NEWLINE�INDENT�DEDENTr �generate_tokens�io�StringIO�readline) �input�skip�tokens� quintuple�type�value�start�end� line_texts r �tokenize_wrapperr% sd � �� � �M�M�5�<�<�� � � 6�D�
�
%�
%�b�k�k�%�&8�&A�&A�
B�F�� �-6�*��e�U�C�� �t� ��O� �s �A3A=�6A=c �0 � e Zd Zdd�Zdd�Zd� Zdd�Zd� Zy) �PatternCompilerNc � � |�+t j | _ t j | _ n>t j |� | _ t j | j � | _ t j | _ t j | _ t j | j t �� | _ y)z^Initializer.
Takes an optional alternative filename for the pattern grammar.
N)�convert)r �pattern_grammarr �pattern_symbols�symsr � load_grammar�Symbols�python_grammar� pygrammar�python_symbols�pysyms�Driver�pattern_convert)�self� grammar_files r �__init__zPatternCompiler.__init__( sz � �
� �!�1�1�D�L� �.�.�D�I�!�.�.�|�<�D�L� ���t�|�|�4�D�I� �.�.��� �+�+�� � �m�m�D�L�L�/�J�� r c � � t |� } | j j ||�� }|r| j |� |fS | j |� S # t j $ r}t t
|� � d�d}~ww xY w)z=Compiles a pattern string to a nested pytree.*Pattern object.)�debugN)r% r � parse_tokensr �
ParseErrorr
�str� compile_node)r5 r r9 � with_treer �root�es r �compile_patternz PatternCompiler.compile_pattern7 s � �!�%�(�� 7��;�;�+�+�F�%�+�@�D� ��$�$�T�*�D�0� 0��$�$�T�*� *�� � � � 7�$�S��V�,�$� 6�� 7�s � A �A=�#A8�8A=c � � |j | j j k( r|j d }|j | j j k( rx|j ddd� D �cg c] }| j |� �� }}t
|� dk( r|d S t j |D �cg c] }|g�� c}dd�� }|j � S |j | j j k( rd|j D �cg c] }| j |� �� }}t
|� dk( r|d S t j |gdd�� }|j � S |j | j j k( rC| j |j dd � }t j |� }|j � S d}|j } t
| � dk\ r4| d j t j k( r| d j }| dd } d}
t
| � dk\ r0| d j | j j" k( r
| d }
| dd } | j | |
� }|
��|
j } | d } | j t j$ k( rd}
t j& }n�| j t j( k( rd}
t j& }nV| j t j* k( r9| j- | d � x}
}t
| � dk( r| j- | d � }
dk7 sdk7 r*|j � }t j |gg|
�� }|�||_ |j � S c c}w c c}w c c}w ) zXCompiles a node, recursively.
This is one big switch on the node type.
r N� r ��min�max� ������ )r r, �Matcher�children� Alternativesr= �lenr
�WildcardPattern�optimize� Alternative� NegatedUnit�
compile_basic�NegatedPatternr �EQUALr! �Repeater�STAR�HUGE�PLUS�LBRACE�get_int�name)r5 �node�ch�alts�a�p�units�patternr[ �nodes�repeatrK �childrE rF s r r= z PatternCompiler.compile_nodeC s � � �9�9�� � �)�)� )��=�=��#�D� �9�9�� � �.�.� .�48�M�M�#�A�#�4F�G�4F�b�D�%�%�b�)�4F�D�G��4�y�A�~��A�w���&�&�T�':�T����T�':��q�I�A��:�:�<� � �9�9�� � �-�-� -�59�]�]�C�]�r�T�&�&�r�*�]�E�C��5�z�Q�� �Q�x���&�&��w�A�1�=�A��:�:�<� � �9�9�� � �-�-� -��(�(�� � �q�r�):�;�G��%�%�g�.�A��:�:�<� � ���
�
�� �u�:��?�u�Q�x�}�}�� � � ;���8�>�>�D��!�"�I�E��� �u�:��?�u�R�y�~�~����1C�1C� C��2�Y�F��#�2�J�E� �$�$�U�F�3�� � � ���H� �Q�K�E��z�z�U�Z�Z�'��� �k�k�����u�z�z�)��� �k�k�����u�|�|�+� !�L�L��!�� 5�5��c��x�=�A�%� �,�,�x��{�3�C� �a�x�3�!�8�!�*�*�,�� �0�0�7�)��#�3�O�� � � �G�L�� � �!�!��s H�� (;�� Ds �*M3�*
M8�?M=c � � |d }|j t j k( rGt t j
|j � � }t j t |� |� S |j t j k( r�|j }|j � rB|t vrt d|z � �|dd r t d� �t j t | � S |dk( rd }n8|j d� s't | j |d � }|�t d|z � �|dd r#| j# |d j$ d � g}nd }t j& |� S |j dk( r| j# |d � S |j d k( r.| j# |d � }t j( |ggdd�
� S y ) Nr zInvalid token: %rr z Can't have details for token�any�_zInvalid symbol: %r�(�[rD )r r �STRINGr< r �
evalStringr! r
� LeafPattern�_type_of_literal�NAME�isupper� TOKEN_MAPr
�
startswith�getattrr2 r= rK � NodePatternrN )r5 rc rd r\ r! r �content�
subpatterns r rR z PatternCompiler.compile_basic� s� � � �Q�x�� �9�9�� � � $���+�+�D�J�J�7�8�E��%�%�&6�u�&=�u�E� E�
�Y�Y�%�*�*�
$��J�J�E��}�}��� �)�,�-@�5�-H�I�I����9�,�-K�L�L� �)�)�)�E�*:�;�;��E�>� �D� �)�)�#�.�"�4�;�;��t�<�D��|� 0�1E��1M� N�N����9� #� 0� 0��q��1B�1B�1�1E� F� G�G� "�G� �)�)�$��8�8�
�Z�Z�3�
��$�$�U�1�X�.� .�
�Z�Z�3�
� �*�*�5��8�4�J��)�)�J�<�.�a�Q�G� G� r c �, � t |j � S �N)�intr! )r5 r\ s r rZ zPatternCompiler.get_int� s � ��4�:�:�� r rx )FF)r r r r7 rA r= rR rZ r r r r' r' & s � �
K�
+�E"�N!�F r r' )ro rk �NUMBER�TOKENc � � | d j � rt j S | t j v rt j | S y )Nr )�isalphar ro r �opmap)r! s r rn rn � s: � � �Q�x�����z�z�� �'�-�-� ��}�}�U�#�#�r c � � |\ }}}}|s|| j v rt j |||�� S t j |||�� S )z9Converts raw node information to a Node or Leaf instance.)�context)�
number2symbolr
�Node�Leaf)r �
raw_node_infor r! r� rK s r r4 r4 � sF � �%2�"�D�%��(��4�7�0�0�0��{�{�4��7�;�;��{�{�4���8�8r c �4 � t � j | � S rx )r' rA )rb s r rA rA � s � � � � ,� ,�W� 5�5r )�__doc__�
__author__r �pgen2r r r r r r � r
r � Exceptionr
r% �objectr' ro rk rz rq rn r4 rA r r r �r� sw ��� 3�
�
� E� D� � � �� � �I �f� I �Z �Z�Z� �|�|� �|�|��
� � �9�6r