3
�Pf�' � @ s� d dl mZ d dlmZmZ yd dlmZ W n" ek
rN G dd� d�ZY nX yd dlmZ W n ek
r� ddl mZ Y nX ddl
m Z m Z m
Z
d d
gZe� ZG d d � d e�ZG d d
� d
e�Zd
S )� )�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� )�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__2 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__9 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__H 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__O 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__S 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 r/ V s
zRecentlyUsedContainer.clearc
C s | j � tt| j��S Q R X d S )N)r r. r r )r r r r �keys` s zRecentlyUsedContainer.keys)r N)r r r �__doc__r r r r# r) r* r+ r- r/ r1 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 �ZeZeZeZeZd d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z!e"d+d,� �Z#� Z$S ).r 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 )Nr1 Fc s s | ]\}}|j � |fV qd S )N)r: )�.0�k�vr r r � � s z(HTTPHeaderDict.__eq__..c s s | ]\}}|j � |fV qd S )N)r: )r> r? r@ r r r rA � s )r4 r �hasattr�type�dict�
itermerged)r �otherr r r �__eq__� s zHTTPHeaderDict.__eq__c C s | j |� S )N)rG )r rF 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 r0 )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)rJ )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_valsrI 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 r1 N) r& � TypeError�formatr4 r � iteritemsrP r rB r1 �items)r �argsr8 rF r! r; r( r r r r6 � 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: rJ rK )r r! rL rI r r r �getlist� s
zHTTPHeaderDict.getlistc C s dt | �jt| j� �f S )Nz%s(%s))rC r rD rE )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)rV r4 r. r r: )r rF r! r; r r r r5 s
zHTTPHeaderDict._copy_fromc C s t | �� }|j| � |S )N)rC r5 )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! rI r; r r r rS 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 rE % s
zHTTPHeaderDict.itermergedc C s t | j� �S )N)r. rS )r r r r rT + s zHTTPHeaderDict.itemsc C sn g }x`|j D ]V}|jd�r@|d \}}||d |j� f|d< q |jdd�\}}|j||j� f� q W | |�S ) z4Read headers from a Python 2 httplib message object.� � r z
�:)rY rZ �����r\ )r7 �
startswith�rstrip�splitrO �strip)�cls�messager7 �liner! r( r r r � from_httplib. s
zHTTPHeaderDict.from_httplib)N)%r r r r2 r r) r# r* r= rG rH r r r r �objectrK r+ r- r rM rP r6 rV Z
getheadersZgetallmatchingheadersZigetZget_allrW r5 rX rS rE rT � classmethodrd �
__classcell__r r )r9 r r e s<
N)Z
__future__r � collectionsr r Z threadingr � ImportErrorr Zpackages.ordered_dictZ packages.sixr r r �__all__re r$ r r r r r r � s J