File: //opt/saltstack/salt/lib/python3.10/site-packages/cherrypy/lib/__pycache__/encoding.cpython-310.pyc
o
�N�g�B � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z dZ
dZ dd d
�Z G d d � d �Z
G d
d� d�Zdd� Zdd� Zdd� Zdddgdfdd�ZdS ) � N)�
text_or_bytes)�file_generator)�is_closable_iterator)�set_vary_header� � �utf-8c C sR t jj}| durt| t�s| g} | |_dS |r't|t�s |g}|j| |_dS dS )a; Replace or extend the list of charsets used to decode a request entity.
Either argument may be a single string or a list of strings.
encoding
If not None, restricts the set of charsets attempted while decoding
a request entity to the given set (even if a different charset is
given in the Content-Type request header).
default_encoding
Only in effect if the 'encoding' argument is not given.
If given, the set of charsets attempted while decoding a request
entity is *extended* with the given value(s).
N)�cherrypy�request�body�
isinstance�listZattempt_charsets)�encoding�default_encodingr � r �I/opt/saltstack/salt/lib/python3.10/site-packages/cherrypy/lib/encoding.py�decode s
�r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�UTF8StreamEncoderc C s
|| _ d S �N)� _iterator)�self�iteratorr r r �__init__, s
zUTF8StreamEncoder.__init__c C s | S r r �r r r r �__iter__/ s zUTF8StreamEncoder.__iter__c C s | � � S r )�__next__r r r r �next2 s zUTF8StreamEncoder.nextc C s" t | j�}t|t�r|�d�}|S �Nr )r r r �str�encode)r �resr r r r 5 s
zUTF8StreamEncoder.__next__c C s t | j�r | j�� d S d S r )r r �closer r r r r! ; s
�zUTF8StreamEncoder.closec C s |� d�r
t| |��t| j|�S )N�__)�
startswith�AttributeError�getattrr )r �attrr r r � __getattr__? s
z UTF8StreamEncoder.__getattr__N) �__name__�
__module__� __qualname__r r r r r! r' r r r r r + s r c @ sP e Zd ZdZdZdZdZdZdZdZ dd� Z
d d
� Z d d � Z d
d� Z
dd� ZdS )�ResponseEncoderr z+Response body could not be encoded with %r.N�strictTFc K s` |� � D ]
\}}t| ||� qt� | _tjj}|jd ur.| jr%t� dd� |j| _
| |_d S d S )NzReplacing request.handler� TOOLS.ENCODE) �items�setattr�set�attempted_charsetsr �servingr
�handler�debug�log�
oldhandler)r �kwargs�k�vr
r r r r O s
�zResponseEncoder.__init__c s8 � �j v rdS �j �� � � �fdd�}|�j��_dS )z�Encode a streaming response body.
Use a generator wrapper, and just pray it works as the stream is
being written out.
Fc 3 s. � | D ]}t |t�r|�� �j�}|V qd S r )r r r �errors)r �chunk�r r r r �encoderf s �
�z.ResponseEncoder.encode_stream.<locals>.encoderT)r1 �addr )r r r= r r< r �
encode_stream\ s
z ResponseEncoder.encode_streamc
C sv || j v rdS | j �|� g }| jD ]#}t|t�r0z |�|| j�}W n
ttfy/ Y dS w |� |� q|| _dS )z Encode a buffered response body.FT)
r1 r> r r r r r: � LookupError� UnicodeError�append)r r r r; r r r �
encode_stringn s
� z ResponseEncoder.encode_stringc C s t jj}t jj}| jrt �d|j d� |jr| j}n | j}d|j v r'|j d= |j �
d�}dd� |D �}| jrAt �dt |� d� | j d urr| j �
� }| jrVt �d| d� |r`d |v s`||v rq| jrkt �d
| d� ||�rq|S nt|s�| jr�t �d | j d� || j�r�| jS t �d | j| j ��|D ]8}|jd
kr�|jd kr�| jr�t �d| d� || j�r�| j S q�|j}| jr�t �d| d� ||�r�| S q�d |vr�d}||vr�| jr�t �dd� ||�r�|S |j �d�} | d u r�d}
nd| }
d�t| j��} |
d| f 7 }
t �d|
��)Nzresponse.stream %rr- �Content-LengthzAccept-Charsetc S s g | ]}|j �� �qS r )�value�lower)�.0�encr r r �
<listcomp>� s z;ResponseEncoder.find_acceptable_charset.<locals>.<listcomp>z charsets %szSpecified encoding %r�*zAttempting encoding %rz Attempting default encoding %ri� r z%Attempting default encoding due to %rz"Attempting encoding %s (qvalue >0)z
iso-8859-1z Attempting ISO-8859-1 encodingz2Your client did not send an Accept-Charset header.z0Your client sent this Accept-Charset header: %s.z, z We tried these charsets: %s.� )r r2 r
�responser4 r5 �streamr? rC �headers�elements�reprr rF r � HTTPError�failmsg�qvaluerE �get�join�sortedr1 ) r r
rL r= ZencsZcharsetsr �elementZisoZac�msgZ _charsetsr r r �find_acceptable_charset~ s� ��
�������
�
��
����� z'ResponseEncoder.find_acceptable_charsetc O s t jj}| j|i |��| _t| j�| _|j�d�}| jr)t � ddd� |D � d� |r| j
r|d }| j rY|j �
� �d�rK| jrHt � d| d� d }n| jrVt � d
| d� d }n | jrbt � d d� d }|r| �� |jd
< | jrxt � d| d� t|�|jd< | jS )N� Content-TypezContent-Type: %rc S s g | ]}t |��qS r )r )rG �hr r r rI � s �z,ResponseEncoder.__call__.<locals>.<listcomp>r- r ztext/z#Content-Type %s starts with "text/"Tz?Not finding because Content-Type %s does not start with "text/"Fz Finding because not text_only�charsetzSetting Content-Type %s)r r2 rL r6 r � prepare_iterrN rO r4 r5 � add_charset� text_onlyrE rF r# rY �paramsr )r �argsr7 rL �ctZdo_findr r r �__call__� sP ��
����
�zResponseEncoder.__call__)r( r) r* r rR r r: r_ r^ r4 r r? rC rY rc r r r r r+ E s
_r+ c C sF t | t�r| r | g} | S g } | S t| d�rt| �} | S | du r!g } | S )zL
Ensure response body is iterable and resolves to False when empty.
�readN)r r �hasattrr )rE r r r r] s
�
��r] c c s� � ddl }dV dV dV t�dtt�� �tdd�@ �V |tkr&d V n |tkr.d
V ndV d V |�d �}d}|�||j |j
|j d�}| D ]}|t |�7 }|�||�}|�
|�V qJ|�� V t�d|tdd�@ �V t�d|tdd�@ �V dS )
z,Compress 'body' at the given compress_level.r Ns �� � z<LZFFFFFFFF� � � � �� )�zlib�struct�pack�int�time�_COMPRESSION_LEVEL_BEST�_COMPRESSION_LEVEL_FAST�crc32� compressobj�DEFLATED� MAX_WBITS�
DEF_MEM_LEVEL�len�compress�flush)r �compress_levelrm �crc�sizeZzobj�liner r r rz s2 �
�
rz c C sF dd l }t�� }|�| � |�d� |jd|d�}|�� }|�� |S )Nr �rb)�mode�fileobj)�gzip�io�BytesIO�write�seek�GzipFilerd r! )r r� ZzbufZzfile�datar r r �
decompressN s
r� � z text/htmlz
text/plainFc C s. t jj}t jj}t|d� |js|rt jddd� dS t|dd�r,|r*t jddd� dS |j� d�}|s?|r=t jd dd� dS |j�
d
d �� d �d
}|D ]�}|j dkrg|j
d
krg|rdt jd| dd� dS |j dv �r|j
d
kr�|r}t jd| dd� dS ||vr�d}d|v r�|� d�\} }
|D ];} d| v r�| � d�\} }
| | kr�|
dkr�d} n#d|
v r�d|
v r�|
� d�\}}|
� d�\}}|dkr�||kr�d} nq�|s�|r�t jd||f dd� dS |r�t jddd� d|jd< t|j| �|_d|jv r�|jd= dS qM|�r
t jdd d� t �d d ��� dS ) a� Try to gzip the response body if Content-Type in mime_types.
cherrypy.response.headers['Content-Type'] must be set to one of the
values in the mime_types arg before calling this function.
The provided list of mime-types must be of one of the following form:
* `type/subtype`
* `type/*`
* `type/*+subtype`
No compression is performed if any of the following hold:
* The client sends no Accept-Encoding request header
* No 'gzip' or 'x-gzip' is present in the Accept-Encoding header
* No 'gzip' or 'x-gzip' with a qvalue > 0 is present
* The 'identity' value is given with a qvalue > 0.
zAccept-EncodingzNo response bodyz
TOOLS.GZIP)�contextN�cachedFz Not gzipping cached responsezNo Accept-EncodingrZ � �;r �identityz Non-zero identity qvalue: %s)r� zx-gzipzZero gzip qvalue: %s�/rJ T�+z$Content-Type %s not in mime_types %rZGzippingr� zContent-EncodingrD z No acceptable encoding found.ZGZIPrK zidentity, gzip)r r2 r
rL r r r5 r% rN rO rT �splitrE rS rz rQ Z set_response)r| Z
mime_typesr4 r
rL Z
acceptablerb Zcoding�foundZ
ct_media_typeZ ct_sub_typeZ mime_typeZ
media_typeZsub_typeZct_leftZct_right�left�rightr r r r� Z s�
�
����
�1r� r )rn rq r� r Zcherrypy._cpcompatr Z cherrypy.libr r r rs rr r r r+ r] rz r� r� r r r r �<module> s(
C. �