a dw%@szUddlmZmZmZmZddlmZmZmZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZmZmZddlmZdd lmZdd lmZmZmZmZd d Z d dZ!iZ"de#d<Gddde Z$ddddZ%ddZ&GdddZ'ddZ(ddZ)d3dd Z*e*e"d!<Gd"d#d#Z+d4d$d%Z,d&d'Z-d(d)Z.d*d+Z/Gd,d-d-Z0e/e"d.<e0e"d/<eed0d1d2Z1dS)5)AnyCallableDictTuple)ConfigurationError GrammarError assert_config)get_regexp_width Serialize)GrammarAnalyzer) LexerThread BasicLexerContextualLexerLexer)earleyxearleycyk) LALR_Parser)Tree) LexerConf ParserConf_ParserArgType _LexerArgTypecs0tdd}|rSGfdddt}|SdS)NZ__future_interface__Fcs eZdZfddZddZdS)z'_wrap_lexer..CustomLexerWrappercs||_dSN)lexer)self lexer_conf lexer_class^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lark/parser_frontends.py__init__sz0_wrap_lexer..CustomLexerWrapper.__init__cSs|j|jSr)rlextext)r lexer_state parser_stater r r!r#sz+_wrap_lexer..CustomLexerWrapper.lexN__name__ __module__ __qualname__r"r#r rr r!CustomLexerWrappers r+)getattrr)rZfuture_interfacer+r rr! _wrap_lexers  r-cCsPt|d|}|r |jdp"t}||d|||j}||_t||||dS)N parser_confrparser)r/)rZ deserialize_pluginsgetrdebug callbacksParsingFrontend)datamemorr3optionsr.clsr/r r r!_deserialize_parsing_frontends r9z/Dict[str, Callable[[LexerConf, Any, Any], Any]]_parser_creatorsc@s@eZdZdZd ddZdddZddZdd d Zdd d ZdS)r4)rr.r/NcCs||_||_||_|r||_n2t|j}|dusBJd|j|||||_|j}d|_ |dvr||j dusrJd|_ dSzt t d|}Wn2t yt|tsJ|t|||_Yn0|||j|j ||_|j rt|j|j |_dS)Nz&{} is not supported in standalone modeF)dynamicdynamic_completeTbasicZ contextual)r.rr7r/r:r1 parser_typeformat lexer_type skip_lexerpostlexcreate_basic_lexercreate_contextual_lexerKeyError issubclassrr-rPostLexConnector)rrr.r7r/ create_parserrAZ create_lexerr r r!r")s8  zParsingFrontend.__init__cCsR|dur.|jj}t|dkr&td||\}n ||jjvrNtd||jjf|S)Nrz]Lark initialized with more than 1 possible start rule. Must specify which start rule to parsez(Unknown start rule %s. Must be one of %r)r.startlenr)rrJZ start_declsr r r! _verify_startNs   zParsingFrontend._verify_startcCs0|jr|jjdpt}|jr"|S||j|S)Nr )r7r0r1r rBZ from_textr)rr$r8r r r!_make_lexer_threadXsz"ParsingFrontend._make_lexer_threadcCs>||}|durind|i}||}|jj||fi|S)Non_error)rLrMr/parse)rr$rJrN chosen_startkwstreamr r r!rO\s  zParsingFrontend.parsecCs6||}|jjdkrtd||}|j||S)Nlalrzr;r<)r>z9Parser %r does not support lexer %%r, expected one of %%s)r isinstancetype)r/rexpectedr r r!_validate_frontend_argsjs  rYcCs2i}|D]$}t||jd}|dur|||j<q|Sr)r,name)Z transformer terminalsresultZterminalcallbackr r r!_get_lexer_callbacksus  r^c@seZdZddZddZdS)rHcCs||_||_dSr)r postlexer)rrr_r r r!r"~szPostLexConnector.__init__cCs|j||}|j|Sr)rr#r_process)rr%r&ir r r!r#szPostLexConnector.lexNr'r r r r!rH}srHcCs|r|jdpt}||S)Nr)r0r1r)rr/rCr7r8r r r!rDsrDcCsF|r|jdpt}dd|jjD}|r4|jnd}||||dS)NrcSsi|]\}}|t|qSr )listkeys).0idxtr r r! z+create_contextual_lexer..r ) always_accept)r0r1rZ _parse_tablestatesitemsri)rr/rCr7r8rjrir r r!rEsrEcCs.|r |jnd}|r|jdp t}|||dS)NFr)r2)r2r0r1r)rr.r7r2r8r r r!create_lalr_parsersrlrSc@seZdZddZdddZdS)EarleyRegexpMatcherc Csi|_|jD]~}|j}zt|d}Wn$tyNtd|j|fYn0|dkrbtd||jrr| d}|j ||j |j|j<q dS)NrzBad regexp in token %s: %sz/Dynamic Earley doesn't allow zero-width regexpszutf-8) regexpsr[patternZ to_regexpr ValueErrorrrZ use_bytesencodeZ re_modulecompileZ g_regex_flags)rrrfregexpwidthr r r!r"s     zEarleyRegexpMatcher.__init__rcCs|j|j||Sr)rnrZmatch)rtermr$indexr r r!rvszEarleyRegexpMatcher.matchN)r)r(r)r*r"rvr r r r!rmsrmcKs.|jrtdt|}tj|||jfi|S)Nz7Earley's dynamic lexer doesn't support lexer_callbacks.)r3rrmrParserrv)rr.r7rQZearley_matcherr r r!create_earley_parser__dynamicsrzcCs |j|jkSr)rZrW)rwtokenr r r!_match_earley_basicsr|cKstj||tfi|Sr)rryr|)rr.r7rQr r r!create_earley_parser__basicsr}cCs||jdk}|r|jnd}|jdkr,|jp.tnd}i}|jdkrDt}n|jdkr\d|d<t}nt}||||f|||d|S) NresolveFZforestr;r<TZ complete_lex)resolve_ambiguityr2 tree_class)Z ambiguityr2rrrArzr})rr.r7rr2rextrafr r r!create_earley_parsers   rc@s.eZdZd ddZddZddZdd ZdS) CYK_FrontEndNcCs$t||_t|j|_|j|_dSr)r Z _analysisrryrulesr/r3)rrr.r7r r r!r"s zCYK_FrontEnd.__init__cCs&t|d}|j||}||Sr)rbr#r/rO _transform)rZ lexer_threadrJtokenstreer r r!rOszCYK_FrontEnd.parsecs6t|}|D]}fdd|jD|_q|S)Ncs$g|]}t|tr|n|qSr )rVr_apply_callback)rdcrr r! rhz+CYK_FrontEnd._transform..)rbZ iter_subtreeschildrenr)rrZsubtreesZsubtreer rr!rs zCYK_FrontEnd._transformcCs|j|j|jSr)r3Zruler)rrr r r!rszCYK_FrontEnd._apply_callback)N)r(r)r*r"rOrrr r r r!rs rrr)r?rAcCs4t|tsJt|tsJ||_||_t|||Sr)rVrrr?rAr4)r?rArr.r7r r r!_construct_parsing_frontends r)N)N)2typingrrrr exceptionsrrr utilsr r Zparsers.grammar_analysisr rr rrrparsersrrrZparsers.lalr_parserrrrcommonrrrrr-r9r:__annotations__r4rYr^rHrDrErlrmrzr|r}rrrr r r r!s<     D