o !@sdZddlZddlZddlZddlmZddlmmm Z ddl mmm Z ddlmmmZddlmZddlmZddlmZejrTddlmZnddlmZdd gZd Zd dZd d Zd dZdddZ dddZ!dddZ"ddZ#ddZ$ddZ%dS)z4Assorted utilities shared between parts of apitools.N) http_client)messages)encoding_helper) exceptions)Iterable DetectGae DetectGcez:/?#[]@!$&'()*+,;=cCs"tjdd}|dp|dS)zDetermine whether or not we're running on GAE. This is based on: https://developers.google.com/appengine/docs/python/#The_Environment Returns: True iff we're running on GAE. SERVER_SOFTWAREz Development/zGoogle App Engine/)osenvironget startswith)server_softwarer>/tmp/google-cloud-sdk/lib/third_party/apitools/base/py/util.pyr-s cCsrdtjdd}zttitj|ddid}Wn t j y)YdSw| t j ko8|jddkS) zDetermine whether or not we're running on GCE. This is based on: https://cloud.google.com/compute/docs/metadata#runninggce Returns: True iff we're running on a GCE instance. z http://{}GCE_METADATA_ROOTzmetadata.google.internalzMetadata-FlavorGoogle)headersFzmetadata-flavor)formatr r r urllib_request build_opener ProxyHandleropenRequest urllib_errorURLErrorgetcoderOKr) metadata_urlorrrr;s cCsXt|tjrt|}t|dSt|tr"dd|D}t|Stdt |f)z)Normalize scope_spec to a set of strings. cSsg|]}t|qSr)six ensure_str).0xrrr Vsz#NormalizeScopes..z5NormalizeScopes expected string or iterable, found %s) isinstancer" string_typesr#setsplitrrTypecheckErrortype) scope_specrrrNormalizeScopesPs   r.cCsLt||s$|durt|trdt||f}ndt||f}t||S)Nz"Type of arg is "%s", not one of %rzType of arg is "%s", not "%s")r'tupler,rr+)argarg_typemsgrrr Typecheck]s   r3c Cs|p|jpd}|jD]r}d|}d}d|}||vr"t}|||}||vr-td|z||}WntyAtd|w|durMtd|zt|tj sXt |}||t | d|}Wq ty|} z td||| fd} ~ ww|S) z(Determine the relative path for request.r z{%s}z{+%s}zMissing path parameter %sz%Request missing required parameter %sNutf_8z3Error setting required parameter %s to value %s: %s) relative_path path_params_RESERVED_URI_CHARSreplacerInvalidUserInputErrorKeyErrorr'r"r(str urllib_parsequoteencode TypeError) method_configparamsr5pathparamparam_templatereserved_charsreserved_templatevalueerrrExpandRelativePathisP      rI<cCs2d|}|d}|t| |7}tdt||S)aCalculates amount of time to wait before a retry attempt. Wait time grows exponentially with the number of attempts. A random amount of jitter is added to spread out retry attempts from different clients. Args: retry_attempt: Retry attempt counter. max_wait: Upper bound for wait time [seconds]. Returns: Number of seconds to wait before retrying request. g@)randomuniformmaxmin) retry_attemptmax_wait wait_time max_jitterrrrCalculateWaitForRetrysrUcs\dvr tddd|D}|rtdd|ddtfd d |DS) aReturn True iff mime_type is acceptable for one of accept_patterns. Note that this function assumes that all patterns in accept_patterns will be simple types of the form "type/subtype", where one or both of these can be "*". We do not support parameters (i.e. "; q=") in patterns. Args: accept_patterns: list of acceptable MIME types. mime_type: the mime type we would like to match. Returns: Whether or not mime_type matches (at least) one of these patterns. /zInvalid MIME type: "%s"cSsg|]}d|vr|qS);rr$prrrr&sz&AcceptableMimeType..z.MIME patterns with parameter unsupported: "%s"z, cSs0|dkrd}tddt|d|dDS)z4Return True iff mime_type is acceptable for pattern.*z*/*css |] \}}|d|fvVqdS)rZNr)r$acceptprovidedrrr sz>AcceptableMimeType..MimeTypeMatches..rV)allzipr*)pattern mime_typerrrMimeTypeMatchess z+AcceptableMimeType..MimeTypeMatchesc3s|]}|VqdSNr)r$r`rbrarrr]sz%AcceptableMimeType..)rr9GeneratedClientErrorjoinany)accept_patternsraunsupported_patternsrrdrAcceptableMimeTypesrjcsfdd|DS)z2Reverse parameter remappings for URL construction.csg|] }tj|dp |qS)) json_name)encodingGetCustomJsonFieldMappingrX request_typerrr&sz!MapParamNames..r)rArorrnr MapParamNamess rpcCstt|}|D]/\}}tj||d}|dur ||||<|}t|tjr7tjt |t |dp4t |||<q|S)aPerform any renames/remappings needed for URL construction. Currently, we have several ways to customize JSON encoding, in particular of field names and enums. This works fine for JSON bodies, but also needs to be applied for path and query parameters in the URL. This function takes a dictionary from param names to values, and performs any registered mappings. We also need the request type (to look up the mappings). Args: params: (dict) Map from param names to values request_type: (protorpc.messages.Message) request type for this API call Returns: A new dict of the same size, with all registered mappings applied. ) python_nameN) dictitemsrlrmpopr'rEnumGetCustomJsonEnumMappingr,r;)rAro new_params param_namerGfield_remappingrrrMapRequestParamss   rzrc)rJ)&__doc__r rMr" six.movesrsix.moves.urllib.errormovesurlliberrorrsix.moves.urllib.parseparser<six.moves.urllib.requestrequestrapitools.base.protorpcliterapitools.base.pyrrlrPY3collections.abcr collections__all__r7rrr.r3rIrUrjrprzrrrrs6      '$