
    g-                        d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
 g dZ G d de      Z G d deeef         Z G d d	e      Ze
eeeef   ee	eef      ef   Zy
)    )annotations)AnyIterableIteratorMappingMutableMappingProtocolTupleUnion)HeadersHeadersLikeMultipleValuesErrorc                  $     e Zd ZdZd fdZ xZS )r   zP
    Exception raised when :class:`Headers` has multiple values for a key.

    c                    t        | j                        dk(  rt        | j                  d         S t        |          S N   r   )lenargsreprsuper__str__)self	__class__s    N/var/www/openai/venv/lib/python3.12/site-packages/websockets/datastructures.pyr   zMultipleValuesError.__str__   s3    tyy>Q		!%%w      returnstr)__name__
__module____qualname____doc__r   __classcell__r   s   @r   r   r      s    
! !r   r   c                       e Zd ZdZddg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Zd fdZd dZd!dZ xZS )"r   a  
    Efficient data structure for manipulating HTTP headers.

    A :class:`list` of ``(name, values)`` is inefficient for lookups.

    A :class:`dict` doesn't suffice because header names are case-insensitive
    and multiple occurrences of headers with the same name are possible.

    :class:`Headers` stores HTTP headers in a hybrid data structure to provide
    efficient insertions and lookups while preserving the original data.

    In order to account for multiple values with minimal hassle,
    :class:`Headers` follows this logic:

    - When getting a header with ``headers[name]``:
        - if there's no value, :exc:`KeyError` is raised;
        - if there's exactly one value, it's returned;
        - if there's more than one value, :exc:`MultipleValuesError` is raised.

    - When setting a header with ``headers[name] = value``, the value is
      appended to the list of values for that header.

    - When deleting a header with ``del headers[name]``, all values for that
      header are removed (this is slow).

    Other methods for manipulating headers are consistent with this logic.

    As long as no header occurs multiple times, :class:`Headers` behaves like
    :class:`dict`, except keys are lower-cased to provide case-insensitivity.

    Two methods support manipulating multiple values explicitly:

    - :meth:`get_all` returns a list of all values for a header;
    - :meth:`raw_items` returns an iterator of ``(name, values)`` pairs.

    _dict_listc                D    i | _         g | _         | j                  |i | y N)r&   r'   update)r   r   kwargss      r   __init__zHeaders.__init__H   s#    +-
,.
T$V$r   c                L    dj                  d | j                  D              dz   S )N c              3  2   K   | ]  \  }}| d | d  yw)z: 
N ).0keyvalues      r   	<genexpr>z"Headers.__str__.<locals>.<genexpr>N   s"     Jze#bt,zs   r0   )joinr'   r   s    r   r   zHeaders.__str__M   s     wwJtzzJJVSSr   c                N    | j                   j                   d| j                  dS )N())r   r   r'   r7   s    r   __repr__zHeaders.__repr__P   s$    ..))*!DJJ>;;r   c                    | j                         }| j                  j                         |_        | j                  j                         |_        |S r)   )r   r&   copyr'   )r   r=   s     r   r=   zHeaders.copyS   s7    ~~ZZ__&
ZZ__&
r   c                4    t        |       j                         S r)   )r   encoder7   s    r   	serializezHeaders.serializeY   s    4y!!r   c                ^    t        |t              xr |j                         | j                  v S r)   )
isinstancer   lowerr&   r   r3   s     r   __contains__zHeaders.__contains___   s#    #s#A		tzz(AAr   c                ,    t        | j                        S r)   )iterr&   r7   s    r   __iter__zHeaders.__iter__b   s    DJJr   c                ,    t        | j                        S r)   )r   r&   r7   s    r   __len__zHeaders.__len__e   s    4::r   c                x    | j                   |j                            }t        |      dk(  r|d   S t        |      r   )r&   rC   r   r   r   r3   r4   s      r   __getitem__zHeaders.__getitem__j   s5    

399;'u:?8O%c**r   c                    | j                   j                  |j                         g       j                  |       | j                  j                  ||f       y r)   )r&   
setdefaultrC   appendr'   rL   s      r   __setitem__zHeaders.__setitem__q   s=    

ciik2.55e<

3,'r   c                    |j                         }| j                  j                  |       | j                  D cg c]  \  }}|j                         |k7  s||f c}}| _        y c c}}w r)   )rC   r&   __delitem__r'   )r   r3   	key_lowerkvs        r   rS   zHeaders.__delitem__u   sQ    IIK	

y))-NAqwwyI7Mq!fN
Ns   A'A'c                `    t        |t              st        S | j                  |j                  k(  S r)   )rB   r   NotImplementedr&   )r   others     r   __eq__zHeaders.__eq__{   s%    %)!!zzU[[((r   c                     i | _         g | _        y)z&
        Remove all headers.

        N)r&   r'   r7   s    r   clearzHeaders.clear   s    
 

r   c                H    t        d |D              }t        |   |i | y)zT
        Update from a :class:`Headers` instance and/or keyword arguments.

        c              3  `   K   | ]&  }t        |t              r|j                         n| ( y wr)   )rB   r   	raw_items)r2   args     r   r5   z!Headers.update.<locals>.<genexpr>   s(      
LPSz#w7CMMOS@Ds   ,.N)tupler   r*   )r   r   r+   r   s      r   r*   zHeaders.update   s/    
  
LP
 
 	''r   c                V    | j                   j                  |j                         g       S )z|
        Return the (possibly empty) list of all values for a header.

        Args:
            key: Header name.

        )r&   getrC   rD   s     r   get_allzHeaders.get_all   s     zz~~ciik2..r   c                ,    t        | j                        S )zO
        Return an iterator of all values as ``(name, value)`` pairs.

        )rG   r'   r7   s    r   r_   zHeaders.raw_items   s    
 DJJr   )r   r   r+   r   r   Noner   )r   r   )r   bytes)r3   objectr   bool)r   zIterator[str])r   intr3   r   r   r   )r3   r   r4   r   r   rf   )r3   r   r   rf   )rY   r   r   ri   )r   rf   )r3   r   r   z	list[str])r   zIterator[tuple[str, str]])r   r    r!   r"   	__slots__r,   r   r;   r=   r@   rE   rH   rJ   rM   rQ   rS   rZ   r\   r*   rd   r_   r#   r$   s   @r   r   r      sh    #J '"I%
T<"B 
+(O)
(/ r   r   c                       e Zd ZdZddZddZy)SupportsKeysAndGetItemz_
    Dict-like types with ``keys() -> str`` and ``__getitem__(key: str) -> str`` methods.

    c                     y r)   r1   r7   s    r   keyszSupportsKeysAndGetItem.keys   s    Sr   c                     y r)   r1   rD   s     r   rM   z"SupportsKeysAndGetItem.__getitem__   s    Cr   N)r   zIterable[str]rk   )r   r    r!   r"   rp   rM   r1   r   r   rn   rn      s    
 )/r   rn   N)
__future__r   typingr   r   r   r   r   r	   r
   r   __all__LookupErrorr   r   r   rn   r   r1   r   r   <module>rv      s    "	 	 	 <
!+ 
!D nS#X& D P0X 0 CHU38_	r   