U
e5d @ s d Z ddlmZmZ ddlmZmZmZmZm Z m
Z
m Z dddddd d
d d gfd
ddddddddddddddd gfd d gfgdd d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5gfd d6d7gfgd8Z e d9
e d: d; dd? d?eZd@S )AzFix changes imports of urllib which are now incompatible.
This is rather similar to fix_imports, but because of the more
complex nature of the fixing for urllib, it has its own fixer.
)
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 sv t } t D ]b\}}|D ]T}|\}}t|}d||f V d|||f V d| V d| V d||f V qqd 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.8/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 dd D ] }|t|d |dt g q&|tt|j d d |d || dS )zTransform for the basic import case. Replaces the old
import name with a comma separated list of its
replacements.
moduleNr prefix) getr r valueextendr r appendreplace)r noderesultsZ
import_modprefnamesnamer r r transform_importL s
zFixUrllib.transform_importc C s& | d}|j}| d}|rt|tr0|d }d}t|j D ] }|j|d kr>|d } q^q>|rv|t||d n | |d ng }i } |d }
|
D ]}|j t
j kr|j d j} |j d j} n
|j} d} | d
krt|j D ]B}| |d kr|d | kr |
|d | |d g
| qqg }
t|}d }d d
}|D ]}| | }g }|dd D ]"}|||| |
t q^|||d | t||}|r|jj|r||_|
| d}qB|
rg }|
dd D ]}||t g q|
|
d || n | |d dS )zTransform for imports of specific module elements. Replaces
the module to be imported from with the appropriate new
module.
mod_membermemberr Nr
r !This is an invalid module elementr ,Tc S sX | j tjkrHt| jd j|d| jd | jd 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_namer Fz All module elements are invalid)r r
isinstancelistr r r" r cannot_convertr. r
r/ r0 r!
setdefaultr r r r parentendswithr )r r# r$ r) r% r* new_namer modulesZmod_dictr Zas_name member_nameZ new_nodesZ indentationfirstr1 r Zeltsr& ZeltnewZnodesZnew_noder r r transform_member\ sh
zFixUrllib.transform_memberc C s~ | d}| d}d}t|tr*|d }t|j D ] }|j|d kr4|d } qTq4|rn|t||jd n | |d dS )z.Transform for calls to module members in code.bare_with_attrr* Nr r
r r+ ) r r2 r3 r r r" r r r4 )r r# r$ Z
module_dotr* r8 r r r r
transform_dot s
zFixUrllib.transform_dotc C sz | dr| || n^| dr0| || nF| drH| || n.| dr`| |d n| drv| |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? r4 )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)__doc__Zlib2to3.fixes.fix_importsr r Zlib2to3.fixer_utilr r r r r r r
r r! r r r r r r s~ $
!