o
    \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   a/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/libfuturize/fixes/fix_UserDict.pyr   5   s   zFixUserdict.build_patternc                    s   |   | _tt|   d S N)r   PATTERNsuperr   compile_patternr   	__class__r   r   r   8   s   
zFixUserdict.compile_patternc                    sH   t t| j  |}|r"d|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   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}|r;|j}t| j| }|t||jd d|v r$|| j|< d|v r7| |}|r9| || d S d S d S |d d }| j |j}|rV|t||jd d S d S )Nmodule_name)prefixname_importmultiple_importsr   r   )	getvalueunicoder   r$   r   r(   r   	transform)r   r!   r"   
import_modmod_namenew_name	bare_namer   r   r   r.   O   s$   


zFixUserdict.transform)__name__
__module____qualname__BM_compatiblekeep_line_orderMAPPINGr   	run_orderr   r   r   r#   r.   __classcell__r   r   r   r   r
   *   s    r
   N)__doc__lib2to3r   lib2to3.fixer_utilr   r   lib2to3.fixes.fix_importsr   r   r   r8   r
   r   r   r   r   <module>   s    	