a
XC?h� � @ s� d dl mZmZ d dlmZmZmZmZmZm Z m
Z
dg d�fdg d�fddgfgdg d �fdd
d gfgd �Z e d
� e d d � dd� Z
G dd� de�ZdS )� )�
alternates�
FixImports)�Name�Comma�
FromImport�Newline�find_indentation�Node�symszurllib.request) Z URLopenerZFancyURLopenerZ urlretrieveZ
_urlopener�urlopenZ
urlcleanupZ pathname2urlZ url2pathnameZ
getproxiesz urllib.parse)ZquoteZ
quote_plusZunquoteZ unquote_plusZ urlencodeZ splitattrZ splithostZ
splitnportZ splitpasswdZ splitportZ
splitqueryZsplittagZ splittypeZ splituserZ
splitvaluez urllib.errorZContentTooShortError)r Zinstall_openerZ build_openerZRequestZOpenerDirectorZ BaseHandlerZHTTPDefaultErrorHandlerZHTTPRedirectHandlerZHTTPCookieProcessorZ ProxyHandlerZHTTPPasswordMgrZ HTTPPasswordMgrWithDefaultRealmZAbstractBasicAuthHandlerZHTTPBasicAuthHandlerZProxyBasicAuthHandlerZAbstractDigestAuthHandlerZHTTPDigestAuthHandlerZProxyDigestAuthHandlerZ HTTPHandlerZ HTTPSHandlerZ FileHandlerZ
FTPHandlerZCacheFTPHandlerZUnknownHandlerZURLErrorZ HTTPError)�urllib�urllib2r
r � c c sv t � } t�� D ]b\}}|D ]T}|\}}t|�}d||f V d|||f V d| V d| V d||f V qqd S )Nz�import_name< 'import' (module=%r
| dotted_as_names< any* module=%r any* >) >
z�import_from< 'from' mod_member=%r 'import'
( member=%s | import_as_name< member=%s 'as' any > |
import_as_names< members=any* >) >
zIimport_from< 'from' module_star=%r 'import' star='*' >
ztimport_name< 'import'
dotted_as_name< module_as=%r 'as' any > >
zKpower< bare_with_attr=%r trailer< '.' member=%s > any* >
)�set�MAPPING�itemsr )ZbareZ
old_moduleZchanges�changeZ
new_module�members� r �=/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_urllib.py�
build_pattern0 s( �����r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � FixUrllibc C s d� t� �S )N�|)�joinr )�selfr r r r I s zFixUrllib.build_patternc C sv |� d�}|j}g }t|j d d� D ] }|�t|d |d�t� g� q&|�tt|j d d |d�� |�|� d S )N�module�����r ��prefix) �getr r �value�extendr r �append�replace)r �node�resultsZ
import_mod�pref�names�namer r r �transform_importL s
zFixUrllib.transform_importc C s& |� d�}|j}|� d�}|r�t|t�r0|d }d }t|j D ] }|j|d v r>|d } q^q>|rv|�t||d�� n | �|d� �n�g }i } |d }
|
D ]�}|j t
j kr�|j d j} |j d j} n
|j} d } | d kr�t|j D ]B}| |d v r�|d | v�r |�
|d � | �|d g ��
|� q�q�g }
t|�}d
}d d � }|D ]�}| | }g }|d d
� D ]"}|�|||�� |�
t� � �q^|�||d
|�� t||�}|�r�|jj�|��r�||_|
�
|� d}�qB|
�rg }|
d d
� D ]}|�|t� g� �q�|�
|
d
� |�|� n | �|d� d S )N�
mod_member�memberr r r �!This is an invalid module elementr � �,Tc S sX | j tjkrHt| jd j|d�| jd �� | jd �� g}ttj|�gS t| j|d�gS )Nr r r r- )�typer
�import_as_namer �childrenr Zcloner )r( r Zkidsr r r � handle_name� s �z/FixUrllib.transform_member..handle_namer Fz All module elements are invalid)r r �
isinstance�listr r r# r �cannot_convertr/ r
r0 r1 r" �
setdefaultr r! r r �parent�endswithr )r r$ r% r* r& r+ �new_namer �modulesZmod_dictr Zas_name� member_nameZ new_nodesZ indentation�firstr2 r �eltsr' �elt�newZnodesZnew_noder r r �transform_member\ sh
zFixUrllib.transform_memberc C s~ |� d�}|� d�}d }t|t�r*|d }t|j D ] }|j|d v r4|d } qTq4|rn|�t||jd�� n | �|d� d S )N�bare_with_attrr+ r r r r, ) r r3 r4 r r r# r r r5 )r r$ r% Z
module_dotr+ r9 r r r r �
transform_dot� s
�
zFixUrllib.transform_dotc C sz |� d�r| �||� n^|� d�r0| �||� nF|� d�rH| �||� n.|� d�r`| �|d� n|� d�rv| �|d� d S )Nr r* rA Z module_starzCannot handle star imports.Z module_asz#This module is now multiple modules)r r) r@ rB r5 )r r$ r% r r r � transform� s
zFixUrllib.transformN)�__name__�
__module__� __qualname__r r) r@ rB rC r r r r r G s
Lr N)Zlib2to3.fixes.fix_importsr r Zlib2to3.fixer_utilr r r r r r r
r r" r r r r r r � s, $����
����!