�~mZ`6 � @ st d d l m Z d d l m Z d d l Z d d l m Z m Z e � Z Gd d � d e � Z
e
Z d d � Z e d
d g � a
Gd d � d � Z Gd
d � d � Z Gd d � d e e � Z Gd d � d e � Z Gd d � d e e � Z Gd d � d e � Z Gd d � d � Z Gd d � d e e j � Z Gd d � d e e j � Z Gd d � d e e j � Z d S) � )�array)�abcN� )� MultiMapping�MutableMultiMappingc @ sC e Z d Z d Z d Z d e j � d d d � Z d d � Z d S)
�istrzCase insensitive str.T� �strictc C sS t | d d � r | St | � t k r+ n t | � } | j � } t j | | � S)N� __is_istr__F)�getattr�type�str�title�__new__)�cls�val�encoding�errors� r �/_multidict_py.pyr s z istr.__new__c C s | S)Nr )�selfr r r r s z
istr.titleN) �__name__�
__module__� __qualname__�__doc__r
�sys�getdefaultencodingr r r r r r r s
r c C s% t | t � s t d � � | j j S)Nz&Parameter should be multidict or proxy)�
isinstance�_Base� TypeError�_impl�_version)Zmdr r r �
getversion$ s r" �Qc @ s. e Z d Z d Z d d � Z d d � Z d S) �_Impl�_itemsr! c C s g | _ | j � d S)N)r% � incr_version)r r r r �__init__0 s z_Impl.__init__c C s% t } | d d 7<| d | _ d S)Nr r )r! )r �vr r r r& 4 s z_Impl.incr_versionN)r% r! )r r r � __slots__r' r& r r r r r$ - s r$ c @ s� e Z d Z d d � Z e d d � Z e d d � Z d d � Z d d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d S) r c C s | S)Nr )r �keyr r r �_title= s z _Base._titlec sc | j | � � � f d d � | j j D� } | r8 | S| rO | t k rO | St d | � � d S)z-Return a list of all values matching the key.c s+ g | ]! \ } } } | � k r | � q Sr r )�.0�i�kr( )�identityr r �
<listcomp>C s z _Base.getall.<locals>.<listcomp>zKey not found: %rN)r+ r r% �_marker�KeyError)r r* �default�resr )r/ r �getall@ s z _Base.getallc C sc | j | � } x- | j j D] \ } } } | | k r | Sq W| t k rO | St d | � � d S)z!Get first value matching the key.zKey not found: %rN)r+ r r% r1 r2 )r r* r3 r/ r- r. r( r r r �getoneJ s z _Base.getonec C s
| j | � S)N)r6 )r r* r r r � __getitem__V s z_Base.__getitem__Nc C s | j | | � S)zVGet first value matching the key.
The method is alias for .getone().
)r6 )r r* r3 r r r �getY s z _Base.getc C s t | j � � S)N)�iter�keys)r r r r �__iter__` s z_Base.__iter__c C s t | j j � S)N)�lenr r% )r r r r �__len__c s z
_Base.__len__c C s
t | j � S)z+Return a new view of the dictionary's keys.)� _KeysViewr )r r r r r: f s z
_Base.keysc C s
t | j � S)zAReturn a new view of the dictionary's items *(key, value) pairs).)�
_ItemsViewr )r r r r �itemsj s z _Base.itemsc C s
t | j � S)z-Return a new view of the dictionary's values.)� _ValuesViewr )r r r r �valuesn s z _Base.valuesc C s t | t j � s t St | t � r� | j j } | j j } t | � t | � k rY d SxK t | | � D]: \ \ } } } \ } } } | | k s� | | k ri d Sqi Wd St | j j � t | � k r� d Sx< | j � D]. \ } }
| j
| t � } |
| k r� d Sq� Wd S)NFT) r r �Mapping�NotImplementedr r r% r< �zipr@ r8 r1 ) r �otherZlftZrhtZi1Zk2Zv1Zi2Zv2r. r( Znvr r r �__eq__r s$ . z _Base.__eq__c C sC | j | � } x- | j j D] \ } } } | | k r d Sq Wd S)NTF)r+ r r% )r r* r/ r- r. r( r r r � __contains__� s
z_Base.__contains__c C s8 d j d d � | j � D� � } d j | j j | � S)Nz, c s s' | ] \ } } d j | | � Vq d S)z
'{}': {!r}N)�format)r, r. r( r r r � <genexpr>� s z!_Base.__repr__.<locals>.<genexpr>z<{}({})>)�joinr@ rI � __class__r )r �bodyr r r �__repr__� s "z_Base.__repr__)r r r r+ r1 r5 r6 r7 r8 r; r= r: r@ rB rG rH rN r r r r r ; s
r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�MultiDictProxyc C s@ t | t t f � s0 t d j t | � � � � | j | _ d S)Nz:ctor requires MultiDict or MultiDictProxy instance, not {})r � MultiDictrO r rI r r )r �argr r r r' � s
zMultiDictProxy.__init__c C s t d j | j j � � � d S)Nzcan't pickle {} objects)r rI rL r )r r r r �
__reduce__� s zMultiDictProxy.__reduce__c C s t | j � � S)zReturn a copy of itself.)rP r@ )r r r r �copy� s zMultiDictProxy.copyN)r r r r' rR rS r r r r rO � s rO c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�CIMultiDictProxyc C s@ t | t t f � s0 t d j t | � � � � | j | _ d S)Nz>ctor requires CIMultiDict or CIMultiDictProxy instance, not {})r � CIMultiDictrT r rI r r )r rQ r r r r' � s
zCIMultiDictProxy.__init__c C s
| j � S)N)r )r r* r r r r+ � s zCIMultiDictProxy._titlec C s t | j � � S)zReturn a copy of itself.)rU r@ )r r r r rS � s zCIMultiDictProxy.copyN)r r r r' r+ rS r r r r rT � s rT c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d d � Z e d d � Z e Z e d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d S)(rP c O s/ t � | _ | j | | | j j | j � d S)N)r$ r �_extendrL r �
_extend_items)r �args�kwargsr r r r' � s zMultiDict.__init__c C s | j t | j � � f f S)N)rL �listr@ )r r r r rR � s zMultiDict.__reduce__c C s | S)Nr )r r* r r r r+ � s zMultiDict._titlec C s) t | t � r t | � St d � � d S)Nz8MultiDict keys should be either str or subclasses of str)r r
r )r r* r r r �_key� s
zMultiDict._keyc C sE | j | � } | j j j | | j | � | f � | j j � d S)N)r+ r r% �appendr[ r& )r r* �valuer/ r r r �add� s %z
MultiDict.addc C s | j } | | j � � S)zReturn a copy of itself.)rL r@ )r r r r r rS � s zMultiDict.copyc O s | j | | d | j � d S)zhExtend current MultiDict with more values.
This method must be used instead of update.
�extendN)rV rW )r rX rY r r r r_ � s zMultiDict.extendc s& t | � d k r0 t d j | t | � � � � | r� | d } t | d t t f � rh | j j } n� t | d � r� | j � } g } xi | D]a } t | � d k s� t d j | � � � | j
� j | d � � j | d � | d f � q� W| | � | � f d d � | j � D� � d S) Nr z1{} takes at most 1 positional argument ({} given)r r@ � z3{} takes either dict or list of (key, value) tuplesc s7 g | ]- \ } } � j | � � j | � | f � q Sr )r+ r[ )r, r* r] )r r r r0 � s z%MultiDict._extend.<locals>.<listcomp>)
r< r rI r rP rO r r% �hasattrr@ r\ r+ r[ )r rX rY �name�methodrQ r@ �itemr )r r rV � s*
zMultiDict._extendc C s. x' | D] \ } } } | j | | � q Wd S)N)r^ )r r@ r/ r* r] r r r rW � s zMultiDict._extend_itemsc C s! | j j j � | j j � d S)z Remove all items from MultiDict.N)r r% �clearr& )r r r r re � s zMultiDict.clearc C s | j | | � d S)N)�_replace)r r* r] r r r � __setitem__ s zMultiDict.__setitem__c C s� | j | � } | j j } d } xE t t | � d d d � D]' } | | d | k r> | | =d } q> W| s~ t | � � n
| j j � d S)NFr r T���rh )r+ r r% �ranger< r2 r& )r r* r/ r@ �foundr- r r r � __delitem__ s #
zMultiDict.__delitem__Nc C sS | j | � } x- | j j D] \ } } } | | k r | Sq W| j | | � | S)zAReturn value for key, set value to default if key is not present.)r+ r r% r^ )r r* r3 r/ r- r. r( r r r �
setdefault s zMultiDict.setdefaultc C s� | j | � } xl t t | j j � � D]R } | j j | d | k r( | j j | d } | j j | =| j j � | Sq( W| t k r� t | � � n | Sd S)z�Remove specified key and return the corresponding value.
If key is not found, d is returned if given, otherwise
KeyError is raised.
r r` N)r+ ri r< r r% r& r1 r2 )r r* r3 r/ r- r] r r r �popone s
zMultiDict.poponec C s� d } | j | � } g } x{ t t | j j � d d d � D]W } | j j | } | d | k r> | j | d � | j j | =| j j � d } q> W| s� | t k r� t | � � q� | Sn | j � | Sd S) z�Remove all occurrences of key and return the list of corresponding
values.
If key is not found, default is returned if given, otherwise
KeyError is raised.
Fr r r` TNrh rh )
r+ ri r< r r% r\ r& r1 r2 �reverse)r r* r3 rj r/ Zretr- rd r r r �popall2 s )
zMultiDict.popallc C sP | j j r@ | j j j d � } | j j � | d | d f St d � � d S)z1Remove and return an arbitrary (key, value) pair.r r r` zempty multidictN)r r% �popr& r2 )r r- r r r �popitemM s
zMultiDict.popitemc O s | j | | d | j � d S)z>Update the dictionary from *other*, overwriting existing keys.�updateN)rV �
_update_items)r rX rY r r r rr V s zMultiDict.updatec
C sz | s
d Si } x� | D]� \ } } } | j | d � } x� t | t | j j � � D]N } | j j | } | d | k rT | d | | <| | | f | j j | <PqT W| j j j | | | f � t | j j � | | <q Wd } x� | t | j j � k rh| j j | } | d } | j | � } | d k r?| d 7} q� | | k r[| j j | =q� | d 7} q� W| j j � d S)Nr r )r8 ri r< r r% r\ r& )
r r@ Z used_keysr/ r* r] �startr- rd �posr r r rs Z s2 "
zMultiDict._update_itemsc C s | j | � } | j | � } | j j } x� t t | � d d d � D]G } | | } | d | k rG | | | f | | <| } | j j � PqG W| j j j | | | f � | j j � d Sd } xH | | k r| | } | d | k r| | =| d 8} q� | d 7} q� Wd S)Nr r rh rh )r[ r+ r r% ri r< r& r\ )r r* r] r/ r@ r- rd Zrgtr r r rf z s( #
zMultiDict._replace)r r r r' rR r+ r[ r^ rS r_ rV rW re rg rk rl r1 rm rp ro rq rr rs rf r r r r rP � s(
rP c @ s e Z d Z d d � Z d S)rU c C s
| j � S)N)r )r r* r r r r+ � s zCIMultiDict._titleN)r r r r+ r r r r rU � s rU c @ s( e Z d Z d d � Z d d � Z d S)� _ViewBasec C s | | _ | j | _ d S)N)r r! )r �implr r r r' � s z_ViewBase.__init__c C s t | j j � S)N)r< r r% )r r r r r= � s z_ViewBase.__len__N)r r r r' r= r r r r rv � s rv c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)r? c C s� t | t � s$ t | t � s$ t � t | � d k s< t � xA | j j D]3 \ } } } | d | k rI | d | k rI d SqI Wd S)Nr` r r TF)r �tuplerZ �AssertionErrorr< r r% )r rd r- r. r( r r r rH � s $ z_ItemsView.__contains__c c sP xI | j j D]; \ } } } | j | j j k r= t d � � | | f Vq
Wd S)Nz#Dictionary changed during iteration)r r% r! � RuntimeError)r r- r. r( r r r r; � s z_ItemsView.__iter__c C sc g } x5 | j j D]' } | j d j | d | d � � q Wd j | � } d j | j j | � S)Nz
{!r}: {!r}r r` z, z{}({}))r r% r\ rI rK rL r )r �lstrd rM r r r rN � s
%z_ItemsView.__repr__N)r r r rH r; rN r r r r r? � s r? c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)rA c C s/ x( | j j D] } | d | k r
d Sq
Wd S)Nr` TF)r r% )r r] rd r r r rH � s z_ValuesView.__contains__c c sE x> | j j D]0 } | j | j j k r4 t d � � | d Vq
Wd S)Nz#Dictionary changed during iterationr` )r r% r! rz )r rd r r r r; � s z_ValuesView.__iter__c C s\ g } x. | j j D] } | j d j | d � � q Wd j | � } d j | j j | � S)Nz{!r}r` z, z{}({}))r r% r\ rI rK rL r )r r{ rd rM r r r rN � s
z_ValuesView.__repr__N)r r r rH r; rN r r r r rA � s rA c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)r> c C s/ x( | j j D] } | d | k r
d Sq
Wd S)Nr TF)r r% )r r* rd r r r rH � s z_KeysView.__contains__c c sE x> | j j D]0 } | j | j j k r4 t d � � | d Vq
Wd S)Nz#Dictionary changed during iterationr )r r% r! rz )r rd r r r r; � s z_KeysView.__iter__c C s\ g } x. | j j D] } | j d j | d � � q Wd j | � } d j | j j | � S)Nz{!r}r z, z{}({}))r r% r\ rI rK rL r )r r{ rd rM r r r rN � s
z_KeysView.__repr__N)r r r rH r; rN r r r r r> � s r> )r � collectionsr r Z_abcr r �objectr1 r
r Zupstrr" r! r$ r rO rT rP rU rv � ItemsViewr? �
ValuesViewrA �KeysViewr> r r r r �<module> s$ W�