a
    W%b                      @   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   n/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/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|v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       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|v rH|| j|< d|v 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   r2   r
   r   r   r   r   <module>   s
   	