o ]<@shdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl mZzddlmZWneyMddlmZYnwgd Zd Zd Ze je je je jefZed gd Zejd)ddZGddde Z!GdddedgdZ"ddZ#ddZ$ddZ%ddZ&ddd e&e#fd!d"Z'd e#fd#d$Z(gZ)d%d&Z*d'd(Z+dS)*zpHTTP wrapper for apitools. This library wraps the underlying http library we use, which is currently httplib2. N) http_client)parse) exceptions)util)HttpAccessTokenRefreshError)AccessTokenRefreshError) CheckResponseGetHttp)HandleExceptionsAndRebuildHttpConnections MakeRequestRebuildHttpConnectionsRequestResponseRethrowExceptionHandler4iExceptionRetryArgs)http http_requestexc num_retriesmax_retry_waittotal_wait_secccs|jdur dVdStj}i}|t_|dur0|jD]\}}d|vr%q|j||<||qdV|t_|durP|D]\}}||jvrO|j||q>dSdS)aTemporarily change the value of httplib2.debuglevel, if necessary. If http_request has a `loggable_body` distinct from `body`, then we need to prevent httplib2 from logging the full body. This sets httplib2.debuglevel for the duration of the `with` block; however, that alone won't change the value of existing HTTP connections. If an httplib2.Http object is provided, we'll also change the level on any cached connections attached to it. Args: http_request: a Request we're logging. level: (int) the debuglevel for logging. http: (optional) an httplib2.Http whose connections we should set the debuglevel on. Yields: None. N:) loggable_bodyhttplib2 debuglevel connectionsitemsset_debuglevel)rlevelr old_level http_levelsconnection_key connectionr$F/tmp/google-cloud-sdk/lib/third_party/apitools/base/py/http_wrapper.py_Httplib2DebuglevelKs,    r&c@sNeZdZdZd ddZeddZejd dZed d Zejd d ZdS)r z1Class encapsulating the data for an HTTP request.GETNcCs,||_||_|p i|_d|_d|_||_dSN)url http_methodheaders_Request__body_Request__loggable_bodybody)selfr*r+r,r/r$r$r%__init__{s   zRequest.__init__cC|jSr))r.r0r$r$r%rzRequest.loggable_bodycCs|jdur td||_dS)Nz0Cannot set loggable body on request with no body)r/r RequestErrorr.)r0valuer$r$r%rs  cCr2r))r-r3r$r$r%r/r4z Request.bodycCsf||_|durt|jddpt|j}t||jd<n|jddt|tdtj fs1d|_ dSdS)z>Sets the request body; handles logging and length measurement.Nlengthcontent-lengthz ) r-getattrlenstrr,pop isinstancetypesix string_typesr)r0r6 body_lengthr$r$r%r/s )r'r(Nr') __name__ __module__ __qualname____doc__r1propertyrsetterr/r$r$r$r%r ws    r c@sLeZdZdZdZddZeddZeddZed d Z ed d Z d S)rz.Class encapsulating data for an HTTP response.r$cCr2r))r7r3r$r$r%__len__szResponse.__len__cCsfdd}d|jvrd|jvr||jdSd|jvr"t|jdSd|jvr.||jdSt|jS)zReturn the length of this response. We expose this as an attribute since using len() directly can fail for responses larger than sys.maxint. Returns: Response length (as int or long) cSsD|d\}}}|d\}}}|d\}}}t|t|dS)N /-) partitionint) content_range_ range_spec byte_rangestartendr$r$r%ProcessContentRangesz,Response.length..ProcessContentRangez-content-encodingz content-ranger8)inforNgetr:content)r0rUr$r$r%r7s    zResponse.lengthcCst|jdS)Nstatus)rNrVr3r$r$r% status_codeszResponse.status_codecCsd|jvr t|jdSdS)Nz retry-after)rVrNr3r$r$r% retry_afters zResponse.retry_aftercCs|jtvo d|jvS)Nlocation)rZ_REDIRECT_STATUS_CODESrVr3r$r$r% is_redirects zResponse.is_redirectN) rBrCrDrE __slots__rHrFr7rZr[r^r$r$r$r%rs   r HttpResponse)rVrX request_urlcCsN|dur td|j|jdks|jtkrtj||jr%tj|dS)Nz,Request to url %s did not return a response.) rr5rarZTOO_MANY_REQUESTSBadStatusCodeError FromResponser[RetryAfterError)responser$r$r%rs    rcCs8t|ddrt|jD] }d|vr|j|=q dSdS)aRebuilds all http connections in the httplib2.Http instance. httplib2 overloads the map in http.connections to contain two different types of values: { scheme string: connection class } and { scheme + authority string : actual http connection } Here we remove all of the entries for actual connections so that on the next request httplib2 will rebuild them from the connection types. Args: http: An httplib2.Http instance. rNr)r9listrkeys)rconn_keyr$r$r%r s r cGsr)r$) unused_argsr$r$r%rsrcCsd}t|jtjtjtjfrtdt|jj |jnt|jt j r*td|jnt|jt j r9td|jn}t|jt j rHtd|jnnt|jtjrWtd|jn_t|jtretd|jnQt|jtrt|jdr|jjtks}|jjd krtd |jn1t|jtjrtd n$t|jtjrtd |jjnt|jtjrtd |jj}n|jt|jtd|jj|jt |pt!j"|j#|j$ddS)zException handler for http failures. This catches known failures and rebuilds the underlying HTTP connections. Args: retry_args: An ExceptionRetryArgs tuple. Nz"Caught HTTP error %s, retrying: %sz!Caught socket error, retrying: %sz)Caught socket address error, retrying: %sz)Caught socket timeout error, retrying: %sz+Caught server not found error, retrying: %sz+Response content was invalid (%s), retryingrYrbz8Caught transient credential refresh error (%s), retryingz&Request returned no response, retryingz%Response returned status %s, retryingz0Response returned a retry-after header, retryingz-Retrying request to url %s after exception %s)max_wait)%r=rr BadStatusLineIncompleteReadResponseNotReadyloggingdebugr>rBsocketerrorgaierrortimeoutrServerNotFoundError ValueErrorTokenRefreshErrorhasattrrYrcrr5rdrZrfr[r rrr*timesleeprCalculateWaitForRetryrr) retry_argsr[r$r$r%r sl           r <c Csd}t}t|drt|jdh|_ z t||||dWStyJ} z |d7}||kr/t|} |t||| ||| WYd} ~ nd} ~ wwq)aSend http_request via the given http, performing error/retry handling. Args: http: An httplib2.Http instance, or a http multiplexer that delegates to an underlying http, for example, HTTPMultiplexer. http_request: A Request to send. retries: (int, default 7) Number of retries to attempt on retryable replies (such as 429 or 5XX). max_retry_wait: (int, default 60) Maximum number of seconds to wait when retrying. redirections: (int, default 5) Number of redirects to follow. retry_func: Function to handle retries on exceptions. Argument is an ExceptionRetryArgs tuple. check_response_func: Function to validate the HTTP response. Arguments are (Response, response content, url). Raises: InvalidDataFromServerError: if there is no response after retries. Returns: A Response object. rredirect_codesrT) redirectionscheck_response_funcrLN)rzrysetr_MakeRequestNoRetry Exceptionr) rrretriesrr retry_funcrretryfirst_req_timeerr$r$r%r 9s*   r c Csd}t|ddrt|jj}|r||jvr|j|}tjdkr"dnd}t|||d|j t |jt |j |j |j ||d\}}Wdn1sKwY|durXtt|||j} || | S)aSend http_request via the given http. This wrapper exists to handle translation between the plain httplib2 request/response types and the Request and Response types above. Args: http: An httplib2.Http instance, or a http multiplexer that delegates to an underlying http, for example, HTTPMultiplexer. http_request: A Request to send. redirections: (int, default 5) Number of redirects to follow. check_response_func: Function to validate the HTTP response. Arguments are (Response, response content, url). Returns: A Response object. Raises: RequestError if no response could be parsed. Nrr)r)methodr/r,rconnection_type)r9rurlsplitr*schemerrrr&requestr;r+r/r,rr5r) rrrrr url_schemenew_debuglevelrVrXrgr$r$r%rks$   rcCst|dSr))_HTTP_FACTORIESappend)factoryr$r$r%_RegisterHttpFactorysrcKs8tD]}|di|}|dur|Sqtjdi|S)Nr$)rrHttp)kwdsrrr$r$r%r s r r)),rE collections contextlibrprrrzrr? six.movesrsix.moves.urllibrapitools.base.pyrroauth2client.clientrrx ImportErrorr__all__RESUME_INCOMPLETErcMOVED_PERMANENTLYFOUND SEE_OTHERTEMPORARY_REDIRECTr] namedtuplercontextmanagerr&objectr rrr rr r rrrr r$r$r$r%sd        +-3 = 2 /