`iTdZddlmZmZmZmZmZmZddlm Z m Z ddl m Z ee e e fZee e e fZee e fZdZGddZGdd e e e e fZGd d Zd ed edefdZGddZde dee fdZdS)zThis module defines utilities for matching and translation tree templates. A tree templates is a tree that contains nodes that are template variables. )UnionOptionalMappingDictTupleIterator)Tree Transformer)MissingVariableError$ceZdZdZd dZdeeeefdeefdZ de deefdZ deedd fd Z de d e deefd ZdS) TemplateConfz~Template Configuration Allows customization for different uses of Template parse() must return a Tree instance. Nc||_dSN)_parse)selfparses g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/lark/tree_templates.py__init__zTemplateConf.__init__s  varreturnc2t|trt|St|tr]|jdkrRt |jdkr:t|jdtrt|jdSdS)zGiven a tree node, if it is a template variable return its name. Otherwise, return None. This method may be overridden for customization Parameters: var: Tree | str - The tree node to test rrN) isinstancestr_get_template_namer datalenchildren)rrs rtest_varzTemplateConf.test_vars c3   +%c** * sD ! ! 7E!!CL!!A%%3<?C00&&cl1o66 6trtemplatect|tr|jsJ||}t|tst d|S)Nz+template parser must return a Tree instance)rrrr TypeErrorrr!s r _get_treezTemplateConf._get_tree3sY h $ $ -;  ;{{8,,H(D)) KIJJ JrTemplatec$t||S)N)conf)r&r$s r__call__zTemplateConf.__call__=st,,,,rtreecX||}|r+t|tstd|||iSt|tr ||kriSdSt|trt|tsJd|d||j|jkr~t |jt |jkrTi}t|j|jD]5\}}| ||}|dS| |6|SdS)zAReturns dict of {var: match} if found a match, else None z6Template variables can only match Tree instances. Not Nz template=z tree=) r rr r#rrrrzip_match_tree_templateupdate)rr!r* template_varrest1t2matchess rr-z!TemplateConf._match_tree_template@sO}}X..  (dD)) c aY] a abbb $' ' h $ $ 4 4(D))hjt.D.DhhFhRZFhFhbfFhFhhhD =DI % %#h.?*@*@C DVDV*V*VCh/?? $ $B33B;;?44 7####Jtrr)__name__ __module__ __qualname____doc__rrr rrr TreeOrCoder%r)Branch MatchResultr-rrrrsE$s)S.1hsm,*c-c-z----Zv(S^J_rrcdeZdZdedeeeefddffd Zdeeffd ZxZ S) _ReplaceVarsr(varsrNcdt||_||_dSr)superr_conf_vars)rr(r> __class__s rrz_ReplaceVars.__init___s+   rct|||}|j|}|r/ |j|S#t $rt d|dwxYw|S)Nz"No mapping for template variable ())r@ __default__rAr rBKeyErrorr )rrrmetar*rrCs rrFz_ReplaceVars.__default__dsww""4488j!!$''  X Xz#& X X X*+VPS+V+V+VWWW X s  AA-) r4r5r6rrrr rrF __classcell__)rCs@rr=r=^s\d3i1HT 49          rr=ceZdZdZefdeedefdZdede e fdZ dede e eee ffdZdeeeefdeefd Zd S) r&aRepresents a tree template, tied to a specific configuration A tree template is a tree that contains nodes that are template variables. Those variables will match any tree. (future versions may support annotations on the variables, to allow more complex templates) r*r(cH||_|||_dSr)r(r%r*)rr*r(s rrzTemplate.__init__xs  NN4(( rrcv|j|}|j|j|S)aMatch a tree template to a tree. A tree template without variables will only match ``tree`` if it is equal to the template. Parameters: tree (Tree): The tree to match to the template Returns: Optional[Dict[str, Tree]]: If match is found, returns a dictionary mapping template variable names to their matching tree nodes. If no match was found, returns None. )r(r%r-r*)rr*s rmatchzTemplate.match|s3y""4((y--di>>>rc#K|j|}|D]}||}|r||fV dS)zISearch for all occurrences of the tree template inside ``tree``. N)r(r% iter_subtreesrM)rr*subtreer0s rsearchzTemplate.searchsly""4(())++ # #G**W%%C #sl""" # #rr>c\t|j||jS)z(Apply vars to the template tree )r=r( transformr*)rr>s r apply_varszTemplate.apply_varss&DIt,,66tyAAArN)r4r5r6r7rr rrr8rr:rMrrrQrrTr;rrr&r&ps>J\^^))T#Y)l))))?*?+)>???? #:#(5cK9O3P*Q####BwsDI~6B49BBBBBBrr&r1r2r*c|j|}||D]:\}}||}||j|j;|S)z=Search tree and translate each occurrence of t1 into t2. )r(r%rQrTsetrr)r1r2r*rPr>r0s r translaterWsg 7  T " "D4,, mmD!! CHcl++++ KrcFeZdZdZdeeeffdZdeefdZ dS)TemplateTranslatorz;Utility class for translating a collection of patterns translationscntd|DsJ||_dS)Nc3pK|]1\}}t|tot|tV2dSr)rr&).0kvs r z.TemplateTranslator.__init__..s?hh41a:a**Fz!X/F/Fhhhhhhr)allitemsrZ)rrZs rrzTemplateTranslator.__init__s@hhS_SeSeSgSghhhhhhhh(rr*cf|jD]\}}t|||}|Sr)rZrbrW)rr*r^r_s rrWzTemplateTranslator.translates<%++-- ) )DAqQ4((DD rN) r4r5r6r7rr&rr rrWr;rrrYrYsb)WXx-?%@))))d3irrYvaluercn|tr|tndSr) startswith_TEMPLATE_MARKERlstrip)rds rrrs/-2-=-=>N-O-O Y5<<( ) ) )UYYrN)r7typingrrrrrrlarkr r lark.exceptionsr rr9r8r:rgrr=r&rWrYrr;rrrls CBBBBBBBBBBBBBBB""""""""000000 tCy#~  49c> " 39o IIIIIIIIX;sDI~.$(B(B(B(B(B(B(B(BV(         ZcZhsmZZZZZZr