3
�ga| � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZmZ d dlZejd�Zeee f Ze
e ZG dd� de�Zd d
� Zeed �d d
�Zd dd�ZG dd� dej j �Z dd� Z dd� Z d ee!e
e d�dd�Z"dd� Z#d d � Z$dS ) � N)� OrderedDict)�parse_ns_headers)�pformat)�Any�List�Optional�Tuplez
, (?=[^ ;]+=)c @ sD e Zd ZdZejd
kZed�dd�Zdd�d d
�Z ed�d d �Z
dS )� JsonDictPreservingDuplicateKeysz2A specialized JSON dict preserving duplicate keys.� � )�itemsc C s || _ | j� d S )N)�_items�_ensure_items_used)�selfr � r �/usr/lib/python3.6/utils.py�__init__ s z(JsonDictPreservingDuplicateKeys.__init__N)�returnc C s | j rd| d< dS )a� HACK: Force `json.dumps()` to use `self.items()` instead of an empty dict.
Two JSON encoders are available on CPython: pure-Python (1) and C (2) implementations.
(1) The pure-python implementation will do a simple `if not dict: return '{}'`,
and we could fake that check by implementing the `__bool__()` method.
Source:
- <https://github.com/python/cpython/blob/9d318ad/Lib/json/encoder.py#L334-L336>
(2) On the other hand, the C implementation will do a check on the number of
items contained inside the dict, using a verification on `dict->ma_used`, which
is updated only when an item is added/removed from the dict. For that case,
there is no workaround but to add an item into the dict.
Sources:
- <https://github.com/python/cpython/blob/9d318ad/Modules/_json.c#L1581-L1582>
- <https://github.com/python/cpython/blob/9d318ad/Include/cpython/dictobject.h#L53>
- <https://github.com/python/cpython/blob/9d318ad/Include/cpython/dictobject.h#L17-L18>
To please both implementations, we simply add one item to the dict.
Z__hack__N)r
)r r r r r s z2JsonDictPreservingDuplicateKeys._ensure_items_usedc C s | j S )z4Return all items, duplicate ones included.
)r
)r r r r r 8 s z%JsonDictPreservingDuplicateKeys.items)r
r ) �__name__�
__module__� __qualname__�__doc__�sys� version_infoZSUPPORTS_SORTING�Itemsr r r r r r r r s
r c C s t j| td�S )N)Zobject_pairs_hook)�json�loadsr )�sr r r �&load_json_preserve_order_and_dupe_keys? s r )�dr c C s t | �S )N)r )r r r r � repr_dictC s r � c C sT ddddddg}| dkr d
S x|D ]\}}| |kr"P q"W | | d|� d��d|� �S ) a* Return a humanized string representation of a number of bytes.
>>> humanize_bytes(1)
'1 B'
>>> humanize_bytes(1024, precision=1)
'1.0 kB'
>>> humanize_bytes(1024 * 123, precision=1)
'123.0 kB'
>>> humanize_bytes(1024 * 12342, precision=1)
'12.1 MB'
>>> humanize_bytes(1024 * 12342, precision=2)
'12.05 MB'
>>> humanize_bytes(1024 * 1234, precision=2)
'1.21 MB'
>>> humanize_bytes(1024 * 1234 * 1111, precision=2)
'1.31 GB'
>>> humanize_bytes(1024 * 1234 * 1111, precision=1)
'1.3 GB'
� �2 �PB�( �TB� �GB� �MB�
�kB�Bz1 B�.�f� � )r1 r$ � )r2 r& � @)r3 r( � )r4 r* � )r5 r, )r" r- r )�nZ precisionZabbrevsZfactor�suffixr r r �humanize_bytesG s r8 c @ s e Zd ZdZdd� ZdS )�ExplicitNullAuthzwForces requests to ignore the ``.netrc``.
<https://github.com/psf/requests/issues/2773#issuecomment-174312831>
c C s |S )Nr )r �rr r r �__call__x s zExplicitNullAuth.__call__N)r r r r r; r r r r r9 s s r9 c C s t j| dd�d S )z�
Return the content type for ``filename`` in format appropriate
for Content-Type headers, or ``None`` if the file type is unknown
to ``mimetypes``.
F)�strictr )� mimetypesZ
guess_type)�filenamer r r �get_content_type| s r? c C s | sg S t j| �S )z�
When ``requests`` stores cookies in ``response.headers['Set-Cookie']``
it concatenates all of them through ``, ``.
This function splits cookies apart being careful to not to
split on ``, `` which may be part of cookie value.
)�RE_COOKIE_SPLIT�split)�cookiesr r r �
split_cookies� s rC )rB �nowr c s\ �p
t j � �tt td��fdd� � tt| ��}dd� |D �} t| �d� � fdd�| D �S )N)�expiresr c s | d k o| � kS )Nr )rE )rD r r �
is_expired� s z'get_expired_cookies.<locals>.is_expiredc S s( g | ] }t |d d� |d d d��qS )r" Nr )�name)�dict)�.0Zattrsr r r �
<listcomp>� s z'get_expired_cookies.<locals>.<listcomp>)rB rD c s2 g | ]*}� |j d �d�r|d |j dd�d��qS )rE )rE rG �path�/)rG rK )�get)rI �cookie)rF r r rJ � s )�timer �float�boolr rC �_max_age_to_expires)rB rD Z attr_setsr )rF rD r �get_expired_cookies� s
rS c C sB x<| D ]4}d|krq|j d�}|r|j� r|t|� |d< qW dS )z�
Translate `max-age` into `expires` for Requests to take it into account.
HACK/FIXME: <https://github.com/psf/requests/issues/5743>
rE zmax-ageN)rM �isdigitrP )rB rD rN Zmax_ager r r rR � s
rR c
C s� | j d�}|d j� |dd� }}i }d}xh|D ]`}|j� }|r2|d }}|jd�} | dkr�|d| � j|�}|| d d� j|�}|||j� < q2W ||fS ) zBorrowed from requests.�;r r" Nz"' T�=���)rA �strip�find�lower)
�header�tokensZ content_typeZparamsZ params_dictZitems_to_stripZparam�key�valueZindex_of_equalsr r r �parse_content_type_header� s
r_ )r! )N)%r r= �rer rO � collectionsr Zhttp.cookiejarr �pprintr Ztypingr r r r Z
requests.authZrequests�compiler@ �strZItemr r r rH r r8 ZauthZAuthBaser9 r? rC rP rS rR r_ r r r r �<module> s.
-
,