U
    h                      @   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   P/var/www/html/venv/lib/python3.8/site-packages/libfuturize/fixes/fix_UserDict.pyr   5   s    zFixUserdict.build_patternc                    s   |   | _tt|   d S N)r   ZPATTERNsuperr   compile_patternr   	__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 r   r   ).0objmatchr   r   	<genexpr>F   s     z$FixUserdict.match.<locals>.<genexpr>parentF)r   r   r   anyr   )r   noderesultsr   r   r   r   ?   s    zFixUserdict.matchc                    s   t t| || i | _d S r   )r   r   
start_treereplace)r   treefilenamer   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 )Nmodule_name)prefixZname_importZmultiple_importsr   r   )	getvalue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   r1   r
   r   r   r   r   <module>   s
   	