o
    bc	                     @   sR   d Z ddlmZ ddlmZ ddlmZ ddlm	Z	m
Z
mZ G dd dejZdS )	a7  
Fixer that changes zip(seq0, seq1, ...) into list(zip(seq0, seq1, ...)
unless there exists a 'from future_builtins import zip' statement in the
top-level namespace.

We avoid the transformation if the zip() call is directly contained in
iter(<>), list(<>), tuple(<>), sorted(<>), ...join(<>), or for V in <>:.
   )
fixer_base)Node)python_symbols)NameArgListin_special_contextc                   @   s    e Zd ZdZdZdZdd ZdS )FixZipTzN
    power< 'zip' args=trailer< '(' [any] ')' > [trailers=trailer*]
    >
    zfuture_builtins.zipc                 C   s   |  |rd S t|rd S |d  }d|_g }d|v r-dd |d D }|D ]}d|_q'ttjtd|gdd}ttjtdt|gg| }|j|_|S )	Nargs trailersc                 S   s   g | ]}|  qS  )clone).0nr   r   ,/usr/lib/python3.10/lib2to3/fixes/fix_zip.py
<listcomp>'   s    z$FixZip.transform.<locals>.<listcomp>zip)prefixlist)	should_skipr   r   r   r   symspowerr   r   )selfnoderesultsr	   r   r   newr   r   r   	transform   s   
zFixZip.transformN)__name__
__module____qualname__BM_compatiblePATTERNskip_onr   r   r   r   r   r      s
    r   N)__doc__r
   r   pytreer   pygramr   r   
fixer_utilr   r   r   ConditionalFixr   r   r   r   r   <module>   s    
