U

<O�X�!�@s�dZd
dl
mZmZmZmZ
d
dlmZ
d
dlZd
dl	Z	d
dl
Z
ddlmZ
ddl
mZmZmZmZmZ
e
jdkr�d
d	lmZmZ
d
d
lmZmZ
nd
dlmZmZmZmZ
dd
�Zdd�Zdd�Ze� de�
ddd�
Z!ddd�
Z"dS)z�
Functions to convert unicode IRIs into ASCII byte string URIs and back. Exports
the following items:

 - iri_to_uri()
 - uri_to_iri()
�)�unicode_literals�division�absolute_import�print_function)
�idnaN�
)
�unwrap)�byte_cls�str_cls�	type_name�
bytes_to_list�	int_types)
�)�urlsplit�
urlunsplit)�quote�unquote)r�unquote_to_bytesrrc
Cst|t
�sttd
t|�
��
�
d}
tjdkr�|�d�
s�|�d�
s�d}t�	d|�}|rp|�
d�
}d|t|�
d�}t|�
}|r�||dd�}t
|dd	��
}
nt|�
}|
dkr�t
|j�
}
|j}|dk	r�|�d
�
}t
|jdd�}t
|jdd�}|j}|dk	�
r
t
|�
�d
�
}d}	|dk	�
r:|	|7}	|�
r2|	d|7}	|	d7}	|dk	�
rL|	|7}	|dk	�
r�|
dk�
of|dk}
|
dk�
ox|dk}|
�
s�|�
s�|	d|7}	t
|jdd�}t
|jdd�}
t
|jdd�}|
dk�
r�|dk�
r�|dk�
r�d}|dk�
r�d}t|
|	||
|f�
}t|t
��r|�d�
}|S)z�
    Normalizes and encodes a unicode IRI into an ASCII byte string URI

    :param value:
        A unicode string of an IRI

    :return:
        A byte string of the ASCII-encoded URI
    z@
            value must be a unicode string, not %s
            N)��zhttp://zhttps://z	^[^:]*://r
r���rz!$&'()*+,;=�
�safe�ascii��
:�
@shttps80shttpss443z/!$&'()*+,;=@:z/?!$&'()*+,;=@:�
/��latin1)�
isinstancer
�	TypeErrorrr�sys�version_info�
startswith�re�match�group�lenr�	_urlquote�scheme�hostname�encode�username�password�port�path�query�fragmentr)�valuer*Zreal_prefixZprefix_match�parsedr+r-r.r/�netlocZdefault_httpZ
default_httpsr0r1r2�output�r7�/_iri.py�
iri_to_uri%sf


�










































r9c
Cs(
t|t
�sttd
t|�
��
�
t|�
}
|
j}|dk	r<|�d�
}t|
j	ddgd�}t|
j
ddgd�}|
j}|rt|�d�
}|
j}|r�t|t
�s�|�d�
}d}|dk	r�||7}|r�|d|7}|d7}|dk	r�||7}|dk	r�|dt|�
7}t|
jd	g
d
d�}t|
jdd
gd
d�}	t|
j�
}
t||||	|
f�
S)z�
    Converts an ASCII URI byte string into a unicode IRI

    :param value:
        An ASCII-encoded byte string of the URI

    :return:
        A unicode string of the IRI
    z=
            value must be a byte string, not %s
            Nr�
:�
@)
�remaprr�
/T)r<�preserve�
&�
=)r r	r!rrrr*�decode�_urlunquoter-r.r+r/r
r
r0r1r2r)r3r4r*r-r.r+r/r5r0r1r2r7r7r8�
uri_to_irius>


�























rCc
Cs4t|j
|j|j��
}
d
d�|
D�
}d�|�
|jfS)a>

    Error handler for decoding UTF-8 parts of a URI into an IRI. Leaves byte
    sequences encoded in %XX format, but as part of a unicode string.

    :param exc:
        The UnicodeDecodeError exception

    :return:
        A 2-element tuple of (replacement unicode string, integer index to
        resume at)
    c
Ssg|]}
d|
�qS)
�%%%02xr7)�.0Znumr7r7r8�
<listcomp>�sz,_iri_utf8_errors_handler.<locals>.<listcomp>r)r�object�start�end�join)�excZ
bytes_as_ints�replacementsr7r7r8�_iri_utf8_errors_handler�s


rM�iriutf8rcs�|d
ks|dkrd
Sg�t�
d|�rX�
f
dd�}t�d||�}�f
dd�}t�d||�}t|�d	�
�
�d	�
d
�}t|t�s�|�d�
}t��
dkr��f
d
d�}t�d||�}|S)a


    Quotes a unicode string for use in a URL

    :param string:
        A unicode string

    :param safe:
        A unicode string of character to not encode

    :return:
        None (if string is None) or an ASCII byte string of the quoted string
    Nrz%[0-9a-fA-F]{2}c
s@t|�
d
�
�
}
|
�dd�}t��
D]}|�|dt|�
�}q"|S)Nr
�utf-8rNrD)rr'rA�list�replace�ord)r&�byte_stringZunicode_stringZ	safe_charrr7r8�
_try_unescape�s





z _urlquote.<locals>._try_unescapez(?:%[0-9a-fA-F]{2})+c

s��|�
d
�
�d�
�

dS)Nr
r�
)�appendr'r,)
r&�
�escapesr7r8�_extract_escape�s

z"_urlquote.<locals>._extract_escaperOrrr
c

s
��d
�
S)Nr
)
�pop)
�
_rWr7r8�_return_escape�s
z!_urlquote.<locals>._return_escapes%00)r%�search�sub�urlquoter,r r	r()�stringrrTrYr6r\r7)rXrr8r)�s






r)c	Cs�|d
kr|S|dkrdS|rbdddddg}i}|
D].}|�d	�
}|||<|�
|�d
�
|�d
�
�}q2t|�
}|
r�|
D]$}|�
|�d
�
dt|�
�d
�
�}qr|�dd
�}|r�|��D]\}}|�
||�}q�|S)a
    Unquotes a URI portion from a byte string into unicode using UTF-8

    :param byte_string:
        A byte string of the data to unquote

    :param remap:
        A list of characters (as unicode) that should be re-mapped to a
        %XX encoding. This is used when characters are not valid in part of a
        URL.

    :param preserve:
        A bool - indicates that the chars to be remapped if they occur in
        non-hex form, should be preserved. E.g. / for URL path.

    :return:
        A unicode string
    Nrr�
�
�
�
�
r
rrDrOrN)rZrQr,rrRrA�items)	rSr<r>rLZpreserve_unmap�charZreplacementr6Zoriginalr7r7r8rB�s(










"

rB)
r)NN)#�__doc__Z
__future__rrrrZ	encodingsr�codecsr%r"Z_errorsrZ_typesr	r
rrr
r#ZurlparserrZurllibrr_rrZurllib.parser9rCrM�register_errorr)rBr7r7r7r8�<module>s"






P4
4