U .en@sdZddlmZddlZdddddd gZGd d d ZGd d d eZGdddeZGdddeZGdddeZ GdddeZ GdddeZ dddddZ dd ddd Z dS)a Parser for parsing a regular expression. Take a string representing a regular expression and return the root node of its parse tree. usage:: root_node = parse_regex('(hello|world)') Remarks: - The regex parser processes multiline, it ignores all whitespace and supports multiple named groups with the same name and #-style comments. Limitations: - Lookahead is not supported. ) annotationsNRepeatVariableRegex Lookaheadtokenize_regex parse_regexc@s0eZdZdZdddddZdddddZd S) NodezT Base class for all the grammar nodes. (You don't initialize this one.) NodeSequence other_nodereturncCs t||gSN)r selfr rY/tmp/pip-unpacked-wheel-_x9ru1vw/prompt_toolkit/contrib/regular_languages/regex_parser.py__add__%sz Node.__add__AnyNodecCs t||gSr)rrrrr__or__(sz Node.__or__N)__name__ __module__ __qualname____doc__rrrrrrr sr c@s>eZdZdZdddddZdddd d Zd d d dZdS)rz Union operation (OR operation) between several grammars. You don't initialize this yourself, but it's a result of a "Grammar1 | Grammar2" operation. list[Node]Nonechildrenr cCs ||_dSrrrrrrr__init__3szAnyNode.__init__r r cCst|j|gSr)rrrrrrr6szAnyNode.__or__strr cCs|jjd|jdSN() __class__rrrrrr__repr__9szAnyNode.__repr__N)rrrrr rr)rrrrr,src@s>eZdZdZdddddZdddd d Zd d d dZdS)r z Concatenation operation of several grammars. You don't initialize this yourself, but it's a result of a "Grammar1 + Grammar2" operation. rrrcCs ||_dSrrrrrrr CszNodeSequence.__init__r r cCst|j|gSr)r rrrrrrFszNodeSequence.__add__r!r"cCs|jjd|jdSr#r&r(rrrr)IszNodeSequence.__repr__N)rrrrr rr)rrrrr =sr c@s.eZdZdZdddddZdddd Zd S) rz Regular expression. r!r)regexr cCst|||_dSr)recompiler*)rr*rrrr Rs zRegex.__init__r"cCs|jjd|jdS)Nz(/z/))r'rr*r(rrrr)WszRegex.__repr__Nrrrrr r)rrrrrMsc@s2eZdZdZdddddddZd d d d Zd S)rz Lookahead expression. Fr boolr) childnodenegativer cCs||_||_dSr)r/r0)rr/r0rrrr `szLookahead.__init__r!r"cCs|jjd|jdSr#r'rr/r(rrrr)dszLookahead.__repr__N)Fr-rrrrr[sc@s2eZdZdZd ddddddZdd d d Zd S)ra Mark a variable in the regular grammar. This will be translated into a named group. Each variable can have his own completer, validator, etc.. :param childnode: The grammar which is wrapped inside this variable. :param varname: String. r r!r)r/varnamer cCs||_||_dSr)r/r3)rr/r3rrrr qszVariable.__init__r"cCsd|jj|j|jS)Nz {}(childnode={!r}, varname={!r}))formatr'rr/r3r(rrrr)us zVariable.__repr__N)r2r-rrrrrhsc@s2eZdZddddddd d d Zd d ddZdS)rrNTr intz int | Noner.r)r/ min_repeat max_repeatgreedyr cCs||_||_||_||_dSr)r/r6r7r8)rr/r6r7r8rrrr ~szRepeat.__init__r!r"cCs|jjd|jdS)Nz (childnode=r%r1r(rrrr)szRepeat.__repr__)rNT)rrrr r)rrrrr}s  r!z list[str])inputr cCshtdtj}g}|rd||}|rZ|d|||d}}|sb||qtdq|S)z Takes a string, representing a regular expression as input, and tokenizes it. :param input: string, representing a regular expression. :returns: List of tokens. a^( \(\?P\<[a-zA-Z0-9_-]+\> | # Start of named group. \(\?#[^)]*\) | # Comment \(\?= | # Start of lookahead assertion \(\?! | # Start of negative lookahead assertion \(\?<= | # If preceded by. \(\?< | # If not preceded by. \(?: | # Start of group. (non capturing.) \( | # Start of group. \(?[iLmsux] | # Flags. \(?P=[a-zA-Z]+\) | # Back reference to named group \) | # End of group. \{[^{}]*\} | # Repetition \*\? | \+\? | \?\?\ | # Non greedy repetition. \* | \+ | \? | # Repetition \#.*\n | # Comment \\. | # Character group. \[ ( [^\]\\] | \\.)* \] | [^(){}] | . )NzCould not tokenize input regex.)r+r,VERBOSEmatchendisspaceappend Exception)r9ptokensmtokenrrrrs  "  ) regex_tokensr cs^dg|dddddddddd fd d }td krVtd n|SdS)zN Takes a list of tokens from the tokenizer, and returns a parse tree. r%Nrr )lstr cSs t|dkr|dSt|SdS)z7Turn list into sequence when it contains several items.rN)lenr )rFrrrwraps zparse_regex..wrapr"csggddfdd }r̈}|drXt|ddd}|q|d kr~|d k}td|d d<q|d kr|d k}tdd|dd<q|dkrgkrtdtn |dk}tddd|dd<q|dkrgq|dkrq|dkr<tddq|dkr\tddq|dkrl|S|drzq|drt|dq|d rtd!|q|rqt |qtd"dS)#Nr r"cs4gkrStfddDSdS)Ncsg|] }|qSrr).0i)rIrr szGparse_regex.._parse..wrapped_result..)r>rr)or_listresultrIrrwrapped_results z3parse_regex.._parse..wrapped_resultz(?P<rE)r3)*z*?rQ)r8)+z+?rRrG)r6r8)?z??zNothing to repeat.rSr)r6r7r8|)r$z(?:z(?!T)r0z(?=Fr%#{z#-style repetition not yet supportedz(?z%r not supportedzExpecting ')' token) pop startswithrr>rr?reprrr=r)rOtvariabler8_parserArI)rMrNrr]sZ             zparse_regex.._parserzUnmatched parentheses.)rHr?)rDrNrr\rrsD  )r __future__rr+__all__r rr rrrrrrrrrrs$    5