3
\4 � 1 @ s� d Z ddlmZ ddlmZmZ ddddddd d
d d d d
d
ddddddddddddddd d d d d d!d"d"d#d$d%d&d'd(d(d(d)d*d*d+d,d-�0Zd.d/� Zefd0d1�ZG d2d3� d3ej �Z
d4S )5z/Fix incompatible imports and module references.� )�
fixer_base)�Name�
attr_chain�io�pickle�builtins�copyregZqueueZ socketserverZ configparser�reprlibztkinter.filedialogztkinter.simpledialogztkinter.colorchooserztkinter.commondialogztkinter.dialogz tkinter.dndz tkinter.fontztkinter.messageboxztkinter.scrolledtextztkinter.constantsz tkinter.tixz tkinter.ttkZtkinterZ _markupbase�winreg�_threadZ
_dummy_threadzdbm.bsdzdbm.dumbzdbm.ndbmzdbm.gnuz
xmlrpc.clientz
xmlrpc.serverz http.clientz
html.entitiesz html.parserz http.cookieszhttp.cookiejarz http.server�
subprocess� collectionsz urllib.parsezurllib.robotparser)0�StringIOZ cStringIOZcPickleZ __builtin__Zcopy_regZQueueZ SocketServerZ ConfigParser�reprZ
FileDialogZ tkFileDialogZ SimpleDialogZtkSimpleDialogZtkColorChooserZtkCommonDialogZDialogZTkdndZtkFontZ tkMessageBoxZ ScrolledTextZ TkconstantsZTixZttkZTkinterZ
markupbase�_winregZthreadZ dummy_threadZdbhashZdumbdbmZdbmZgdbmZ xmlrpclibZDocXMLRPCServerZSimpleXMLRPCServerZhttplibZhtmlentitydefsZ
HTMLParserZCookieZ cookielibZBaseHTTPServerZSimpleHTTPServerZ
CGIHTTPServerZcommands�
UserString�UserListZurlparseZ robotparserc C s ddj tt| �� d S )N�(�|�))�join�mapr )�members� r �1/usr/lib64/python3.6/lib2to3/fixes/fix_imports.py�
alternates= s r c c sT dj dd� | D ��}t| j� �}d||f V d| V d||f V d| V d S )Nz | c S s g | ] }d | �qS )zmodule_name='%s'r )�.0�keyr r r �
<listcomp>B s z!build_pattern.<locals>.<listcomp>zyname_import=import_name< 'import' ((%s) |
multiple_imports=dotted_as_names< any* (%s) any* >) >
z�import_from< 'from' (%s) 'import' ['(']
( any | import_as_name< any 'as' any > |
import_as_names< any* >) [')'] >
z�import_name< 'import' (dotted_as_name< (%s) 'as' any > |
multiple_imports=dotted_as_names<
any* dotted_as_name< (%s) 'as' any > any* >) >
z3power< bare_with_attr=(%s) trailer<'.' any > any* >)r r �keys)�mappingZmod_listZ
bare_namesr r r �
build_patternA s
r! c sT e Zd ZdZdZeZdZdd� Z� fdd�Z � fdd�Z
� fd d
�Z d d � Z � Z
S )
�
FixImportsT� c C s dj t| j��S )Nr )r r! r )�selfr r r r! ` s zFixImports.build_patternc s | j � | _tt| �j� d S )N)r! ZPATTERN�superr" �compile_pattern)r$ )� __class__r r r& c s
zFixImports.compile_patternc sH t t| �j� � |�}|rDd|kr@t� fdd�t|d�D ��r@dS |S dS )N�bare_with_attrc 3 s | ]}� |�V qd S )Nr )r �obj)�matchr r � <genexpr>q s z#FixImports.match.<locals>.<genexpr>�parentF)r% r" r* �anyr )r$ �node�results)r' )r* r r* j s zFixImports.matchc s t t| �j||� i | _d S )N)r% r" �
start_tree�replace)r$ Ztree�filename)r' r r r0 v s zFixImports.start_treec C s� |j d�}|rh|j}| j| }|jt||jd�� d|krD|| j|< d|kr�| j|�}|r�| j||� n2|d d }| jj |j�}|r�|jt||jd�� d S )NZ module_name)�prefixZ name_importZmultiple_importsr( � )�get�valuer r1 r r3 r* � transform)r$ r. r/ Z
import_modZmod_name�new_nameZ bare_namer r r r7 z s
zFixImports.transform)�__name__�
__module__� __qualname__Z
BM_compatibleZkeep_line_order�MAPPINGr Z run_orderr! r& r* r0 r7 �
__classcell__r r )r' r r" U s r" N) �__doc__� r Z
fixer_utilr r r< r r! ZBaseFixr" r r r r �<module> sj