o
     JAf                     @   sh   	 d dl mZ d dl mZ d dl mZ d dlmZmZmZ d dl mZ ejdhB Z	G dd dej
Zd	S )
   )pytree)patcomp)
fixer_base)NameCallDot)
fixer_utiliterc                   @   s@   e Zd ZdZdZdd ZdZeeZ	dZ
ee
Zdd Zd	S )
FixDictTa  
    power< head=any+
         trailer< '.' method=('keys'|'items'|'values'|
                              'iterkeys'|'iteritems'|'itervalues'|
                              'viewkeys'|'viewitems'|'viewvalues') >
         parens=trailer< '(' ')' >
         tail=any*
    >
    c              	   C   s  |d }|d d }|d }| j }|j}|d}|d}	|s"|	r(|dd  }dd	 |D }d
d	 |D }| o>| ||}
|t|jt t||j	dg|d 
 g }t|j|}|
sp|	spd|_	tt|rjdnd|g}|r|t|j|g| }|j	|_	|S )Nheadmethod    tailr	   view   c                 S      g | ]}|  qS  clone.0nr   r   -/usr/lib/python3.10/lib2to3/fixes/fix_dict.py
<listcomp>A       z%FixDict.transform.<locals>.<listcomp>c                 S   r   r   r   r   r   r   r   r   B   r   )prefixZparens list)symsvalue
startswithin_special_contextr   ZNodeZtrailerr   r   r   r   Zpowerr   )selfnoderesultsr   r   r   r   Zmethod_nameisiterZisviewZspecialargsnewr   r   r   	transform6   s:   



zFixDict.transformz3power< func=NAME trailer< '(' node=any ')' > any* >zmfor_stmt< 'for' any 'in' node=any ':' any* >
            | comp_for< 'for' any 'in' node=any any* >
         c                 C   s   |j d u rdS i }|j j d ur/| j|j j |r/|d |u r/|r'|d jtv S |d jtjv S |s3dS | j|j |o@|d |u S )NFr#   func)parentp1matchr   iter_exemptr   consuming_callsp2)r"   r#   r%   r$   r   r   r   r!   Z   s   
zFixDict.in_special_contextN)__name__
__module____qualname__ZBM_compatibleZPATTERNr(   ZP1r   compile_patternr+   ZP2r/   r!   r   r   r   r   r
   )   s    


r
   N)r   r   r   r   r   r   r   r   r.   r-   ZBaseFixr
   r   r   r   r   <module>   s   