SrSSKJrJrJrJrJrJr SSKJ r J r SSK J r \\ \ \ 4r\\ \ \ 4r\\ \ 4rSr"SS5r"SS \ \ \ \ 45r"S S 5rS \S \S\4Sjr"SS5rS\ S\\ 4Sjrg)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$c\rSrSrSrSSjrS\\\\4S\ \4Sjr S\ S\\4S jr S\\SS 4S jr S\ S \S\ \4S jrSrg) TemplateConfzrTemplate Configuration Allows customization for different uses of Template parse() must return a Tree instance. NcXlgN_parse)selfparses &lib/third_party/lark/tree_templates.py__init__TemplateConf.__init__s varreturnc4[U[5(a [U5$[U[5(acURS:XaS[ UR 5S:a:[UR S[5(a[UR S5$g)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_varTemplateConf.test_varss c3  %c* * sD ! !E!CLL!A%3<<?C00%cll1o6 6rtemplatec[U[5(a$UR(deURU5n[U[5(d [ S5eU$)Nz+template parser must return a Tree instance)rrrr TypeErrorrr%s r _get_treeTemplateConf._get_tree3sG h $ $;; ;{{8,H(D))IJ JrTemplatec[XS9$)N)conf)r+r(s r__call__TemplateConf.__call__=s ,,rtreec^URU5nU(a'[U[5(d[SU<35eX20$[U[5(aX:Xa0$g[U[5(a[U[5(d SUSU35eUR UR :Xa[ UR5[ UR5:XaT0n[URUR5H,upVURXV5nUc gURU5 M. U$g)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'rr r!r"zip_match_tree_templateupdate)rr%r0 template_varrest1t2matchess rr3!TemplateConf._match_tree_template@s }}X. dD))"XY]X` abb ' ' h $ $ (D))jt.D.Dh RZQ[[abfagFhhD ==DII %#h.?.?*@C DV*VCh//?33B;? 7# @Jrrr)__name__ __module__ __qualname____firstlineno____doc__rrr rrr# TreeOrCoder)r.Branch MatchResultr3__static_attributes__rrrrs~E$s)S.1hsm,*c-c-z-Zv(S^J_rrcb^\rSrSrS\S\\\\4SS4U4SjjrS\\4U4Sjjr Sr U=r $) _ReplaceVars^r-varsrNc:>[TU]5 XlX lgr)superr_conf_vars)rr-rH __class__s rr_ReplaceVars.__init___s   rc>[TU]XU5nURRU5nU(aURU$U$![ a [ SUS35ef=f)Nz"No mapping for template variable ())rJ __default__rKr#rLKeyErrorr )rr r"metar0rrMs rrQ_ReplaceVars.__default__dspw"448jj!!$'  Xzz#&  X*-OPSuTU+VWW Xs AA )rKrL) r;r<r=r>rrrr rrQrC __classcell__)rMs@rrFrF^s?\d3i1HT 49  rrFc\rSrSrSr\"54S\\S\4SjjrS\ S\ \ 4Sjr S\ S\ \\\\ 44SjrS \\\\4S\\4S jrS rg ) r+pa Represents 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) r0r-c<X lURU5Ulgr)r-r)r0)rr0r-s rrTemplate.__init__xs NN4( rrcURRU5nURRURU5$)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)r3r0)rr0s rmatchTemplate.match|s3yy""4(yy--dii>>rc## URRU5nUR5H"nURU5nU(dMX#4v M$ g7f)zISearch for all occurrences of the tree template inside ``tree``. N)r-r) iter_subtreesr[)rr0subtreer6s rsearchTemplate.searchsIyy""4())+G**W%Csl",s AA ArHc`[URU5RUR5$)z(Apply vars to the template tree )rFr- transformr0)rrHs r apply_varsTemplate.apply_varss$DIIt,66tyyAAr)r-r0N)r;r<r=r>r?rr rrr@rrBr[rrr`rrdrCrDrrr+r+ps>J^)T#Y)l)?*?+)>? #:#(5cK9O3P*Q#BwsDI~6B49Brr+r7r8r0cURRU5nURU5H<up4URU5nUR UR UR 5 M> U$)z=Search tree and translate each occurrence of t1 into t2. )r-r)r`rdsetr r")r7r8r0r_rHr6s r translaterhsW 77  T "D4 mmD! CHHcll+) KrcD\rSrSrSrS\\\44SjrS\\ 4Sjr Sr g) TemplateTranslatorz;Utility class for translating a collection of patterns translationsc^[SUR555(deXlg)Nc3t# UH.up[U[5=(a [U[5v M0 g7fr)rr+).0kvs r .TemplateTranslator.__init__..s*hSg41:a*Fz!X/FFSgs68)allitemsrl)rrls rrTemplateTranslator.__init__s)hS_SeSeSghhhhh(rr0cdURR5Hup#[X#U5nM U$r)rlrurh)rr0rprqs rrhTemplateTranslator.translates.%%++-DAQ4(D. r)rlN) r;r<r=r>r?rr+rr rrhrCrDrrrjrjs-)WXx-?%@)d3irrjvaluercdUR[5(aUR[5$S$r) startswith_TEMPLATE_MARKERlstrip)rys rrrs)-2-=-=>N-O-O5<<( )YUYYrN)r?typingrrrrrrlarkr r lark.exceptionsr rrAr@rBr|rrFr+rhrjrrDrrrs CB"0 tCy#~  49c> " 39o IIX;sDI~.$(B(BV(   ZcZhsmZr