֦i& d Z ddlmZ ddlmZmZ ddlmZmZ eZ eZ
ej Z eZ
dZdZdZ G d d e Zdd Zd Zd
Zy
)z0Utility functions used by the btm_matcher module )pytree)grammartoken)pattern_symbolspython_symbolsc 0 e Zd ZdZddZd Zd Zd Zd Zy) MinNodezThis class serves as an intermediate representation of the
pattern tree during the conversion to sets of leaf-to-root
subpatternsNc f || _ || _ g | _ d| _ d | _ g | _ g | _ y )NF)typenamechildrenleafparent alternativesgroup)selfr r s */usr/lib64/python3.12/lib2to3/btm_utils.py__init__zMinNode.__init__ s4
c ^ t | j dz t | j z S )N )strr r )r s r __repr__zMinNode.__repr__ s" 499~#c$))n44r c D | }g }|r|j t k( r|j j | t |j t |j
k( r*t
|j g}g |_ |j }|j }d} |S |j t k( r|j j | t |j t |j
k( r*t |j }g |_ |j }|j }d} |S |j t j k( r(|j r |j |j n|j |j |j }|r|S )zInternal method. Returns a characteristic path of the
pattern tree. This method must be run for all leaves until the
linear subpatterns are merged into a singleN)r TYPE_ALTERNATIVESr appendlenr tupler
TYPE_GROUPr get_characteristic_subpattern token_labelsNAMEr )r nodesubps r leaf_to_rootzMinNode.leaf_to_root! sP yy --!!((.t(()S -?? !$"3"3 45D(*D% ;;D ;;D D, ) yyJ&
!!$'tzz?c$--&888DD!#DJ ;;D ;;D D yyL - --$)) DII& DII&;;DC D r c Z | j D ] }|j }|s|c S y)a Drives the leaf_to_root method. The reason that
leaf_to_root must be run multiple times is because we need to
reject 'group' matches; for example the alternative form
(a | b c) creates a group [b c] that needs to be matched. Since
matching multiple linear patterns overcomes the automaton's
capabilities, leaf_to_root merges each group into a single
choice based on 'characteristic'ity,
i.e. (a|b c) -> (a|b) if b more characteristic than c
Returns: The most 'characteristic'(as defined by
get_characteristic_subpattern) path for the compiled pattern
tree.
N)leavesr( )r lr' s r get_linear_subpatternz MinNode.get_linear_subpatternK s( A>>#D r c # K | j D ] }|j E d{ | j s| yy7 w)z-Generator that returns the leaves of the treeN)r r* )r childs r r* zMinNode.leaves` s9 ]]E ||~ % % #}}J
&s #A >A )NN) __name__
__module__ __qualname____doc__r r r( r, r* r r r r s! 5(T *r r Nc
d}| j t j k( r| j d } | j t j k( rt | j dk rt
| j d | }nt t }| j D ]K }| j j | dz r"t
|| }|1|j j | M na| j t j k( rt | j dkD r\t t }| j D ], }t
|| }|s|j j | . |j sd}nt
| j d | }n| j t j k( rt | j d t j r5| j d j" dk( rt
| j d | S t | j d t j r | j d j" dk( sMt | j dkD r6t% | j d d r | j d j" dk( ryd }d}d}d
}d} d
}
| j D ] }|j t j& k( rd
}|}nA|j t j( k( rd }|} n |j t j k( r|}t% |d ss|j" d k( sd }
|
r:| j d } t% | d r.| j" dk( r | j d } n| j d } | j t* j, k( r| j" d
k( rt t. }nt% t* | j" r%t t1 t* | j" }nt t1 t2 | j" }n| j t* j4 k( rS| j" j7 d } | t8 v rt t8 | }nEt t* j, | }n)| j t j k( r t
|| }|rB| j d j" dk( rd}n#| j d j" dk( rnt: |r@|>|j dd D ], }t
|| }||j j | . |r||_ |S )z
Internal function. Reduces a compiled pattern tree to an
intermediate representation suitable for feeding the
automaton. This also trims off any optional pattern elements(like
[a], a*).
N )r r ([valueTF= any')r r *+r ) r symsMatcherr Alternativesr reduce_treer r indexr Alternativer" Unit
isinstancer Leafr9 hasattrDetailsRepeaterr$ r% TYPE_ANYgetattrpysymsSTRINGstriptokensNotImplementedErrorr )
r& r new_noder. reducedr details_nodealternatives_node has_repeater
repeater_nodehas_variable_name name_leafr s
r rC rC g sI H yyDLL }}Q yyD%%% t}} ""4==#3V>\ . . .%'"1< 9 &GL)//,R SH &GFIOO,L MH
^^| 2 2
2 ??((-Dv~"t 5" (9(9E
^^t 0 0
0"#4f=H %%a(..#5 ''*00C7 *) H 0%..q4%eX6& %%,,W5 5
Or c t | t s| S t | dk( r| d S g }g }g dg }d| D ]~ }t t |d st t |fd r|j | Dt t |fd r|j | n|j | |r|} n |r|} n|r|} t
| t S ) zPicks the most characteristic from a list of linear patterns
Current order used is:
names > common_names > common_chars
r r5 )inforifnotNonez[]().,:c $ t | t u S N)r r )xs r z/get_characteristic_subpattern.. s d1gnr c 0 t | t xr | v S rb rG r )rc common_charss r rd z/get_characteristic_subpattern.. s jC&8&NQ,=N&Nr c 0 t | t xr | v S rb rf )rc common_namess r rd z/get_characteristic_subpattern.. s
1c(:(PqL?P(Pr )key)rG listr r< rec_testr max) subpatternssubpatterns_with_names subpatterns_with_common_names subpatterns_with_common_chars
subpatternrg ri s @@r r# r# s
k4 (
;1~ $&!6L$&! L!
x
$<= >8J NP Q-44Z@Xj PR S-44Z@ '--j9 " , &3 &3 { $$r c # K | D ]7 }t |t t f rt || E d{ . || 9 y7 w)zPTests test_func on all items of sequence and items of included
sub-iterablesN)rG rk r! rl )sequence test_funcrc s r rl rl s= a$ ' 9- - -A, -s +AA Arb )r2 r pgen2r r pygramr r r@ rN opmaprQ r$ rL r r" objectr rC r# rl r3 r r r{ sZ 2 ! 3
Uf UnBJ#%J r