B
    	Q[                  @   sP   d Z ddlmZ ddlmZmZ ddlmZmZm	Z	 ddiZ
G dd de	Zd	S )
zCFix UserDict.

Incomplete!

TODO: base this on fix_urllib perhaps?
    )
fixer_base)Name
attr_chain)
alternatesbuild_pattern
FixImportsUserDictcollectionsc                   sT   e Zd ZdZdZeZdZdd Z fddZ	 fddZ
 fd	d
Zdd Z  ZS )FixUserdictT   c             C   s   d t| jS )N|)joinr   mapping)self r   =lib/python3.7/site-packages/libfuturize/fixes/fix_UserDict.pyr   5   s    zFixUserdict.build_patternc                s   |   | _tt|   d S )N)r   ZPATTERNsuperr   compile_pattern)r   )	__class__r   r   r   8   s    
zFixUserdict.compile_patternc                sH   t t| j  |}|rDd|kr@t fddt|dD r@dS |S dS )Nbare_with_attrc             3   s   | ]} |V  qd S )Nr   ).0obj)matchr   r   	<genexpr>F   s    z$FixUserdict.match.<locals>.<genexpr>parentF)r   r   r   anyr   )r   noderesults)r   )r   r   r   ?   s    zFixUserdict.matchc                s   t t| || i | _d S )N)r   r   
start_treereplace)r   Ztreefilename)r   r   r   r   K   s    zFixUserdict.start_treec             C   s   | d}|rl|j}t| j| }|t||jd d|krH|| j|< d|kr| |}|r| || n2|d d }| j |j}|r|t||jd d S )NZmodule_name)prefixZname_importZmultiple_importsr   r   )	getvalueZunicoder   r   r   r!   r   	transform)r   r   r   Z
import_modmod_namenew_nameZ	bare_namer   r   r   r$   O   s    


zFixUserdict.transform)__name__
__module____qualname__ZBM_compatibleZkeep_line_orderMAPPINGr   Z	run_orderr   r   r   r   r$   __classcell__r   r   )r   r   r
   *   s   r
   N)__doc__Zlib2to3r   Zlib2to3.fixer_utilr   r   Zlib2to3.fixes.fix_importsr   r   r   r*   r
   r   r   r   r   <module>   s
   