3
Z'íc+ ã @ s d Z ddlmZ ddlZddlZddlZddlmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ dd lmZ dd
lmZ d d lmZ ydd
lmZ W n ek
r¸ dZY nX ej e Z G dd deZ G dd deZ G dd deZ!ej"Z"ej#Z#ej$Z$ej%Z%ej&Z&dS )aC
This module provides a pool manager that uses Google App Engine's
`URLFetch Service <https://cloud.google.com/appengine/docs/python/urlfetch>`_.
Example usage::
from urllib3 import PoolManager
from urllib3.contrib.appengine import AppEngineManager, is_appengine_sandbox
if is_appengine_sandbox():
# AppEngineManager uses AppEngine's URLFetch API behind the scenes
http = AppEngineManager()
else:
# PoolManager uses a socket-level API behind the scenes
http = PoolManager()
r = http.request('GET', 'https://google.com/')
There are `limitations <https://cloud.google.com/appengine/docs/python/urlfetch/#Python_Quotas_and_limits>`_ to the URLFetch service and it may not be
the best choice for your application. There are three options for using
urllib3 on Google App Engine:
1. You can use :class:`AppEngineManager` with URLFetch. URLFetch is
cost-effective in many circumstances as long as your usage is within the
limitations.
2. You can use a normal :class:`~urllib3.PoolManager` by enabling sockets.
Sockets also have `limitations and restrictions
<https://cloud.google.com/appengine/docs/python/sockets/ #limitations-and-restrictions>`_ and have a lower free quota than URLFetch.
To use sockets, be sure to specify the following in your ``app.yaml``::
env_variables:
GAE_USE_SOCKETS_HTTPLIB : 'true'
3. If you are using `App Engine Flexible
<https://cloud.google.com/appengine/docs/flexible/>`_, you can use the standard
:class:`PoolManager` without any configuration or special environment variables.
é )Úabsolute_importNé )Ú HTTPErrorÚ HTTPWarningÚ
MaxRetryErrorÚ
ProtocolErrorÚSSLErrorÚ TimeoutError)Úurljoin)ÚRequestMethods)Ú HTTPResponse)ÚRetry)ÚTimeouté )Ú_appengine_environ)Úurlfetchc @ s e Zd ZdS )ÚAppEnginePlatformWarningN)Ú__name__Ú
__module__Ú __qualname__© r r ú</tmp/pip-build-gk9425m9/urllib3/urllib3/contrib/appengine.pyr G s r c @ s e Zd ZdS )ÚAppEnginePlatformErrorN)r r r r r r r r K s r c @ sX e Zd ZdZdddZdd Zdd Zddddejfd
d Z d d
Z
dd Z dd Z dS )ÚAppEngineManagera
Connection manager for Google App Engine sandbox applications.
This manager uses the URLFetch service directly instead of using the
emulated httplib, and is subject to URLFetch limitations as described in
the App Engine documentation `here
<https://cloud.google.com/appengine/docs/python/urlfetch>`_.
Notably it will raise an :class:`AppEnginePlatformError` if:
* URLFetch is not available.
* If you attempt to use this on App Engine Flexible, as full socket
support is available.
* If a request size is more than 10 megabytes.
* If a response size is more than 32 megabytes.
* If you use an unsupported request method such as OPTIONS.
Beyond those cases, it will raise normal urllib3 errors.
NTc C s@ t s tdtjdt tj| | || _|| _|p8t j
| _ d S )Nz.URLFetch is not available in this environment.zÃurllib3 is using URLFetch on Google App Engine sandbox instead of sockets. To use sockets directly instead of URLFetch see https://urllib3.readthedocs.io/en/1.26.x/reference/urllib3.contrib.html.) r r ÚwarningsÚwarnr r Ú__init__Úvalidate_certificateÚurlfetch_retriesr
ÚDEFAULTÚretries)ÚselfÚheadersr r r r r r r c s zAppEngineManager.__init__c C s | S )Nr )r! r r r Ú __enter__| s zAppEngineManager.__enter__c C s dS )NFr )r! Úexc_typeÚexc_valÚexc_tbr r r Ú__exit__ s zAppEngineManager.__exit__c K sà | j ||}yF|o |jdko |j} tj||||p2i d| jo<| | j|| jd}
W nB tjk
r } zt | | W Y d d } ~ X n tj
k
r } z$dt | krªt d| t
| W Y d d } ~ X nÔ tjk
r } z(dt | krðt| || dt
| W Y d d } ~ X n tjk
r6 } zt d| W Y d d } ~ X n` tjk
rb } zt| W Y d d } ~ X n4 tjk
r } zt d | | W Y d d } ~ X nX | j|
fd
|i|} |o¸| j }
|
rr| jrÞ|jrÞt| |d n| jd krîd
}y|j||| | d}W n* tk
r. |jr*t| |d | S X |j| tjd||
t ||
}| j ||||f|||d|S t | j j d}|j!|| j|rÜ|j||| | d}tjd| |j"| | j ||f|||||d|S | S )Nr F)ÚpayloadÚmethodr" Zallow_truncatedÚfollow_redirectsÚdeadliner z too largezOURLFetch request too large, URLFetch only supports requests up to 10mb in size.zToo many redirects)ÚreasonzPURLFetch response too large, URLFetch only supportsresponses up to 32mb in size.z$URLFetch does not support method: %sr ztoo many redirectsi/ ÚGET)ÚresponseÚ_poolzRedirecting %s -> %s)r ÚredirectÚtimeoutz Retry-Afterz Retry: %s)Úbodyr" r r0 r1 )#Ú _get_retriesr0 Útotalr Úfetchr Ú_get_absolute_timeoutr ZDeadlineExceededErrorr ZInvalidURLErrorÚstrr r Z
DownloadErrorr ZResponseTooLargeErrorZSSLCertificateErrorr ZInvalidMethodErrorÚ#_urlfetch_response_to_http_responseÚget_redirect_locationÚraise_on_redirectÚstatusÚ incrementÚsleep_for_retryÚlogÚdebugr
ÚurlopenÚboolr" ÚgetÚis_retryÚsleep)r! r) Úurlr2 r" r r0 r1 Ú response_kwr* r. ÚeÚ
http_responseÚredirect_locationZ redirect_urlÚhas_retry_afterr r r r@ s
zAppEngineManager.urlopenc K s¤ t r"|jjd}|dkr"|jd= |jjd}|dkrZ|jd}|jd dj||jd< tf tj|j |j
|j|j d|}tf tj|j |j|j |d|S )Nzcontent-encodingÚdeflateztransfer-encodingÚchunkedú,)r2 Úmsgr" r; )r2 r" r; Úoriginal_response) Úis_prod_appenginer" rB ÚsplitÚremoveÚjoinr ÚioÚBytesIOÚcontentZ
header_msgÚ status_code)r! Z
urlfetch_resprF Úcontent_encodingZtransfer_encodingÚ encodingsrO r r r r8 õ s*
z4AppEngineManager._urlfetch_response_to_http_responsec C sB |t jkrd S t|t r>|jd k s,|jd k r8tjdt |jS |S )NzdURLFetch does not support granular timeout settings, reverting to total or default URLFetch timeout.) r ÚDEFAULT_TIMEOUTÚ
isinstanceÚ_readÚ_connectr r r r4 )r! r1 r r r r6 s
z&AppEngineManager._get_absolute_timeoutc C s> t |ts tj||| jd}|js.|js.|jr:tjdt |S )N)r0 ÚdefaultzhURLFetch only supports total retries and does not recognize connect, read, or redirect retry parameters.)
r[ r
Úfrom_intr ÚconnectÚreadr0 r r r )r! r r0 r r r r3 & s
z AppEngineManager._get_retries)NNTT)
r r r Ú__doc__r r# r' r rZ r@ r8 r6 r3 r r r r r O s
j$
r )'rb Ú
__future__r rT Úloggingr Ú
exceptionsr r r r r r Z packages.six.moves.urllib.parser
Úrequestr r. r Z
util.retryr
Z util.timeoutr Ú r Zgoogle.appengine.apir Ú ImportErrorÚ getLoggerr r> r r r Z is_appengineZis_appengine_sandboxZis_local_appenginerP Zis_prod_appengine_mvmsr r r r Ú<module>' s2
h