( ddlZddlZddlZddlmZddlmZejeZ dZ ejfZ Gdde ZdZdZd Zd Zd Zd Zd Zdddej,dfdZej0eZy)N) http_client)_helpers)readseektellc(eZdZdZdZdZdZdZy) MemoryCachez8httplib2 Cache implementation which only caches locally.ci|_yNcache)selfs 5platform/bq/third_party/oauth2client_4_0/transport.py__init__zMemoryCache.__init__$s  c8|jj|Sr )r getrkeys rrzMemoryCache.get'szz~~c""rc"||j|<yr r )rrvalues rsetzMemoryCache.set*s 3rc<|jj|dyr )r poprs rdeletezMemoryCache.delete-s sD!rN)__name__ __module__ __qualname____doc__rrrrrrr r !sB# "rr ctS)aReturn an HTTP object which caches results returned. This is intended to be used in methods like oauth2client_4_0.client.verify_id_token(), which calls to the same URI to retrieve certs. Returns: httplib2.Http, an HTTP object with a MemoryCache ) _CACHED_HTTPr rrget_cached_httpr#1s  rc,tj|i|S)a>Return a new HTTP object. Args: *args: tuple, The positional arguments to be passed when contructing a new HTTP object. **kwargs: dict, The keyword arguments to be passed when contructing a new HTTP object. Returns: httplib2.Http, an HTTP object. )httplib2Http)argskwargss rget_http_objectr)>s ==$ )& ))rc |iSt|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. )dict)headerss r_initialize_headersr-Ms23d7m3rc<|d|vr|dz|dz|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. z user-agent r )r, user_agents r_apply_user_agentr1ZsC 7 "%/#%5 8M%MGL ! N%/GL ! Nrcli} tj|D]z\}}t|tjs t |}t|tjs t |}t j ||t j |<| |S#t$rddlm }|dwxYw)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_typestrr _to_bytesUnicodeEncodeErroroauth2client_4_0.clientr3)r,cleankvr3s r clean_headersr?os E .MM'*DAqa1Fa1F+3+=+=a+@E($$Q' ( + L .?!!T1--.s BBB3c|jdddtjdffd }||_|j_y)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  js&tjd j t |} j |t | jd}tfdtDrj}t ||t|||\}}d} t| D]} |jtvr||fStjd|j| dz|  j  j ||j!|t ||t|||\}}||fS)Nz1Attempting refresh to obtain initial access_tokenc38K|]}t|dywr )getattr).0 stream_propbodys r z:wrap_http_for_auth..new_request..s##!5@wt[$/!sz&Refreshing due to a %s (attempt %s/%s)) access_token_LOGGERinfo_refreshr-applyr1r0all_STREAM_PROPERTIESrrequestr?rangestatusREFRESH_STATUS_CODESr) urimethodrGr, redirectionsconnection_typebody_stream_positionrespcontentmax_refresh_attemptsrefresh_attempt credentialsorig_request_methods ` r new_requestz'wrap_http_for_auth..new_requests\'' LL0 1  !4 5&g.'"';#9#9:# #!# ##'99;  3S&$ -g 6 ,o? g !$%9:O{{"66W} LLAo&9- /  !4 5   g &#/ ./#$7fd$1'$:$0/CMD' ;W}r)rRr%DEFAULT_MAX_REDIRECTSr_)r_httprar`s` @rwrap_http_for_authrdsA,,!&D$!)!?!?$(*ZDL +DLLrc|jt||jdddtjdffd }||_|j_y)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. rANc pd jvr: j jr jdt ||||||St |}t | j|jddd} jd|i\}}d|z|d<t |||t|||S)Naud?rJrzBearer Authorization) _kwargsrKaccess_token_expiredrefreshrRr-r1r0split _create_tokenr?) rVrWrGr,rXrYuri_roottoken unused_expiryauthenticated_request_methodr_r`s rraz-wrap_http_for_jwt_access..new_requests K'' '((044##D)7!4,*, , *'2G g{'='= >yya(+H#.#<#2 LL$fc2D)T2M V$&2)8 ::r)loggingr%r4 six.movesroauth2client_4_0r getLoggerrrLrQ UNAUTHORIZEDrUobjectr r#r)r-r1r?rdrsrbrRr&r"r rrr|s  !% '  H %-$002 "& "  * 4*6@+F/+d$$!77 :@x}}[]+ r