
U                 @   sn   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
 Gd 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                   sv   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   S)FixUserdictT   c             C   s   d j  t |  j   S)N|)joinr   mapping)self r   @/usr/lib/python3/dist-packages/libfuturize/fixes/fix_UserDict.pyr   5   s    zFixUserdict.build_patternc                s&   |  j    |  _ t t |   j   d  S)N)r   ZPATTERNsuperr   compile_pattern)r   )	__class__r   r   r   8   s    zFixUserdict.compile_patternc                sd   t  t |   j     |  } | r` d | k r\ t   f d d   t | d  D  r\ d S| Sd 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 |   j | |  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   | j  d  } | r | j } t |  j |  } | j t | d | j  d | k rf | |  j | <d | k r |  j |  } | r |  j | |  nE | d d } |  j j  | j  } | r | j t | d | j  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$   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
   