o
    nkj                     @  s   d dl mZ d dlZd dlmZmZmZmZ d dlm	Z	m
Z
 g dZG dd deZedjZG d	d
 d
eeef ZG dd de
Zeeeef B eeeef  B eB Z	 ddlmZ dS )    )annotationsN)IterableIteratorMappingMutableMapping)AnyProtocol)HeadersHeadersLikeMultipleValuesErrorc                      s"   e Zd ZdZd fddZ  ZS )r   zP
    Exception raised when :class:`Headers` has multiple values for a key.

    returnstrc                   s&   t | jdkrt| jd S t  S N   r   )lenargsreprsuper__str__self	__class__ W/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/websockets/datastructures.pyr      s   
zMultipleValuesError.__str__r   r   )__name__
__module____qualname____doc__r   __classcell__r   r   r   r   r      s    r   z[\x09\x20-\x7e\x80-\xff]*c                      s   e Zd ZdZddgZd7d
dZd8ddZd8ddZd9ddZd:ddZ	d;ddZ
d<ddZd=ddZd>d d!Zd?d#d$Zd@d%d&ZdAd)d*ZdBd+d,Zd7 fd-d.ZdCd0d1ZdDd3d4Zd?d5d6Z  ZS )Er	   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_listr   r
   kwargsr   r   Nonec                 O  s    i | _ g | _| j|i | d S N)r!   r"   updater   r   r#   r   r   r   __init__I   s   zHeaders.__init__c                 C  s   d dd | jD d S )N c                 s  s$    | ]\}}| d | dV  qdS )z: 
Nr   ).0keyvaluer   r   r   	<genexpr>O   s   " z"Headers.__str__.<locals>.<genexpr>r*   )joinr"   r   r   r   r   r   N   s   zHeaders.__str__c                 C  s   | j j d| jdS )N())r   r   r"   r   r   r   r   __repr__Q   s   zHeaders.__repr__c                 C  s$   |   }| j |_| j |_|S r%   )r   r!   copyr"   )r   r3   r   r   r   r3   T   s   zHeaders.copybytesc                 C  s   t |  S r%   )r   encoder   r   r   r   	serializeZ   s   zHeaders.serializer,   objectboolc                 C  s   t |to| | jv S r%   )
isinstancer   lowerr!   r   r,   r   r   r   __contains__`   s   zHeaders.__contains__Iterator[str]c                 C  
   t | jS r%   )iterr!   r   r   r   r   __iter__c      
zHeaders.__iter__intc                 C  r>   r%   )r   r!   r   r   r   r   __len__f   rA   zHeaders.__len__c                 C  s*   | j |  }t|dkr|d S t|r   )r!   r:   r   r   r   r,   r-   r   r   r   __getitem__k   s   zHeaders.__getitem__r-   c                 C  sB   t t|st||| j| g | | j||f d S r%   )is_valid_header_valuer   InvalidHeaderValuer!   
setdefaultr:   appendr"   rD   r   r   r   __setitem__r   s   
zHeaders.__setitem__c                   s.   |   | j   fdd| jD | _d S )Nc                   s$   g | ]\}}|   kr||fqS r   )r:   )r+   kv	key_lowerr   r   
<listcomp>|   s   $ z'Headers.__delitem__.<locals>.<listcomp>)r:   r!   __delitem__r"   r;   r   rM   r   rP   x   s   zHeaders.__delitem__otherr   c                 C  s   t |tstS | j|jkS r%   )r9   r	   NotImplementedr!   )r   rQ   r   r   r   __eq__~   s   
zHeaders.__eq__c                 C  s   i | _ g | _dS )z&
        Remove all headers.

        N)r!   r"   r   r   r   r   clear   s   
zHeaders.clearc                   s(   t dd |D }t j|i | dS )zT
        Update from a :class:`Headers` instance and/or keyword arguments.

        c                 s  s&    | ]}t |tr| n|V  qd S r%   )r9   r	   	raw_items)r+   argr   r   r   r.      s    
z!Headers.update.<locals>.<genexpr>N)tupler   r&   r'   r   r   r   r&      s   zHeaders.update	list[str]c                 C  s   | j | g S )z|
        Return the (possibly empty) list of all values for a header.

        Args:
            key: Header name.

        )r!   getr:   r;   r   r   r   get_all   s   zHeaders.get_allIterator[tuple[str, str]]c                 C  r>   )zO
        Return an iterator of all values as ``(name, value)`` pairs.

        )r?   r"   r   r   r   r   rU      s   
zHeaders.raw_itemsc                 C  s,   | j | g | | j||f dS )z=
        Set a header without validating its value.

        N)r!   rH   r:   rI   r"   rD   r   r   r   set_insecure   s   zHeaders.set_insecure)r   r
   r#   r   r   r$   r   )r   r	   )r   r4   )r,   r7   r   r8   )r   r=   )r   rB   r,   r   r   r   )r,   r   r-   r   r   r$   )r,   r   r   r$   )rQ   r   r   r8   )r   r$   )r,   r   r   rX   )r   r[   )r   r   r   r   	__slots__r(   r   r2   r3   r6   r<   r@   rC   rE   rJ   rP   rS   rT   r&   rZ   rU   r\   r    r   r   r   r   r	       s(    %















	r	   c                   @  s$   e Zd ZdZdddZddd	Zd
S )SupportsKeysAndGetItemz_
    Dict-like types with ``keys() -> str`` and ``__getitem__(key: str) -> str`` methods.

    r   Iterable[str]c                 C     d S r%   r   r   r   r   r   keys       zSupportsKeysAndGetItem.keysr,   r   c                 C  ra   r%   r   r;   r   r   r   rE      rc   z"SupportsKeysAndGetItem.__getitem__N)r   r`   r]   )r   r   r   r   rb   rE   r   r   r   r   r_      s    
r_   r   )rG   )
__future__r   recollections.abcr   r   r   r   typingr   r   __all__LookupErrorr   compile	fullmatchrF   r   r	   r_   rW   r
   
exceptionsrG   r   r   r   r   <module>   s     "
