�
!A?h� � �� � 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� ZdS )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 ZdS )�PatternSyntaxErrorN)�__name__�
__module__� __qualname__� � �z0PatternCompiler.compile_node..O s' �� �G�G�G�b�D�%�%�b�)�)�G�G�Gr N� r c � � g | ]}|g��S r r )rF �as r rH z0PatternCompiler.compile_node..R s � �':�':�':����':�':�':r ��min�maxc �: �� g | ]}�� |� � ��S r rD rE s �r rH z0PatternCompiler.compile_node..V s' �� �C�C�C�r�T�&�&�r�*�*�C�C�Cr � ������ )r r, �Matcher�children� Alternatives�lenr
�WildcardPattern�optimize� Alternative� NegatedUnit�
compile_basic�NegatedPatternr �EQUALr! �Repeater�STAR�HUGE�PLUS�LBRACE�get_int�name)
r5 �node�alts�p�units�patternrd �nodes�repeatrT �childrM rN s
` r r= z PatternCompiler.compile_nodeC s� �� � �9�� �)� )� )��=��#�D� �9�� �.� .� .�G�G�G�G�D�M�#�#�A�#�4F�G�G�G�D��4�y�y�A�~�~��A�w���&�':�':�T�':�':�':��q�I�I�I�A��:�:�<�<� � �9�� �-� -� -�C�C�C�C�T�]�C�C�C�E��5�z�z�Q��� �Q�x���&��w�A�1�=�=�=�A��:�:�<�<� � �9�� �-� -� -��(�(�� �q�r�r�):�;�;�G��%�g�.�.�A��:�:�<�<� � ���
�� �u�:�:��?�?�u�Q�x�}�� � ;� ;���8�>�D��!�"�"�I�E��� �u�:�:��?�?�u�R�y�~���1C� C� C��2�Y�F��#�2�#�J�E� �$�$�U�F�3�3�� � � ��H� �Q�K�E��z�U�Z�'�'��� �k�����u�z�)�)��� �k�����u�|�+�+� !�L�L��!�� 5� 5�5��c��x�=�=�A�%�%� �,�,�x��{�3�3�C� �a�x�x�3�!�8�8�!�*�*�,�,�� �0�7�)��#�3�O�O�O�� � � �G�L�� � �!�!�!r c �� � |d }|j t j k rHt t j |j � � � � }t j t |� � |� � S |j t j
k r�|j }|� � � rS|t vrt d|z � � �|dd � rt d� � �t j t | � � S |dk rd }n?|� d� � s*t | j |d � � }|�t d|z � � �|dd � r(| � |d j d � � g}nd }t j ||� � S |j dk r| � |d � � S |j d k r4| � |d � � }t j |ggdd�
� � S d S ) Nr zInvalid token: %rr z Can't have details for token�any�_zInvalid symbol: %r�(�[rL )r r �STRINGr< r �
evalStringr! r
� LeafPattern�_type_of_literal�NAME�isupper� TOKEN_MAPr
�
startswith�getattrr2 r= rT � NodePatternrW )r5 rj rk re r! r �content�
subpatterns r r[ z PatternCompiler.compile_basic� s� � � �Q�x�� �9�� � $� $���+�D�J�7�7�8�8�E��%�&6�u�&=�&=�u�E�E� E�
�Y�%�*�
$�
$��J�E��}�}���
9�� �)�)�,�-@�5�-H�I�I�I�����9� M�,�-K�L�L�L� �)�)�E�*:�;�;�;��E�>�>� �D�D� �)�)�#�.�.� O�"�4�;��t�<�<�D��|� 0�1E��1M� N� N�N�����9� #� #� 0� 0��q��1B�1�1E� F� F� G�G�G� "�G� �)�$��8�8�8�
�Z�3�
�
��$�$�U�1�X�.�.� .�
�Z�3�
�
� �*�*�5��8�4�4�J��)�J�<�.�a�Q�G�G�G� G� �
r c �* � t |j � � S �N)�intr! )r5 re s r rc zPatternCompiler.get_int� s � ��4�:��� r r )FF)r r r r7 rA r= r[ rc r r r r' r' & sw � � � � � �
K�
K�
K�
K�
+�
+�
+�
+�E"� E"� E"�N!� !� !� !�F � � � � r r' )rv rr �NUMBER�TOKENc � � | d � � � r t j S | t j v rt j | S d S )Nr )�isalphar rv r �opmap)r! s r ru ru � sA � � �Q�x����� ��z�� �'�-� � ��}�U�#�#��tr 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� rT s r r4 r4 � sS � �%2�"�D�%��(�� 9�4�7�0�0�0��{�4��7�;�;�;�;��{�4���8�8�8�8r c �D � t � � � | � � S r )r' rA )ri s r rA rA � s � � � � � ,� ,�W� 5� 5�5r )�__doc__�
__author__r �pgen2r r r r r r � r
r � Exceptionr
r% �objectr' rv rr r� rx ru r4 rA r r r �r� s= ��� � 3�
�
� � � � E� D� D� D� D� D� D� D� D� D� D� D� D� D� D� D� � � � � � � � � � � � � � � � � �� � � � � � �I � I � I � I � I �f� I � I � I �Z �Z� �|� �|��
�
� � � � �9� 9� 9�6� 6� 6� 6� 6r