o bi@sdZddlmZmZmZmZmZmZddlm Z m Z ddl m Z ee e e fZee e e fZee e fZdZGdddZGdd d e e e e fZGd d d Zd ed edefddZGdddZde dee fddZdS)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@seZdZdZdddZdeeeefdeefddZ d e deefd d Z d eedd fd dZ d e de deefddZdS) TemplateConfz~Template Configuration Allows customization for different uses of Template parse() must return a Tree instance. NcCs ||_dSN)_parse)selfparserG/home/ubuntu/.local/lib/python3.10/site-packages/lark/tree_templates.py__init__s zTemplateConf.__init__varreturncCsVt|tr t|St|tr)|jdkr)t|jdkr)t|jdtr)t|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_namerdatalenchildren)rrrrrtest_vars  zTemplateConf.test_vartemplatecCs4t|tr|js J||}t|tstd|S)Nz+template parser must return a Tree instance)rrrr TypeErrorrrrrr _get_tree3s    zTemplateConf._get_treeTemplatecCs t||dS)N)conf)r!rrrr__call__=s zTemplateConf.__call__treecCs||}|rt|tstd|||iSt|tr$||kr"iSdSt|tr.t|ts8Jd|d||j|jkrjt|jt|jkrji}t|j|jD]\}}| ||}|durbdS| |qQ|SdS)zAReturns dict of {var: match} if found a match, else None z6Template variables can only match Tree instances. Not Nz template=z tree=) rrrrrrrrzip_match_tree_templateupdate)rrr$ template_varrest1t2matchesrrrr&@s&   (   z!TemplateConf._match_tree_templater )__name__ __module__ __qualname____doc__rrrrrr TreeOrCoder r#Branch MatchResultr&rrrrr s " r csLeZdZdedeeeefddffdd Zdeeffdd ZZ S) _ReplaceVarsr"varsrNcst||_||_dSr )superr_conf_vars)rr"r5 __class__rrr_s  z_ReplaceVars.__init__csPt|||}|j|}|r&z|j|WSty%td|dw|S)Nz"No mapping for template variable ())r6 __default__r7rr8KeyErrorr )rrrmetar$rr9rrr<ds   z_ReplaceVars.__default__) r-r.r/r rrrrr< __classcell__rrr9rr4^s&r4c@seZdZdZefdeedefddZdede e fddZ dede e eee ffd d Zd eeeefdeefd d ZdS)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"cCs||_|||_dSr )r"r r$)rr$r"rrrrxszTemplate.__init__rcCs|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$rrrmatch|s zTemplate.matchccs8|j|}|D]}||}|r||fVq dS)zISearch for all occurrences of the tree template inside ``tree``. N)r"r iter_subtreesr@)rr$subtreer)rrrsearchs    zTemplate.searchr5cCst|j||jS)z(Apply vars to the template tree )r4r" transformr$)rr5rrr apply_varsszTemplate.apply_varsN)r-r.r/r0r rrrr1rr3r@rrrCrrErrrrr!ps "& r!r*r+r$cCs>|j|}||D]\}}||}||j|jq |S)z=Search tree and translate each occurrence of t1 into t2. )r"r rCrEsetrr)r*r+r$rBr5r)rrr translates  rGc@s8eZdZdZdeeeffddZdeefddZ dS) TemplateTranslatorz;Utility class for translating a collection of patterns translationscCs$tdd|Ds J||_dS)Ncss(|]\}}t|tot|tVqdSr )rr!).0kvrrr s&z.TemplateTranslator.__init__..)allitemsrI)rrIrrrrs zTemplateTranslator.__init__r$cCs$|jD] \}}t|||}q|Sr )rIrOrG)rr$rKrLrrrrGszTemplateTranslator.translateN) r-r.r/r0rr!rrrrGrrrrrHsrHvaluercCs|tr |tSdSr ) startswith_TEMPLATE_MARKERlstrip)rPrrrrsrN)r0typingrrrrrrlarkrr lark.exceptionsr rr2r1r3rRr r4r!rGrHrrrrrs   L+