�
�A�[c@s�dZd
dl
mZ
d
dlmZ
d
dlmZmZmZm	Z	m
Z
mZmZm
Z
mZmZ
d
dlmZmZmZ
d�Zdejf
d��YZd	S(
uD
Fixer for:
(a,)* *b (,c)* [,] = s
for (a,)* *b (,c)* [,] in d: ...
i����(
t
fixer_base(
tcount(
tAssigntCommatCalltNewlinetNametNumberttokentsymstNodetLeaf(tindentationtsuitifyt	commatizecCs�g}t|�
}t|
�
}|d
kr�t
tjt|�
t
tjttjd�t
tj	ttj
d�t|�
g�ttjd�g�g�}|j
|�

|j
ttjddd�
�

nt
tjttjddd�
t|�
t
tjttjd�t
tj	|d
kr(
t|�
ntdd	�ttj
d�|
d
kry
t
tjttjd
�t|�
g�ntdd	�g�ttjd�ttjd�g�g�}|j
|�

|
d
kr�|j
ttjddd�
�

t
tjt|dd�

t
tjttjd�t
tj	t
tjttjd
�t|�
g�ttj
d�g�ttjd�g�g�}	|j
|	�

nt
tj|�}
|
S(u�
    Accepts num_pre and num_post, which are counts of values
    before and after the starg (not including the starg)
    Returns a source fit for Assign() from fixer_util
    iu
[u
:u
]u
+tprefixu
 i
uu
-(tunicodeR
R	tpowerRttrailerRRtLSQBt	subscripttCOLONRtRSQBtappendtPLUStfactortMINUSt
arith_expr(tnum_pretnum_posttLISTNAMEtITERNAMEtchildrentpretposttpre_partt	main_partt	post_parttsource((sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pytassignment_sources


o


"
�




�


tFixUnpackingc
Bs)eZ
dZd
�Zd�Zd�ZRS(u9

    expl=expr_stmt< testlist_star_expr<
        pre=(any ',')*
            star_expr< '*' name=NAME >
        post=(',' any)* [','] > '=' source=any > |
    impl=for_stmt< 'for' lst=exprlist<
        pre=(any ',')*
            star_expr< '*' name=NAME >
        post=(',' any)* [','] > 'in' it=any ':' suite=any>csF
�f
d
�d	D�
\}}}}g|D]$}|jt
jkr)|j�^q)}d|_g|D]$}|jt
jkrc|j�^qc}gt||j�g
|�
D]}|j�^q�}|jt��

d|_tt	|j
�
tt	d�
|j�g
��}	t|tt
|�
t
|�
|j
|j��}
|	|
fS(
Nc
3s|]}
�j|
�
V
qdS(
N(
tget(t.0t
n(
tresults(sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pys	<genexpr>3supreunameupostusourceu
 uulist(upreunameupostusource(ttypeRtNAMEtcloneRRRRRRRRR'tlenR(tselftnodeR,R!tnameR"R&R+ttargett
setup_linet
power_line((
R,sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pytfix_explicit_context2s
"
1
	
1
6
	
0
0
csU
�f
d
�d	D�
\}}}}g|D]$}|jt
jkr)|j�^q)}d|_g|D]$}|jt
jkrc|j�^qc}gt||j�g
|�
D]}|j�^q�}|jt��

|j�}	d|	_tt	|j
�
tt	d�
t	|j�
g
��}
t|t
t|�
t|�
|j
|j��}|
|fS(
u_
        Only example of the implicit context is
        a for loop, so only fix that.
        c
3s|]}
�j|
�
V
qdS(
N(
R)(R*R+(
R,(sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pys	<genexpr>Esupreunameupostuitu
 uulist(upreunameupostuit(R-RR.R/RRRRRRRRRR'R0(R1R2R,R!R3R"titR+R4R&R5R6((
R,sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pytfix_implicit_context@s"
1
	
1
6

	
3
0
cCs�
|jd
�
|_
|jd�
|_|jd�
|jd�
}}|dk	r�|j|
|�\}}|j|_t|j�
|_|j	t
��

|
j}|
j�}|j||�
|j||�
n
|dk	r�
|j
|
|�\}}t|
�

g|
jD]}	|	jtjkr
|	^q
d}
d|_|
jdj|_t|
jd�
|
jd_|
jdt
��
|
jd|�
|
jdt
��
|
jd|�
|jd	�
jt|jd
d�

�

ndS(
u�

        a,b,c,d,e,f,*g,h,i = range(100) changes to
        _3to2list = list(range(100))
        a,b,c,d,e,f,g,h,i, = _3to2list[:6] + [_3to2list[6:-2]] + _3to2list[-2:]

        and

        for a,b,*c,d,e in iter_of_iters: do_stuff changes to
        for _3to2iter in iter_of_iters:
            _3to2list = list(_3to2iter)
            a,b,c,d,e, = _3to2list[:2] + [_3to2list[2:-2]] + _3to2list[-2:]
            do_stuff
        u	_3to2listu	_3to2iteruexpluimpliui
iulstRu
 N(tnew_nameRRR)tNoneR7RRtparenttappend_childRtremovetinsert_childR9R
R R-R	tsuitetvaluetreplaceR(R1R2R,texpltimplR5R6R<t
it
kR@((sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pyt	transformSs0







	







2
	






(t__name__t
__module__tPATTERNR7R9RG(((sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pyR(&s
		N(t__doc__tlib2to3Rt	itertoolsR
tlib2to3.fixer_utilRRRRRRRR	R
Rtlibfuturize.fixer_utilRR
RR'tBaseFixR((((sR/opt/alt/python27/lib/python2.7/site-packages/libpasteurize/fixes/fix_unpacking.pyt<module>s

F