Yf @ s d Z d Z d d l Z d d l Z d d l m Z m Z m Z m Z m Z m
Z
d d l m Z d d l m
Z
e j j e j j e d Z Gd d
d
e Z d d Z Gd
d d e Z d e j d e j d e j d d i Z d d Z d d Z d d Z d S)zPattern compiler.
The grammar is taken from PatternGrammar.txt.
The compiler compiles a pattern to a pytree.*Pattern instance.
z#Guido van Rossum <guido@python.org> N )driverliteralstokentokenizeparsegrammar)pytree)pygramzPatternGrammar.txtc @ s e Z d Z d S)PatternSyntaxErrorN)__name__
__module__ __qualname__ r r 4/opt/alt/python35/lib64/python3.5/lib2to3/patcomp.pyr s r c c sn t j t j t j h } t j t j | j } x4 | D], } | \ } } } } } | | k r: | Vq: Wd S)z6Tokenizes a string suppressing significant whitespace.N) r NEWLINEINDENTDEDENTr generate_tokensioStringIOreadline) inputskiptokensZ quintupletypevaluestartendZ line_textr r r tokenize_wrapper! s
r c @ sX e Z d Z e d d Z d d d d Z d d Z d d d
Z d d Z d S)
PatternCompilerc C s^ t j | | _ t j | j | _ t j | _ t j | _ t j
| j d t | _ d S)z^Initializer.
Takes an optional alternative filename for the pattern grammar.
ZconvertN) r Z load_grammarr r
ZSymbolssymsZpython_grammarZ pygrammarZpython_symbolspysymsZDriverpattern_convert)selfZ grammar_filer r r __init__- s
zPatternCompiler.__init__Fc C s t | } y | j j | d | } Wn: t j k
rd } z t t | WYd d } ~ Xn X| r~ | j | | f S| j | Sd S)z=Compiles a pattern string to a nested pytree.*Pattern object.debugN)r r Z parse_tokensr Z
ParseErrorr str compile_node)r$ r r& Z with_treer rooter r r compile_pattern8 s %z PatternCompiler.compile_patternc
s6 | j j j k r" | j d } | j j j k r f d d | j d d d D } t | d k rz | d St j d d | D d d d d } | j S| j j j k r f d
d | j D } t | d k r | d St j | g d d d d } | j S| j j j
k ri j | j d d } t j | } | j Sd } | j } t | d k r| d j t
j k r| d j } | d d } d } t | d k r| d
j j j k r| d } | d d } j | | } | d k r| j } | d }
|
j t
j k r]d } t j } nu |
j t
j k rd } t j } nQ |
j t
j k r j | d } } t | d k r j | d } n | d k s| d k r| j } t j | g g d | d | } | d k r,| | _ | j S)zXCompiles a node, recursively.
This is one big switch on the node type.
r c s g | ] } j | q Sr )r( ).0ch)r$ r r
<listcomp>P s z0PatternCompiler.compile_node.<locals>.<listcomp>N r c S s g | ] } | g q Sr r )r, ar r r r. S s minmaxc s g | ] } j | q Sr )r( )r, r- )r$ r r r. W s r5 r5 )r r! ZMatcherchildrenZ Alternativeslenr WildcardPatternoptimizeZ AlternativeZ NegatedUnit
compile_basicZNegatedPatternr EQUALr ZRepeaterSTARZHUGEPLUSLBRACEget_intname)
r$ nodeZaltspZunitspatternr@ nodesrepeatr6 Zchildr1 r2 r )r$ r r( D s^
)(
(
+
! z PatternCompiler.compile_nodeNc C s | d } | j t j k rJ t t j | j } t j t | | S| j t j
k rX| j } | j r | t k r t
d | | d d r t
d t j t | S| d k r d } n@ | j d st | j | d } | d k rt
d | | d d r?| j | d j d g } n d } t j | | Sna | j d k rx| j | d S| j d k r| j | d } t j | g g d
d d d Sd S) Nr zInvalid token: %rr z Can't have details for tokenany_zInvalid symbol: %r([r1 r2 )r r STRINGr' r Z
evalStringr r Z LeafPattern_type_of_literalNAMEisupper TOKEN_MAPr
startswithgetattrr" r( r6 Z NodePatternr8 )r$ rD rE rA r r Zcontent
subpatternr r r r: s8
z PatternCompiler.compile_basicc C s
t | j S)N)intr )r$ rA r r r r? s zPatternCompiler.get_int) r r
r _PATTERN_GRAMMAR_FILEr% r+ r( r: r? r r r r r + s
G#r rL rJ NUMBERZTOKENc C s9 | d j r t j S| t j k r1 t j | Sd Sd S)Nr )isalphar rL r Zopmap)r r r r rK s
rK c C sW | \ } } } } | s' | | j k r= t j | | d | St j | | d | Sd S)z9Converts raw node information to a Node or Leaf instance.contextN)Z
number2symbolr ZNodeZLeaf)r Z
raw_node_infor r rV r6 r r r r# s r# c C s t j | S)N)r r+ )rC r r r r+ s r+ ) __doc__
__author__r osZpgen2r r r r r r r r
pathjoindirname__file__rS Exceptionr r objectr rL rJ rT rN rK r# r+ r r r r <module> s$ .