o i@sdZddlmZddlZddlmZmZgdZGdddZGdd d eZ Gd d d eZ Gd d d eZ GdddeZ GdddeZ GdddeZdddZdddZdS)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. ) annotationsN)ListOptional)RepeatVariableRegex Lookaheadtokenize_regex parse_regexc@s$eZdZdZd ddZd dd Zd S) NodezT Base class for all the grammar nodes. (You don't initialize this one.) other_nodereturn NodeSequencecC t||gSN)rselfr rp/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/prompt_toolkit/contrib/regular_languages/regex_parser.py__add__& z Node.__add__AnyNodecCrr)rrrrr__or__)rz Node.__or__Nr r r rr r r r)__name__ __module__ __qualname____doc__rrrrrrr s r c@.eZdZdZdddZdd d Zdd 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. children list[Node]r NonecC ||_dSrr rr rrr__init__4 zAnyNode.__init__r r cCt|j|gSr)rr rrrrr7zAnyNode.__or__strcC|jjd|jdSN() __class__rr rrrr__repr__:zAnyNode.__repr__Nr r!r r"rr r*)rrrrr&rr2rrrrr-s   rc@r)rz Concatenation operation of several grammars. You don't initialize this yourself, but it's a result of a "Grammar1 + Grammar2" operation. r r!r r"cCr#rr$r%rrrr&Dr'zNodeSequence.__init__r r cCr(r)rr rrrrrGr)zNodeSequence.__add__r*cCr+r,r/r1rrrr2Jr3zNodeSequence.__repr__Nr4rr5)rrrrr&rr2rrrrr>s   rc@s$eZdZdZd ddZd dd Zd S) rz Regular expression. regexr*r r"cCst|||_dSr)recompiler6)rr6rrrr&Ss  zRegex.__init__cCs|jjd|jdS)Nz(/z/))r0rr6r1rrrr2Xr3zRegex.__repr__N)r6r*r r"r5rrrrr&r2rrrrrNs rc@s&eZdZdZddd d Zdd d ZdS)rz Lookahead expression. F childnoder negativeboolr r"cC||_||_dSr)r:r;)rr:r;rrrr&a zLookahead.__init__r*cCr+r,r0rr:r1rrrr2er3zLookahead.__repr__N)F)r:r r;r<r r"r5r9rrrrr\s rc@s&eZdZdZddd d Zdd 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 varnamer*r r"cCr=r)r:rA)rr:rArrrr&rr>zVariable.__init__cCsd|jj|j|jS)Nz {}(childnode={!r}, varname={!r}))formatr0rr:rAr1rrrr2vs zVariable.__repr__N)r@)r:r rAr*r r"r5r9rrrrris rc@s(eZdZ   ddddZdddZdS)rrNTr:r min_repeatint max_repeat int | Nonegreedyr<r r"cCs||_||_||_||_dSr)r:rCrErG)rr:rCrErGrrrr&s zRepeat.__init__r*cCr+)Nz (childnode=r.r?r1rrrr2r3zRepeat.__repr__)rNT) r:r rCrDrErFrGr<r r"r5)rrrr&r2rrrrr~s   rinputr*r list[str]cCsjtdtj}g}|r3||}|r-|d|||d}}|s,||ntd|s |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.)r7r8VERBOSEmatchendisspaceappend Exception)rHptokensmtokenrrrr s  "  r regex_tokenscsNdg|dddddd dfd d }td kr%td |S)zN Takes a list of tokens from the tokenizer, and returns a parse tree. r.Nlstr!r r cSst|dkr |dSt|S)z7Turn list into sequence when it contains several items.r)lenr)rVrrrwraps zparse_regex..wrapcsggd#fdd }r؈}|dr)t|ddd}|n|d vr<|d k}td|d d<n|d vrP|d k}tdd|dd<n|dvrqgkr`tdt|dk}tddd|dd<ne|dkr}gnY|dvrnN|dkrtddn?|dkrtddn0|dkr|S|drn#|drt|d|d rtd!||rnt |std")$Nr r cs0gkrStfddDS)Ncsg|]}|qSrr).0i)rYrr szGparse_regex.._parse..wrapped_result..)rNrr)or_listresultrYrrwrapped_results z3parse_regex.._parse..wrapped_resultz(?P<rU)rA)*z*?ra)rG)+z+?rbrW)rCrG)?z??zNothing to repeat.rcr)rCrErG|)r-z(?:z(?!T)r;z(?=Fr.#{z#-style repetition not yet supportedz(?z%r not supportedzExpecting ')' tokenr r ) pop startswithrrNrrOreprrrMr)r_tvariablerG_parserQrY)r]r^rrnsV         7zparse_regex.._parserzUnmatched parentheses.)rVr!r r rg)rXrO)rTr^rrmrr s D r )rHr*r rI)rTrIr r )r __future__rr7typingrr__all__r rrrrrrr r rrrrs     5