3
��yi+, � @ s� d dl mZ yd dlmZmZ W n$ ek
rD d dlmZmZ Y nX yd dlmZ W n" ek
rx G dd� d�ZY nX d dlm Z ddl
m Z dd l m
Z
mZmZ d
d gZe� ZG d d
� d
e�ZG d
d � d e�ZdS )� )�absolute_import)�Mapping�MutableMapping)�RLockc @ s e Zd Zdd� Zdd� ZdS )r c C s d S )N� )�selfr r �"/usr/lib/python3.6/_collections.py� __enter__
s zRLock.__enter__c C s d S )Nr )r �exc_type� exc_value� tracebackr r r �__exit__
s zRLock.__exit__N)�__name__�
__module__� __qualname__r r
r r r r r s r )� OrderedDict� )�
InvalidHeader)�iterkeys�
itervalues�PY3�RecentlyUsedContainer�HTTPHeaderDictc @ sV e Zd ZdZeZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )r a�
Provides a thread-safe dict-like container which maintains up to
``maxsize`` keys while throwing away the least-recently-used keys beyond
``maxsize``.
:param maxsize:
Maximum number of recent elements to retain.
:param dispose_func:
Every time an item is evicted from the container,
``dispose_func(value)`` is called. Callback which will get called
�
Nc C s" || _ || _| j� | _t� | _d S )N)�_maxsize� dispose_func� ContainerCls�
_containerr �lock)r �maxsizer r r r �__init__, s
z RecentlyUsedContainer.__init__c
C s, | j � | jj|�}|| j|< |S Q R X d S )N)r r �pop)r �key�itemr r r � __getitem__3 s
z!RecentlyUsedContainer.__getitem__c
C sl t }| j�@ | jj|t �}|| j|< t| j�| jkrF| jjdd�\}}W d Q R X | jrh|t k rh| j|� d S )NF)Zlast)�_Nullr r �get�lenr �popitemr )r r" �valueZ
evicted_valueZ_keyr r r � __setitem__: s
z!RecentlyUsedContainer.__setitem__c C s2 | j � | jj|�}W d Q R X | jr.| j|� d S )N)r r r! r )r r" r) r r r � __delitem__I s z!RecentlyUsedContainer.__delitem__c C s | j � t| j�S Q R X d S )N)r r' r )r r r r �__len__P s z RecentlyUsedContainer.__len__c C s t d��d S )Nz7Iteration over this class is unlikely to be threadsafe.)�NotImplementedError)r r r r �__iter__T s z RecentlyUsedContainer.__iter__c
C sL | j � tt| j��}| jj� W d Q R X | jrHx|D ]}| j|� q6W d S )N)r �listr r �clearr )r �valuesr) r r r r0 W s
zRecentlyUsedContainer.clearc
C s | j � tt| j��S Q R X d S )N)r r/ r r )r r r r �keysa s zRecentlyUsedContainer.keys)r N)r r r �__doc__r r r r$ r* r+ r, r. r0 r2 r r r r r s
c s� e Zd ZdZd/� fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e sZe j
Z
e jZe� Zdd� Zdd� Zefdd�Zdd� Zdd� Zdd � Zefd d �Zd d � ZeZeZeZeZ d!d"� Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z!d+d,� Z"e#d-d.� �Z$� Z%S )0r ap
:param headers:
An iterable of field-value pairs. Must not contain multiple field names
when compared case-insensitively.
:param kwargs:
Additional field-value pairs to pass in to ``dict.update``.
A ``dict`` like container for storing HTTP Headers.
Field names are stored and compared case-insensitively in compliance with
RFC 7230. Iteration provides the first case-sensitive key seen for each
case-insensitive pair.
Using ``__setitem__`` syntax overwrites fields that compare equal
case-insensitively in order to maintain ``dict``'s api. For fields that
compare equal, instead create a new ``HTTPHeaderDict`` and use ``.add``
in a loop.
If multiple fields that are equal case-insensitively are passed to the
constructor or ``.update``, the behavior is undefined and some will be
lost.
>>> headers = HTTPHeaderDict()
>>> headers.add('Set-Cookie', 'foo=bar')
>>> headers.add('set-cookie', 'baz=quxx')
>>> headers['content-length'] = '7'
>>> headers['SET-cookie']
'foo=bar, baz=quxx'
>>> headers['Content-Length']
'7'
Nc sP t t| �j� t� | _|d k r>t|t�r4| j|� n
| j|� |rL| j|� d S )N)�superr r r r �
isinstance�
_copy_from�extend)r �headers�kwargs)� __class__r r r � s
zHTTPHeaderDict.__init__c C s ||g| j |j� < | j |j� S )N)r �lower)r r" �valr r r r* � s zHTTPHeaderDict.__setitem__c C s | j |j� }dj|dd � �S )Nz, r )r r; �join)r r" r< r r r r$ � s zHTTPHeaderDict.__getitem__c C s | j |j� = d S )N)r r; )r r" r r r r+ � s zHTTPHeaderDict.__delitem__c C s |j � | jkS )N)r; r )r r" r r r � __contains__� s zHTTPHeaderDict.__contains__c C sb t |t� r t|d� r dS t |t| ��s6t| �|�}tdd� | j� D ��tdd� |j� D ��kS )Nr2 Fc s s | ]\}}|j � |fV qd S )N)r; )�.0�k�vr r r � <genexpr>� s z(HTTPHeaderDict.__eq__.<locals>.<genexpr>c s s | ]\}}|j � |fV qd S )N)r; )r? r@ rA r r r rB � s )r5 r �hasattr�type�dict�
itermerged)r �otherr r r �__eq__� s zHTTPHeaderDict.__eq__c C s | j |� S )N)rH )r rG r r r �__ne__� s zHTTPHeaderDict.__ne__c C s
t | j�S )N)r' r )r r r r r, � s zHTTPHeaderDict.__len__c c s" x | j j� D ]}|d V q W d S )Nr )r r1 )r �valsr r r r. � s zHTTPHeaderDict.__iter__c C s< y | | }W n t k
r, || jkr(� |S X | |= |S dS )z�D.pop(k[,d]) -> v, remove specified key and return the corresponding value.
If key is not found, d is returned if given, otherwise KeyError is raised.
N)�KeyError�_HTTPHeaderDict__marker)r r" �defaultr) r r r r! � s
zHTTPHeaderDict.popc C s$ y
| |= W n t k
r Y nX d S )N)rK )r r" r r r �discard� s
zHTTPHeaderDict.discardc C s4 |j � }||g}| jj||�}||k r0|j|� dS )z�Adds a (name, value) pair, doesn't overwrite the value if it already
exists.
>>> headers = HTTPHeaderDict(foo='bar')
>>> headers.add('Foo', 'baz')
>>> headers['foo']
'bar, baz'
N)r; r �
setdefault�append)r r" r< Z key_lowerZnew_valsrJ r r r �add� s
zHTTPHeaderDict.addc O s t |�dkr tdjt |����t |�dkr2|d nf }t|t�rdx�|j� D ]\}}| j||� qJW nvt|t�r�xj|D ]}| j||| � qtW nLt|d�r�x@|j � D ]}| j||| � q�W n x |D ]\}}| j||� q�W x |j
� D ]\}}| j||� q�W dS )z�Generic import function for any type of header-like object.
Adapted version of MutableMapping.update in order to insert items
with self.add instead of self.__setitem__
r z9extend() takes at most 1 positional arguments ({0} given)r r2 N) r' � TypeError�formatr5 r � iteritemsrQ r rC r2 �items)r �argsr9 rG r" r< r) r r r r7 � s"
zHTTPHeaderDict.extendc C sF y| j |j� }W n" tk
r4 || jkr0g S |S X |dd� S dS )zmReturns a list of all the values for the named field. Returns an
empty list if the key doesn't exist.r N)r r; rK rL )r r" rM rJ r r r �getlist� s
zHTTPHeaderDict.getlistc C s. dddddddg}x|D ]}| j |� qW | S )z�
Remove content-specific header fields before changing the request
method to GET or HEAD according to RFC 9110, Section 15.4.
zContent-EncodingzContent-LanguagezContent-Locationz Content-TypezContent-LengthZDigestz
Last-Modified)rN )r Zcontent_specific_headers�headerr r r �_prepare_for_method_change s
z)HTTPHeaderDict._prepare_for_method_changec C s dt | �jt| j� �f S )Nz%s(%s))rD r rE rF )r r r r �__repr__! s zHTTPHeaderDict.__repr__c C sB x<|D ]4}|j |�}t|t�r&t|�}|g| | j|j� < qW d S )N)rW r5 r/ r r; )r rG r" r< r r r r6 $ s
zHTTPHeaderDict._copy_fromc C s t | �� }|j| � |S )N)rD r6 )r Zcloner r r �copy, s
zHTTPHeaderDict.copyc c sD x>| D ]6}| j |j� }x"|dd� D ]}|d |fV q&W qW dS )z8Iterate over all header lines, including duplicate ones.r Nr )r r; )r r" rJ r< r r r rT 1 s
zHTTPHeaderDict.iteritemsc c s< x6| D ].}| j |j� }|d dj|dd� �fV qW dS )z:Iterate over all headers, merging duplicate ones together.r z, r N)r r; r= )r r" r< r r r rF 8 s
zHTTPHeaderDict.itermergedc C s t | j� �S )N)r/ rT )r r r r rU > s zHTTPHeaderDict.itemsc C s� d}g }xr|j D ]h}|j|�rV|s0td| ��n&|d \}}||d |j� f|d< q|jdd�\}}|j||j� f� qW | |�S ) z4Read headers from a Python 2 httplib message object.� � z/Header continuation with no previous header: %sr �:)r\ r] ���r_ )r8 �
startswithr �strip�splitrP )�cls�messageZobs_fold_continued_leadersr8 �liner" r) r r r � from_httplibA s
zHTTPHeaderDict.from_httplib)N)&r r r r3 r r* r$ r+ r> rH rI r r r r �objectrL r, r. r! rN rQ r7 rW rY Z
getheadersZgetallmatchingheadersZigetZget_allrZ r6 r[ rT rF rU � classmethodrf �
__classcell__r r )r: r r f s> N)Z
__future__r �collections.abcr r � ImportError� collectionsZ threadingr r �
exceptionsr Z packages.sixr r r �__all__rg r% r r r r r r �<module> s J