3
\ @ s d dl mZmZ d dlmZmZmZmZmZm Z m
Z
ddddddd d
d gfd d
ddddddddddddddgfd d gfgddd d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4gfd d5d6gfgd7Z e d8 j e d9 d: d;d< Z
G d=d> d>eZd?S )@ )
alternates
FixImports)NameComma
FromImportNewlinefind_indentationNodesymszurllib.requestZ URLopenerZFancyURLopenerZ urlretrieveZ
_urlopenerZurlopenZ
urlcleanupZ pathname2urlZ url2pathnamez urllib.parseZquoteZ
quote_plusZunquoteZ unquote_plusZ urlencodeZ splitattrZ splithostZ
splitnportZ splitpasswdZ splitportZ
splitqueryZsplittagZ splittypeZ splituserZ
splitvaluez urllib.errorZContentTooShortErrorZinstall_openerZ build_openerZRequestZOpenerDirectorZ BaseHandlerZHTTPDefaultErrorHandlerZHTTPRedirectHandlerZHTTPCookieProcessorZ ProxyHandlerZHTTPPasswordMgrZ HTTPPasswordMgrWithDefaultRealmZAbstractBasicAuthHandlerZHTTPBasicAuthHandlerZProxyBasicAuthHandlerZAbstractDigestAuthHandlerZHTTPDigestAuthHandlerZProxyDigestAuthHandlerZ HTTPHandlerZ HTTPSHandlerZ FileHandlerZ
FTPHandlerZCacheFTPHandlerZUnknownHandlerZURLErrorZ HTTPError)urlliburllib2r r c c s~ t } xrtj D ]f\}}x\|D ]T}|\}}t|}d||f V d|||f V d| V d| V d||f V q W qW d S )Nzimport_name< 'import' (module=%r
| dotted_as_names< any* module=%r any* >) >
zimport_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* >
)setMAPPINGitemsr )ZbareZ
old_moduleZchangeschangeZ
new_modulemembers r 0/usr/lib64/python3.6/lib2to3/fixes/fix_urllib.py
build_pattern0 s
r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) FixUrllibc C s dj t S )N|)joinr )selfr r r r I s zFixUrllib.build_patternc C sz |j d}|j}g }x6t|j d d D ] }|jt|d |dt g q(W |jtt|j d d |d |j| d S )Nmoduler
r )prefixr ) getr r valueextendr r appendreplace)r noderesultsZ
import_modprefnamesnamer r r transform_importL s
zFixUrllib.transform_importc C s> |j d}|j}|j d}|rt|tr0|d }d }x*t|j D ] }|j|d kr@|d }P q@W |rx|jt||d n | j|d ng }i } |d }
x|
D ]}|j t
j kr|j d j} |j d j} n
|j} d } | d krxPt|j D ]B}| |d kr|d | kr|j
|d | j|d g j
| qW qW g }
t|}d
}d d }x|D ]}| | }g }x2|d d D ]"}|j||| |j
t qlW |j||d | t||}| s|jjj|r||_|
j
| d
}qNW |
r.g }x&|
d d D ]}|j|t g qW |j
|
d |j| n | j|d d S )N
mod_membermemberr r
)r z!This is an invalid module elementr ,Tc S sX | j tjkrHt| jd j|d| jd j | jd j g}ttj|gS t| j|dgS )Nr )r r
r* )typer
import_as_namer childrenr Zcloner )r& r Zkidsr r r handle_name s z/FixUrllib.transform_member..handle_nameFz All module elements are invalidr r r r )r r
isinstancelistr r r! r cannot_convertr, r
r- r. r
setdefaultr r r r parentendswithr )r r" r# r( r$ r) new_namer modulesZmod_dictr Zas_name member_nameZ new_nodesZ indentationfirstr/ r Zeltsr% ZeltnewZnodesZnew_noder r r transform_member\ sh
zFixUrllib.transform_memberc C s |j d}|j d}d }t|tr*|d }x*t|j D ] }|j|d kr6|d }P q6W |rp|jt||jd n | j|d d S )Nbare_with_attrr) r r
)r z!This is an invalid module element) r r0 r1 r r r! r r r2 )r r" r# Z
module_dotr) r6 r r r r
transform_dot s
zFixUrllib.transform_dotc C sz |j dr| j|| n^|j dr0| j|| nF|j drH| j|| n.|j dr`| j|d n|j drv| j|d d S )Nr r( r< Z module_starzCannot handle star imports.Z module_asz#This module is now multiple modules)r r' r; r= r2 )r r" r# r r r transform s
zFixUrllib.transformN)__name__
__module__ __qualname__r r' r; r= r> 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> $