���Z � @ s� d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m Z d d l
m Z d d l m
Z
m Z m Z m Z d d l m Z d Z e e j j d
� � Z Gd d � d e � Z Gd
d � d e � Z d S)� N� )�hdrs)�
set_exception�
set_result)� StreamWriter)�
server_logger)�HTTPNotModified�HTTPOk�HTTPPartialContent� HTTPRequestRangeNotSatisfiable)�StreamResponse� FileResponseZAIOHTTP_NOSENDFILEc s[ e Z d Z � f d d � Z d d � Z d d � Z � f d d � Z d d
d � Z � S) �SendfileStreamWriterc s g | _ t � j | | � d S)N)�_sendfile_buffer�super�__init__)�self�args�kwargs)� __class__� �/web_fileresponse.pyr s z SendfileStreamWriter.__init__c C s) | j t | � 7_ | j j | � d S)N)� output_size�lenr �append)r �chunkr r r �_write s zSendfileStreamWriter._writec
C s� | r | j | � | j � r# d Sy. t j | | | | � } | d k rP | } WnS t t f k
rq d } Yn6 t k
r� } z t | | � d SWYd d } ~ Xn X| | k r� | j | | j | | | | | | | | d � n
t
| d � d S)Nr T) Z
remove_writer� cancelled�os�sendfile�BlockingIOError�InterruptedError� Exceptionr Z
add_writer� _sendfile_cbr )
r �fut�out_fd�in_fd�offset�count�loopZ
registered�n�excr r r r# ! s"
z!SendfileStreamWriter._sendfile_cbc
� s | j j d � j � } | j d � | j � } | j � } | j � } | j } d j | j � } z� yM | j | | � Id H| j
� } | j | | | | | | d � | Id HWn, t k
r� t
j d � | j j � Yn XWd | j � X| j | 7_ t � j � Id Hd S)N�socketF� z Socket error)� transport�get_extra_info�dup� setblocking�fileno�tellr) �joinr Z sock_sendallZ
create_futurer# r" r �debug�closer r � write_eof)
r �fobjr( Z
out_socketr% r& r' r) �datar$ )r r r r 8 s$
z SendfileStreamWriter.sendfiler- c � s d S)Nr )r r r r r r7 O s z SendfileStreamWriter.write_eof)�__name__�
__module__� __qualname__r r r# r r7 r r )r r r s
r c s� e Z d Z d Z d
� f d d � Z � f d d � Z � f d d � Z e e d
� rj e rj e Z
n e Z
� f d d � Z � S)r
z,A response object can be used to send files.� i c sG t � j | | � t | t � r1 t j | � } | | _ | | _ d S)N)r r �
isinstance�str�pathlibZPath�_path� _chunk_size)r �path�
chunk_sizer r )r r r r V s
zFileResponse.__init__c � s� | j } | j d � s- | j d � d k rJ | j | | | � Id H} nK t | j | | j � } | | _ t � j | � Id H| j | | � Id H| S)NZ
sslcontextr, )
r. r/ �_sendfile_fallbackr �protocolr) Z_payload_writerr �preparer )r �requestr8 r( r. �writer)r r r �_sendfile_system_ s z FileResponse._sendfile_systemc � s� t � j | � Id H} | j } | j | � } xE | j | � Id H| | } | d k r[ P| j t | | � � } q2 W| j � Id H| S)Nr )r rG rB �read�write�minZdrain)r rH r8 r( rI rD r )r r r rE | s
z FileResponse._sendfile_fallbackr c � s� | j } d } d | j j t j d � k r[ | j | j d � } | j � r[ | } d } | j � } | j } | d k r� | j
| j � k r� | j t
j � d | _ t � j | � Id HSt j | j k rt j t | � � \ } } | s� d } d } n | r
d n d } d } t j }
| j } | } y | j }
|
j } |
j } Wn7 t k
r�| j t j � t � j | � Id HSYn X| d k s�| d k r
| d k r�| d k r�| | } | } n | p�| | } | | | k r�| | } | | k r
d } | | k r t j }
| j |
� | r;| | _ | rQ| | j t j <| rjt j | j t j <| j
| _! | | _" | r�| j# d � �2 } | r�| j$ | � | j% | | | � Id HSWd QRXt � j | � Id HS) NF�gzip� z.gzTzapplication/octet-streamr �rb)&rA Zheaders�getr ZACCEPT_ENCODINGZ with_name�name�is_file�statZif_modified_since�st_mtimeZ timestampZ
set_statusr Z status_codeZ
_length_checkr rG Z CONTENT_TYPE� mimetypesZ
guess_typer? r �st_sizeZ
http_range�start�stop�
ValueErrorr r
Z content_typeZCONTENT_ENCODINGZVARYZ
last_modifiedZcontent_length�open�seek� _sendfile)r rH �filepathrN Z gzip_path�stZmodsince�ct�encodingZ
should_set_ct�statusZ file_sizer( �rngrX �endr8 )r r r rG � sp !
zFileResponse.preparei ) r: r; r< �__doc__r rJ rE �hasattrr �
NOSENDFILEr] rG r r )r r r
S s )r
)rV r r@ rO r Zhelpersr r Z http_writerr �logr Zweb_exceptionsr r r
r Z web_responser �__all__�bool�environrQ rg r r
r r r r �<module> s "?