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)HTTPNotModifiedHTTPOkHTTPPartialContent 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_buffersuper__init__)selfargskwargs) __class__ /web_fileresponse.pyr s z SendfileStreamWriter.__init__c C s) | j t | 7_ | j j | d S)N) output_sizelenr 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 cancelledossendfileBlockingIOErrorInterruptedError Exceptionr Z
add_writer _sendfile_cbr )
r futout_fdin_fdoffsetcountloopZ
registerednexcr 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)NsocketF z Socket error) transportget_extra_infodup setblockingfilenotellr) joinr Z sock_sendallZ
create_futurer# r" r debugcloser 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
isinstancestrpathlibZPath_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 readwriteminZdrain)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) NFgzip z.gzTzapplication/octet-streamr rb)&rA Zheadersgetr ZACCEPT_ENCODINGZ with_namenameis_filestatZif_modified_sincest_mtimeZ timestampZ
set_statusr Z status_codeZ
_length_checkr rG Z CONTENT_TYPE mimetypesZ
guess_typer? r st_sizeZ
http_rangestartstop
ValueErrorr r
Z content_typeZCONTENT_ENCODINGZVARYZ
last_modifiedZcontent_lengthopenseek _sendfile)r rH filepathrN Z gzip_pathstZmodsincectencodingZ
should_set_ctstatusZ 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__boolenvironrQ rg r r
r r r r <module> s "?