a DOg@sdZdZddlZddlmZmZmZmZmZm Z ddl m Z ddl m Z Gdd d e Zd d ZGd d d eZejejejddZddZddZddZdS)zPattern compiler. The grammar is taken from PatternGrammar.txt. The compiler compiles a pattern to a pytree.*Pattern instance. z#Guido van Rossum N)driverliteralstokentokenizeparsegrammar)pytree)pygramc@s eZdZdS)PatternSyntaxErrorN)__name__ __module__ __qualname__rr4/opt/alt/python39/lib64/python3.9/lib2to3/patcomp.pyr sr c csLtjtjtjh}tt|j}|D] }|\}}}}}||vr&|Vq&dS)z6Tokenizes a string suppressing significant whitespace.N) rNEWLINEINDENTDEDENTrgenerate_tokensioStringIOreadline) inputskiptokensZ quintupletypevaluestartendZ line_textrrrtokenize_wrappers rc@s:eZdZd ddZdddZddZdd d Zd d ZdS)PatternCompilerNcCsZ|durtj|_tj|_nt||_t|j|_tj|_ tj |_ tj |jt d|_dS)z^Initializer. Takes an optional alternative filename for the pattern grammar. N)Zconvert)r Zpattern_grammarrZpattern_symbolssymsrZ load_grammarZSymbolsZpython_grammarZ pygrammarZpython_symbolspysymsZDriverpattern_convert)selfZ grammar_filerrr__init__(s  zPatternCompiler.__init__Fc Csrt|}z|jj||d}Wn4tjyP}ztt|dWYd}~n d}~00|rd|||fS||SdS)z=Compiles a pattern string to a nested pytree.*Pattern object.)debugN)rrZ parse_tokensrZ ParseErrorr str compile_node)r$rr&Z with_treerrooterrrcompile_pattern7s$zPatternCompiler.compile_patternc sV|jjjkr|jd}|jjjkrzfdd|jdddD}t|dkrX|dStjdd|Dddd}|S|jjj krʇfd d|jD}t|dkr|dStj|gddd}|S|jjj kr |jdd}t |}|Sd}|j}t|d kr>|djt jkr>|dj}|dd}d}t|dkrx|d jjjkrx|d }|dd } ||}|dur>|j} | d} | jt jkrd} tj} nX| jt jkrd} tj} n>| jt jkr| d} } t| d kr| d } n| dks"| dkr>|}tj|gg| | d}|durN||_|S) zXCompiles a node, recursively. This is one big switch on the node type. rcsg|]}|qSrr(.0Zchr$rr Oz0PatternCompiler.compile_node..NrcSsg|] }|gqSrr)r.arrrr0Rr1minmaxcsg|]}|qSrr,r-r/rrr0Vr1)rr!ZMatcherchildrenZ Alternativeslenr WildcardPatternoptimizeZ AlternativeZ NegatedUnit compile_basicZNegatedPatternrEQUALrZRepeaterSTARZHUGEPLUSLBRACEget_intname) r$nodeZaltspZunitspatternrDnodesrepeatr:childr5r6rr/rr(Cs^       "    zPatternCompiler.compile_nodecCs@|d}|jtjkr4tt|j}tt ||S|jtj kr|j}| r|t vrbt d||ddrvt dtt |S|dkrd}n,|dst|j|d}|durt d||ddr||djdg}nd}t||SnH|jdkr||dS|jd kr<||d}tj|ggddd SdS) NrzInvalid token: %rrzCan't have details for tokenany_zInvalid symbol: %r([r4)rrSTRINGr'rZ evalStringrr Z LeafPattern_type_of_literalNAMEisupper TOKEN_MAPr startswithgetattrr"r(r:Z NodePatternr<)r$rHrIrErrZcontent subpatternrrrr>s8         zPatternCompiler.compile_basiccCs t|jSN)intr)r$rErrrrCszPatternCompiler.get_int)N)FF)N)r r rr%r+r(r>rCrrrrr &s   G #r )rQrONUMBERZTOKENcCs.|drtjS|tjvr&tj|SdSdS)Nr)isalpharrQrZopmap)rrrrrPs    rPcCs>|\}}}}|s||jvr*tj|||dStj|||dSdS)z9Converts raw node information to a Node or Leaf instance.)contextN)Z number2symbolr ZNodeZLeaf)rZ raw_node_inforrr[r:rrrr#s r#cCs t|SrW)r r+)rGrrrr+sr+)__doc__ __author__rZpgen2rrrrrrr r Exceptionr robjectr rQrOrYrSrPr#r+rrrrs"