a
\wiĔ @ sx d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z d dl m Z
z d dlZW n eyz dZY n0 ddlmZ ddlmZmZ dd lmZmZmZmZmZmZm Z m Z m Z m Z m Z dd
l!m"Z" dd l#m$Z$m%Z% e&e'Z(G d d
d
e)Z*G dd de)Z+G dd de)Z,edur
zDeflateDecoder.decompressc C s t | jp t | jjo | j S r% )boolr r r. r r r" r" r# has_unconsumed_tailZ s
z"DeflateDecoder.has_unconsumed_tailN)r+ __name__
__module__ __qualname__r$ r* r/ propertyr3 r" r" r" r# r & s
*r c @ s e Zd ZdZdZdZdS )GzipDecoderStater r N)r5 r6 r7 FIRST_MEMBER
OTHER_MEMBERS SWALLOW_DATAr" r" r" r# r9 a s r9 c @ s2 e Zd Zdd Zdd Zd ddZedd Zd
S ) GzipDecoderc C s$ t dt j | _tj| _d| _d S )N r )r r r0 r r9 r; _state_unconsumed_tailr r" r" r# r$ i s zGzipDecoder.__init__c C s t | j|S r% r& r( r" r" r# r* n s zGzipDecoder.__getattr__r+ c C s t }| jtjkrt|S |dkr4| j|7 _dS | j| }|sR| jjrRt|S z&|| jj|t |t
| dd7 }W n@ t j y | j}tj| _d| _|tj
krt| Y S Y n0 | jjp| jj | _}|dkrt
||krq|st|S | jjrRtj
| _t dt j | _qRt|S )Nr r r, r? ) bytearrayr@ r9 r= bytesrA r eofr/ maxlenr r r< r. unused_datar r0 )r! r1 r- retZprevious_stater" r" r# r/ q s<
zGzipDecoder.decompressc C s
t | jS r% )r2 rA r r" r" r# r3 s z GzipDecoder.has_unconsumed_tailN)r+ r4 r" r" r" r# r> h s
2r> c @ s< e Zd Zdd Zd
ddZdddZedd Zd
d Zd S )
BrotliDecoderc C s0 t | _t| jdr"| jj| _n
| jj| _d S )Nr/ )brotliZ Decompressorr hasattrr/ _decompressZprocessr r" r" r# r$ s
zBrotliDecoder.__init__r+ c C s
t d S r% )NotImplementedError)r! r1 output_buffer_limitr" r" r# rL s zBrotliDecoder._decompressc C sV z(|dkr| j ||dW S | |W S W n( tyP tdt | | Y S 0 d S )Nr )rN z;Brotli >= 1.2.0 is required to prevent decompression bombs.)rL TypeErrorwarningswarnr )r! r1 r- r" r" r# r/ s zBrotliDecoder.decompressc C s( z| j W S ty" Y dS 0 d S )NF)r Zcan_accept_more_dataAttributeErrorr r" r" r# r3 s z!BrotliDecoder.has_unconsumed_tailc C s t | jdr| j S dS )Nflushr )rK r rS r r" r" r# rS s
zBrotliDecoder.flushN)r+ )r+ ) r5 r6 r7 r$ rL r/ r8 r3 rS r" r" r" r# rI s
rI c @ s: e Zd ZdZdZdd Zdd Zd
dd Zed
d Z d S ) MultiDecodera
From RFC7231:
If one or more encodings have been applied to a representation, the
sender that applied the encodings MUST generate a Content-Encoding
header field that lists the content codings in the order in which
they were applied.
c C sP dd | dD }t|| jkr r z)MultiDecoder.__init__..,z)Too many content encodings in the chain: z > c S s g | ] }t |qS r" ) _get_decoderrX er" r" r# rZ r )splitrF max_decode_linksr _decoders)r! Zmodes encodingsr" r" r# r$ s zMultiDecoder.__init__c C s | j d S Nr )ra rS r r" r" r# rS s zMultiDecoder.flushr+ c C s |dkr&t | jD ]}||}q|S t }d}t | jD ]"}|j||t| d}|r:d}q:||7 }|rvt||kr~t|S d}q,d S )Nr Fr, Tr )reversedra r/ rB rF rC )r! r1 r- drH Zany_datar" r" r# r/ s zMultiDecoder.decompressc C s t dd | jD S )Nc s s | ] }|j V qd S r% )r3 )rX re r" r" r#
r z3MultiDecoder.has_unconsumed_tail..)anyra r r" r" r# r3 s z MultiDecoder.has_unconsumed_tailN)r+ )
r5 r6 r7 __doc__r` r$ rS r/ r8 r3 r" r" r" r# rT s
rT c C s: d| v rt | S | dkr t S td ur4| dkr4t S t S )Nr[ gzipbr)rT r> rJ rI r )moder" r" r# r\
s r\ c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) BytesQueueBuffera Memory-efficient bytes buffer
To return decoded data in read() and still follow the BufferedIOBase API, we need a
buffer to always return the correct amount of bytes.
This buffer should be filled using calls to put()
Our maximum memory usage is determined by the sum of the size of:
* self.buffer, which contains the full data
* the largest chunk that we will copy in get()
c C s t | _d| _d S rc ) collectionsdequebuffer_sizer r" r" r# r$ ( s
zBytesQueueBuffer.__init__c C s | j S r% )rp r r" r" r# __len__, s zBytesQueueBuffer.__len__c C s" | j | | jt|7 _d S r% )ro appendrp rF )r! r1 r" r" r# put/ s zBytesQueueBuffer.putc C s |dkr dS | j s tdn|dk r,tdt| j d |krft| j d trf| j|8 _| j S d}t }||k r|| }| j }t|}||k r|d | ||d }}|
| | j | | j|8 _qn|
| | j|8 _||7 }| j srqqr| S )Nr r zbuffer is emptyzn should be > 0)
ro RuntimeError
ValueErrorrF
isinstancerC rp popleftioBytesIOwrite
appendleftgetvalue) r! nZfetchedrH Z remainingchunkZ chunk_lengthZ
left_chunkZ right_chunkr" r" r# get3 s6
"
zBytesQueueBuffer.getN)r5 r6 r7 rh r$ rq rs r r" r" r" r# rl s
rl c @ sr e Zd ZdZddgZedur&edg7 Zg dZdLd d Zd
d Zdd Z dd Z
e dd Z e dd Z
dd Zdd Zdd Zd d ZeejfZedureejf7 ZdMd d Zd!d" Zed#d$ ZdNd%d&ZdOd'd(ZdPd*d+Z e d,d- Z d.d/ Z dQd0d1Z d2d3 Z!d4d5 Z"e d6d7 Z#d8d9 Z$d:d; Z%dd? Z'd@dA Z(dBdC Z)dDdE Z*dRdFdGZ+dHdI Z,dJdK Z-dS )S HTTPResponsea
HTTP Response container.
Backwards-compatible with :class:`http.client.HTTPResponse` but the response ``body`` is
loaded and decoded on-demand when the ``data`` property is accessed. This
class is also compatible with the Python standard library's :mod:`io`
module, and can hence be treated as a readable object in the context of that
framework.
Extra parameters for behaviour not present in :class:`http.client.HTTPResponse`:
:param preload_content:
If True, the response's body will be preloaded during construction.
:param decode_content:
If True, will attempt to decode the body based on the
'content-encoding' header.
:param original_response:
When this HTTPResponse wrapper is generated from an :class:`http.client.HTTPResponse`
object, it's convenient to include the original for debug purposes. It's
otherwise unused.
:param retries:
The retries contains the last :class:`~urllib3.util.retry.Retry` that
was used during the request.
:param enforce_content_length:
Enforce content length checking. Body returned by server must match
value of Content-Length header, if present. Otherwise, raise error.
ri ZdeflateNrj )i- i. i/ i3 i4 r TFc C s& t |tr|| _n
t|| _|| _|| _|| _|| _|| _d| _|
| _ || _
|| _ d | _ d | _
d | _| | _d| _| | _|| _|rt |tjtfr|| _
|
| _| | _t|dr|| _d| _d | _| jdd }dd | dD }d |v rd
| _| || _ t | _!|r"| j
s"| j"|d | _
d S ) NFr readztransfer-encodingr c s s | ]}| V qd S r% rV )rX encr" r" r# rf r z(HTTPResponse.__init__..r[ chunkedTdecode_content)#rv r headersstatusversionreasonstrictr _has_decoded_contentretriesenforce_content_length
auto_close_decoder_body_fp_original_response_fp_bytes_readmsg _request_urlr Z string_typesrC _pool _connectionrK r
chunk_leftr lowerr_ _init_lengthlength_remainingrl _decoded_bufferr )r! bodyr r r r r Zpreload_contentr original_responseZpool
connectionr r r request_methodZ request_urlr Ztr_encrb r" r" r# r$ | sF
zHTTPResponse.__init__c C s | j | jv r| jdS dS )a
Should we redirect and where to?
:returns: Truthy redirect location string if we got a redirect status
code and valid location. ``None`` if redirect status and no
location. ``False`` if not a redirect status code.
locationF)r REDIRECT_STATUSESr r r r" r" r# get_redirect_location s z"HTTPResponse.get_redirect_locationc C s( | j r | jsd S | j | j d | _d S r% )r r Z _put_connr r" r" r# release_conn s zHTTPResponse.release_connc C s2 z| j | jd W n ttttfy, Y n0 dS )z
Read and discard any remaining HTTP response data in the response connection.
Unread data in the HTTPResponse connection blocks the connection from being released back to the pool.
r N)r r r
SocketErrorr r r r" r" r#
drain_conn s
zHTTPResponse.drain_connc C s" | j r | j S | jr | jddS d S )NT)
cache_content)r r r r r" r" r# r1 s zHTTPResponse.datac C s | j S r% )r r r" r" r# r s zHTTPResponse.connectionc C s
t | jS r% )r r r r" r" r# isclosed s zHTTPResponse.isclosedc C s | j S )z
Obtain the number of bytes pulled over the wire so far. May differ from
the amount of content returned by :meth:``urllib3.response.HTTPResponse.read``
if bytes are encoded on the wire (e.g, compressed).
)r r r" r" r# tell s zHTTPResponse.tellc C s | j d}|dur| jr(td dS z.r[ r z8Content-Length contained multiple unmatching values (%s)r ) i0 d ZHEAD)
r r r logZwarningsetr_ rF r popru r r )r! r lengthZlengthsr r" r" r# r s6
&zHTTPResponse._init_lengthc sh j dd } jdu rd| jv r2t| _n2d|v rd fdd|dD }t|rdt| _dS )z=
Set-up the _decoder attribute if necessary.
content-encodingr Nr[ c s" g | ]}| jv r| qS r" )rW CONTENT_DECODERSr] r r" r# rZ = s z.HTTPResponse._init_decoder..)r r r r r r\ r_ rF )r! content_encodingrb r" r r#
_init_decoder2 s
zHTTPResponse._init_decoderc
C s |s| j rtd|S |du s"|r&d}z | jrD| jj||d}d| _ W nF | jy } z,| jdd }td| |W Y d}~n
d}~0 0 |r|| 7 }|S ) zN
Decode the data passed in and potentially flush the decoder.
_Calling read(decode_content=False) is not supported after read(decode_content=True) was called.Nr+ r, Tr r zEReceived response with content-encoding: %s, but failed to decode it.)
r rt r r/ DECODER_ERROR_CLASSESr r r r _flush_decoder)r! r1 r
flush_decoderr- r^ r r" r" r# _decodeI s.
zHTTPResponse._decodec C s$ | j r | j d}|| j S dS )zk
Flushes the decoder. Should only be called if the decoder is actually
being used.
r )r r/ rS )r! bufr" r" r# r h s zHTTPResponse._flush_decoderc
c s: d}zz
dV W n t y0 t| jddY nz tyt } z.dt|vrRt|t| jddW Y d}~n>d}~0 ttfy } ztd| |W Y d}~n
d}~0 0 d}W |s| j r| j
| j r| j
| j r| j r|
nH|s| j r| j
| j r| j
| j r4| j r4|
0 dS )z
Catch low-level python exceptions, instead re-raising urllib3
variants, so that low-level exceptions are not leaked in the
high-level api.
On exit, release the connection back to the pool.
FNzRead timed out.zread operation timed outzConnection broken: %rT)
SocketTimeoutr r r strr r r r r closer r r )r! Z
clean_exitr^ r" r" r# _error_catchers s6
"$
zHTTPResponse._error_catcherc C s | j du rdS t| j dd}| |du rB|s<| j nd}n|sR| j |nd}|dur|dkr|s| j | jr| jdur| jdkrt| j| jW d n1 s0 Y |r| jt |7 _| jdur| jt |8 _|S )z7
Reads `amt` of bytes from the socket.
NclosedFr r )
r r' r r r r r r r rF )r! amtZ fp_closedr1 r" r" r# _raw_read s*
,
zHTTPResponse._raw_readc C s~ | |du r| j}|durrd}| jrX| jjrX| jd|d|t| j d}| j| t| j|krr| j|S | |}d}|du rd}n|dkr|sd}|st| jdkr| jr| jjs|S |du r| |||}|r|| _
n|s| j rt d|S | j||||t| j d}| j| t| j|k rn|rn| |}| j||||t| j d}| j| q"| j|}|S ) aT
Similar to :meth:`http.client.HTTPResponse.read`, but with two additional
parameters: ``decode_content`` and ``cache_content``.
:param amt:
How much of the content to read. If specified, caching is skipped
because it doesn't make sense to cache partial content as the full
response.
:param decode_content:
If True, will attempt to decode the body based on the
'content-encoding' header.
:param cache_content:
If True, will save the returned data such that the same result is
returned despite of the state of the underlying file object. This
is useful if you want the ``.data`` property to continue working
after having ``.read()`` the file object. (Overridden if ``amt`` is
set.)
NFr )r r- Tr r r, )
r r r r3 r rF r rs r r r r rt )r! r r r Z decoded_datar1 r r" r" r# r st
zHTTPResponse.read c c sn | j r*| r*| j||dD ]
}|V q n@t| jrPt| jdksP| jrj| jjrj| j ||d}|r*|V q*dS )a_
A generator wrapper for the read() method. A call will block until
``amt`` bytes have been read from the connection or until the
connection is closed.
:param amt:
How much of the content to read. The generator will return up to
much data per iteration, but may return less. This is particularly
likely when using compressed data. However, the empty string will
never be returned.
:param decode_content:
If True, will attempt to decode the body based on the
'content-encoding' header.
r r )r r N)
r supports_chunked_reads read_chunkedr r rF r r r3 r )r! r r liner1 r" r" r# stream2 s
zHTTPResponse.streamc
K sb |j }t|ts.tjr"t|}n t| }t|dd}| f |||j|j |j
||d|}|S )a
Given an :class:`http.client.HTTPResponse` instance ``r``, return a
corresponding :class:`urllib3.response.HTTPResponse` object.
Remaining parameters are passed to the HTTPResponse constructor, along
with ``original_response=r``.
r r )r r r r r r r ) r rv r r ZPY2 from_httplibitemsr' r r r )Z ResponseClsrZ response_kwr r Zrespr" r" r# r P s$
zHTTPResponse.from_httplibc C s | j S r% r r r" r" r#
getheadersq s zHTTPResponse.getheadersc C s | j ||S r% )r r )r! r) defaultr" r" r# getheadert s zHTTPResponse.getheaderc C s | j S r% r r r" r" r# infox s zHTTPResponse.infoc C s6 | j s| j | jr | j | js2tj| d S r% )r r r r r rx IOBaser r" r" r# r | s
zHTTPResponse.closec C sT | j stjj| S | jd u r"dS t| jdr8| j S t| jdrL| jjS dS d S )NTr r )r rx r r __get__r rK r r r" r" r# r s
zHTTPResponse.closedc C s6 | j d u rtdn t| j dr*| j S tdd S )Nz-HTTPResponse has no file to get a fileno fromfilenozOThe file-like object this HTTPResponse is wrapped around has no file descriptor)r IOErrorrK r r r" r" r# r s
zHTTPResponse.filenoc C s2 | j d ur.t| j dr.t| j dds.| j S d S )NrS r F)r rK r' rS r r" r" r# rS s
zHTTPResponse.flushc C s dS )NTr" r r" r" r# readable s zHTTPResponse.readablec C s: | t|}t|dkr dS ||d t|< t|S d S rc )r rF )r! bZtempr" r" r# readinto s
zHTTPResponse.readintoc C s t | jdS )a
Checks if the underlying file-like object looks like a
:class:`http.client.HTTPResponse` object. We do this by testing for
the fp attribute. If it is present we assume it returns raw chunks as
processed by read_chunked().
fp)rK r r r" r" r# r s z#HTTPResponse.supports_chunked_readsc C sd | j d urd S | jj }|ddd }zt|d| _ W n$ ty^ | t| |Y n0 d S )N ;r r r? ) r r r readliner_ r ru r r )r! r r" r" r# _update_chunk_length s
z!HTTPResponse._update_chunk_lengthc C s d }|d u r2| j | j}|}| j d d | _nv|| jk rZ| j |}| j| | _|}nN|| jkr| j |}| j d d | _|}n | j | j}| j d d | _|S )Nr: )r Z
_safe_readr )r! r Zreturned_chunkr~ valuer" r" r#
_handle_chunk s&
zHTTPResponse._handle_chunkc c s | | jstd| s&td| | jrZt| jrZ| j W d dS | j j
du rvW d dS | | j dkrq|
|}| j||d|d}|rv|V qv|r| }|r|V | j j
}|sq|dkrqq| jr| j W d n1 s 0 Y dS )a
Similar to :meth:`HTTPResponse.read`, but with an additional
parameter: ``decode_content``.
:param amt:
How much of the content to read. If specified, caching is skipped
because it doesn't make sense to cache partial content as the full
response.
:param decode_content:
If True, will attempt to decode the body based on the
'content-encoding' header.
zHResponse is not chunked. Header 'transfer-encoding: chunked' is missing.zkBody should be http.client.HTTPResponse like. It should have have an fp attribute which returns raw chunks.Nr F)r r r- s
)r r r r r
r r r r r r r r r r r r )r! r r r~ Zdecodedr r" r" r# r sL
zHTTPResponse.read_chunkedc C s. | j dur$t| j jr$| j jd jS | jS dS )z
Returns the URL that was the source of this response.
If the request that generated this response redirected, this method
will return the final redirect location.
Nr+ )r rF historyZredirect_locationr r r" r" r# geturl) s zHTTPResponse.geturlc c s g }| j ddD ]n}d|v rt|d}d||d d V |dd D ]}|d V qJ|d rn|d g}q~g }q|| q|rd|V d S )NTr
r r r r+ )r r_ joinrr )r! ro r~ xr" r" r# __iter__4 s
zHTTPResponse.__iter__)r Nr r Nr TTNNNNNFNNT)N)N)NNF)r N)N)NN).r5 r6 r7 rh r rJ r r$ r r r r8 r1 r r r r r r r r r r r r r r r r classmethodr r r r r r r rS r r r r r r r r r" r" r" r# r V s|
G
4
7
+
\
I r )3Z
__future__r rm rx Zloggingr rP
contextlibr Zsocketr r r r rJ ImportError _collectionsr r r r
exceptionsr
r r r
r r r r r r r Zpackagesr Z
util.responser r Z getLoggerr5 r objectr r9 r> rI rT r\ rl r r r" r" r" r# s6
4
;@
+8
<