o
    bc                     @   sj   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 ddlmZ ej	dhB Z
G dd	 d	ejZd
S )aj  Fixer for dict methods.

d.keys() -> list(d.keys())
d.items() -> list(d.items())
d.values() -> list(d.values())

d.iterkeys() -> iter(d.keys())
d.iteritems() -> iter(d.items())
d.itervalues() -> iter(d.values())

d.viewkeys() -> d.keys()
d.viewitems() -> d.items()
d.viewvalues() -> d.values()

Except in certain very specific contexts: the iter() can be dropped
when the context is list(), sorted(), iter() or for...in; the list()
can be dropped when the context is list() or sorted() (but not iter()
or for...in!). Special contexts that apply to both: list(), sorted(), tuple()
set(), any(), all(), sum().

Note: iter(d.keys()) could be written as iter(d) but since the
original d.iterkeys() was also redundant we don't fix this.  And there
are (rare) contexts where it makes a difference (e.g. when passing it
as an argument to a function that introspects the argument).
   )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v s2J t|d	d
 |D }dd
 |D }| oH| ||}
|t|jt t	||j
dg|d  g }t|j|}|
sz|	szd|_
tt	|rtdnd|g}|rt|j|g| }|j
|_
|S )Nheadmethod    tailr	   view   )keysitemsvaluesc                 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   )prefixparens list)symsvalue
startswithreprin_special_contextr   Nodetrailerr   r   r   r   powerr   )selfnoderesultsr   r   r   r"   method_nameisiterisview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__BM_compatiblePATTERNr3   P1r   compile_patternr6   P2r:   r&   r   r   r   r   r
   )   s    


r
   N)__doc__r    r   r   r   r   r   r   r   r9   r8   BaseFixr
   r   r   r   r   <module>   s   