U
'�7`A# � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZ ddlmZ ddlmZmZmZmZ ddlmZ d Z er�dd
l m Z e e
e ge d f Z e!ej"�#d ��Z$G d d
� d
e�Z%dS )� N) �IO�
TYPE_CHECKING�Any� Awaitable�Callable�List�Optional�Union�cast� )�hdrs)�AbstractStreamWriter)� LooseHeaders)�HTTPNotModified�HTTPPartialContent�HTTPPreconditionFailed� HTTPRequestRangeNotSatisfiable)�StreamResponse)� FileResponse)� BaseRequestZAIOHTTP_NOSENDFILEc s� e Zd ZdZdeeejf eee e e e
dd�� fdd�
Z e e
e eee d�d d
�Zd e
e eee d �� fd
d� Zd e e d�� fdd� Z� ZS )r z,A response object can be used to send files.� �� N)�path�
chunk_size�status�reason�headers�returnc s6 t � j|||d� t|t�r&t�|�}|| _|| _d S )N)r r r )�super�__init__�
isinstance�str�pathlib�Path�_path� _chunk_size)�selfr r r r r �� __class__� �K/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/web_fileresponse.pyr , s
zFileResponse.__init__)�writer�fobj�offset�countr c � s� | j }t�� }|�d |j|�I d H |�d |j|�I d H }|r~|�|�I d H || }|dkr`q~|�d |jt||��I d H }q:|�� I d H |S )Nr ) r% �asyncio�get_event_loop�run_in_executor�seek�read�write�minZdrain)r& r+ r, r- r. r �loop�chunkr) r) r* �_sendfile_fallback< s z FileResponse._sendfile_fallbackr )�requestr, r- r. r c � s� t � �|�I d H }|d k s t�ts2tjdk s2| jrH| �||||�I d H S |j}|j }|d k s`t�z|�
||||�I d H W n* t k
r� | �||||�I d H Y S X t � � � I d H |S )N)� � )
r �prepare�AssertionError�
NOSENDFILE�sys� version_infoZ compressionr8 Z_loop� transport�sendfile�NotImplementedErrorZ write_eof)r& r9 r, r- r. r+ r6 rA r' r) r* � _sendfileR s zFileResponse._sendfile)r9 r c
� s\ | j }d}d|j�tjd�kr>|�|jd �}|�� r>|}d}t� � }|�
d |j �I d H }|j }|d k r�|j
|�� kr�| �tj� d| _t� �|�I d H S |j}|d k r�|j
|�� kr�| �tj� t� �|�I d H S tj| jk�rt�t|��\} }
| s�d} d} n|�r dnd }
d} | j} |j }
|
}d }|j }|d k�sH|j
|�� k�rTz|j }|j }|j }W nF t!k
�r� d|
� �| jtj"< | �t#j� t� �|�I d H Y S X |d k �s�|d k �rT|dk �r�|d k�r�||
7 }|dk �r�d}|
| }n t$|d k �r�|n|
|
�| }||
k�rDd|
� �| jtj"< | �t#j� t� �|�I d H S t%j} | �| � | �r`| | _&|
�rr|
| jtj'< |�r�tj| jtj(< |j
| _)|| _*d | jtj+< t,t-|�}| t%jk�r�d
�.||| d |
�| jtj"< |j/tj0k�s�| j1d k�r t� �|�I d H S |�
d |j2d
�I d H }|�r"|}nd}z| �4||||�I d H W �S |�
d |j3�I d H X d S )NF�gzip� z.gzTzapplication/octet-streamzbytes */r �byteszbytes {}-{}/{}r )�� i0 �rb)5r$ r �getr ZACCEPT_ENCODINGZ with_name�name�is_filer/ r0 r1 �statZif_modified_since�st_mtimeZ timestampZ
set_statusr Z status_codeZ
_length_checkr r< Zif_unmodified_sincer Z CONTENT_TYPE� mimetypesZ
guess_typer! Z_status�st_sizeZif_rangeZ
http_range�start�stop�
ValueErrorZ
CONTENT_RANGEr r5 r Z content_typeZCONTENT_ENCODINGZVARYZ
last_modifiedZcontent_lengthZ
ACCEPT_RANGESr
�int�format�methodZ METH_HEADr �open�closerD )r& r9 �filepathrE Z gzip_pathr6 �stZmodsinceZ
unmodsinceZct�encodingZ
should_set_ctr � file_sizer. rQ Zifrange�rng�endZ
real_startr, r- r' r) r* r< g s�
�
� zFileResponse.prepare)r r NN)�__name__�
__module__� __qualname__�__doc__r r! r" r# rT r r r r
r r r8 rD r< �
__classcell__r) r) r' r* r ) s4 � � � �r )&r/ rO �osr" r? �typingr r r r r r r r r
rF r �abcr
Ztypedefsr Zweb_exceptionsr r r r Z web_responser �__all__Z web_requestr rG Z_T_OnChunkSent�bool�environrJ r> r r) r) r) r* �<module> s ,