o !@sddlZddlZddlZddlmZddlmZeeZ dZ ej fZ Gddde ZddZd d Zd d Zd dZddZddZddZeeZdS)N) http_client) _to_bytes)readseektellc@s0eZdZdZddZddZddZdd Zd S) MemoryCachez8httplib2 Cache implementation which only caches locally.cCs i|_dSNcache)selfr ?/tmp/google-cloud-sdk/lib/third_party/oauth2client/transport.py__init__#s zMemoryCache.__init__cCs |j|Sr)r getr keyr r r r&s zMemoryCache.getcCs||j|<dSrr )r rvaluer r r set)szMemoryCache.setcCs|j|ddSr)r poprr r r delete,szMemoryCache.deleteN)__name__ __module__ __qualname____doc__rrrrr r r r r s  rcCstS)aReturn an HTTP object which caches results returned. This is intended to be used in methods like oauth2client.client.verify_id_token(), which calls to the same URI to retrieve certs. Returns: httplib2.Http, an HTTP object with a MemoryCache ) _CACHED_HTTPr r r r get_cached_http0s rcCstS)zSReturn a new HTTP object. Returns: httplib2.Http, an HTTP object. )httplib2Httpr r r r get_http_object=srcCs|duriSt|S)zCreates a copy of the headers. Args: headers: dict, request headers to copy. Returns: dict, the copied headers or a new dictionary if the headers were None. N)dict)headersr r r _initialize_headersFs r!cCs4|durd|vr|d|d|d<|S||d<|S)a,Adds a user-agent to the headers. Args: headers: dict, request headers to add / modify user agent within. user_agent: str, the user agent to add. Returns: dict, the original headers passed in, but modified if the user agent is not None. Nz user-agent r )r user_agentr r r _apply_user_agentSs  r$cCs~i}z)t|D] \}}t|tjst|}t|tjs t|}t||t|<qW|Sty>ddlm}||d|w)aForces header keys and values to be strings, i.e not unicode. The httplib module just concats the header keys and values in a way that may make the message header a unicode string, which, if it then tries to contatenate to a binary request body may result in a unicode decode error. Args: headers: dict, A dictionary of headers. Returns: The same dictionary but with all the keys converted to strings. r)NonAsciiHeaderErrorz: ) six iteritems isinstance binary_typestrrUnicodeEncodeErroroauth2client.clientr%)r cleankvr%r r r clean_headershs     r0cs:|jdddtjdffdd }||_t|jddS)aPrepares an HTTP object's request method for auth. Wraps HTTP requests with logic to catch auth failures (typically identified via a 401 status code). In the event of failure, tries to refresh the token used and then retry the original request. Args: credentials: Credentials, the credentials used to identify the authenticated user. http: httplib2.Http, an http object to be used to make auth requests. GETNc sjs tdt|}|t|jd}tfddt Dr- }||t |||\}}d} t | D]7} |j tvrL||fStd|j | d| ||durj|||t |||\}}q@||fS)Nz1Attempting refresh to obtain initial access_tokenc3s|] }t|dVqdSr)getattr).0 stream_propbodyr r sz:wrap_http_for_auth..new_request..z&Refreshing due to a %s (attempt %s/%s)) access_token_LOGGERinfo_refreshr!applyr$r#all_STREAM_PROPERTIESrr0rangestatusREFRESH_STATUS_CODESr) urimethodr6r redirectionsconnection_typebody_stream_positionrespcontentmax_refresh_attemptsrefresh_attempt credentialsorig_request_methodr5r new_requestsB             z'wrap_http_for_auth..new_requestrN)requestrDEFAULT_MAX_REDIRECTSsetattrrNhttprPr rMr wrap_http_for_auths -rVcs>|jt||jdddtjdffdd }||_dS)aPrepares an HTTP object's request method for JWT access. Wraps HTTP requests with logic to catch auth failures (typically identified via a 401 status code). In the event of failure, tries to refresh the token used and then retry the original request. Args: credentials: _JWTAccessCredentials, the credentials used to identify a service account that uses JWT access tokens. http: httplib2.Http, an http object to be used to make auth requests. r1Nc sdjvrjdus jrd||||||St|}t|j|ddd}d|i\}}d||d<|||t |||S)Naud?r9rzBearer Authorization) _kwargsr:access_token_expiredrefreshr!r$r#split _create_tokenr0) rDrEr6r rFrGuri_roottoken unused_expiryauthenticated_request_methodrNrOr r rPs$     z-wrap_http_for_jwt_access..new_request)rQrVrrRrTr rbr wrap_http_for_jwt_accesss  rd)loggingrr& six.movesroauth2client._helpersr getLoggerrr;r@ UNAUTHORIZEDrCobjectrrrr!r$r0rVrdrrr r r r s"      C/