Directory:~$ /proc/self/root/usr/lib64/python2.7/lib2to3/

Upload File

[Home] [System Details] [Kill Me]

�
s�fc@s�dZd
Z
ddlZddlZddlmZ
ddlmZ
ddlm	Z	
de
f
d	��YZd
e
f
d��YZia
d�ZdS(
s�
A bottom-up tree matching algorithm
implementation meant to speed
up 2to3's matching process. After the tree patterns are reduced to
their rarest linear path, a linear Aho-Corasick automaton is
created. The linear automaton traverses the linear paths from the
leaves to the root of the AST and returns a set of nodes for further
matching. This reduces significantly the number of candidate nodes.s+George
Boutsioukis
<gboutsioukis@gmail.com>i����N(
tdefaultdicti
(
tpytree(
treduce_treetBMNodec
Bs#eZ
dZej�Zd
�ZRS(s?Class
for a node of the Aho-Corasick automaton used in
matchingc

Cs1i|_g|_
ttj�
|_d
|_dS(Nt(ttransition_tabletfixerstnextRtcounttidtcontent(
tself((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyt__init__s
	
	

(t__name__t
__module__t__doc__t	itertoolsRR(((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyRs


t
BottomMatcherc
Bs;eZ
dZd
�Zd�Zd�Zd�Zd�ZRS(sgThe
main matcher class. After instantiating the patterns should
    be added using the add_fixer
methodc

CsFt�|_
t�|_|jg
|_g|_tjd
�
|_dS(NtRefactoringTool(	tsettmatchRtroottnodesRtloggingt	getLoggertlogger(
R((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyRs




	
cCsh|jj
|
�

t|
j�
}|j�}|j|d
|j�

}x|D]}|jj
|
�

qJWdS(s�Reduces
a fixer's pattern tree to a linear path and adds it
        to the matcher(a common Aho-Corasick automaton). The fixer is
        appended on the matching states and called when they are
       
reachedtstartN(RtappendRtpattern_treetget_linear_subpatterntaddR(Rtfixerttreetlineartmatch_nodest
match_node((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyt	add_fixer%s





cCs�|
s
|g
St|
d
t
�r�g}xU|
d
D]I}|j|d|�

}x+|D]#}|j|j|
d|��

qSWq1W|S|
d
|jkr�t�}||j|
d
<n|j|
d
}|
dr�|j|
dd|�

}n	|g
}|SdS(s5Recursively
adds a linear pattern to the AC automatoniRi
N(t
isinstancettupleRtextendRR(RtpatternRR"talternativet	end_nodestendt	next_node((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyR1s"




%
	


	
c	Cs�
|j}t
t�
}x�
|
D]y
}|}xj
|r�
t|_x?|jD]4}t|tj�rD|j	d
krDt
|_PqDqDW|jdkr�|j	}n	|j}||jkr�|j|}x�|j
D]0}||kr�g||<n||j|�

q�Wn�|j}|jdk	r%
|jjr%
Pn||jkr�
|j|}xD|j
D]6}||j�krp
g||<n||j|�

qK
Wn|j}q+WqW|S(suThe
main interface with the bottom matcher. The tree is
        traversed from the bottom using the constructed
        automaton. Nodes are only checked once as the tree is
        retraversed. When the automaton fails, we give it one more
        shot(in case the above tree matches as a whole with the
        rejected leaf), then we break for the next leaf. There is the
        special case of multiple arguments(see code comments) where we
        recheck the nodes

        Args:
           The leaves of the AST tree to be matched

        Returns:
           A dictionary of node matches with fixers as the keys
       
u
;i
N(RRtlisttTruetwas_checkedtchildrenR%R
tLeaftvaluetFalsettypeRRRtparenttNonetkeys(	Rtleavestcurrent_ac_nodetresultstleaftcurrent_ast_nodetchildt
node_tokenR((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pytrunSs>	




	
	
!
	

	





	








c

s*d
GH�f
d���|j�

dGHdS(s<Prints
a graphviz diagram of the BM automaton(for debugging)s
digraph
g{c
svxo|jj
�D]^}
|j|
}d
|j|jt|
�
t|j�
fGH|
dkrd|jGHn�|�

qWdS(Ns%d
-> %d [label=%s] //%si
(RR7R	t	type_reprtstrRR
(tnodetsubnode_keytsubnode(
t
print_node(s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyRE�s




'


t
}N(
R(
R((
REs+/usr/lib64/python2.7/lib2to3/btm_matcher.pytprint_ac�s


(R
RRRR$RR?RG(((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyRs			"	=c
CshtsXd
dl
m}

x?|
jj�D]+\}}t|�
tkr&|t|<q&q&Wntj||�S(Ni
(
tpython_symbols(t_type_reprstpygramRHt__dict__titemsR4tintt
setdefault(ttype_numRHtnametval((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyR@�s


(Rt
__author__RRtcollectionsRRR
t	btm_utilsRtobjectRRRIR@(((s+/usr/lib64/python2.7/lib2to3/btm_matcher.pyt<module>s


	�