File: //opt/gsutil/third_party/apitools/apitools/base/py/__pycache__/util.cpython-39.pyc
a
�z�g! � @ s d Z ddlZddlZddlZddlmZ ddlm m m Z
ddl m m m Z
ddlm m mZ ddlmZ ddlmZ ddlmZ ejr�ddlmZ n ddlmZ dd gZd
Z d d� Z d d � Z d
d� Z d dd�Z d dd�Z!d dd�Z"dd� Z#dd� Z$dd� Z%dS ) z4Assorted utilities shared between parts of apitools.� N)� http_client)�messages)�encoding_helper)�
exceptions)�Iterable� DetectGae� DetectGcez:/?#[]@!$&'()*+,;=c C s" t j�dd�} | �d�p | �d�S )z�Determine 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_SOFTWARE� z Development/zGoogle App Engine/)�os�environ�get�
startswith)�server_software� r �9/opt/gsutil/third_party/apitools/apitools/base/py/util.pyr - s
�c C sr d� tj�dd��} z(t�t�i ���tj| ddid��}W n t j
yR Y dS 0 |� � t j
kop|j�d�dkS ) z�Determine 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-Flavor�Google)�headersFzmetadata-flavor)�formatr r r
�urllib_request� build_opener� ProxyHandler�open�Request� urllib_error�URLError�getcoder �OKr )� metadata_url�or r r r ; s ����c C s\ t | tj�r$t�| �} t| �d��S t | t�rDdd� | D �} t| �S t�dt | �f ��dS )z)Normalize scope_spec to a set of strings.� c S s g | ]}t �|��qS r )�six�
ensure_str)�.0�xr r r �
<listcomp>V � z#NormalizeScopes.<locals>.<listcomp>z5NormalizeScopes expected string or iterable, found %sN)
�
isinstancer" � string_typesr# �set�splitr r �TypecheckError�type)Z
scope_specr r r �NormalizeScopesP s
��r. c C sL t | |�sH|d u r>t |t�r.dt| �|f }ndt| �|f }t�|��| S )Nz"Type of arg is "%s", not one of %rz Type of arg is "%s", not "%s")r( �tupler- r r, )�argZarg_type�msgr r r � Typecheck] s
�
r2 c
C s |p | j p d}| jD ]�}d| }d}d| }||v rDt}|�||�}||vrZt�d| ��z || }W n ty� t�d| ��Y n0 |du r�t�d| ��z2t|tj �s�t
|�}|�|t � |�
d�|��}W q t�y
} z t�d||| f ��W Y d} ~ qd} ~ 0 0 q|S ) z(Determine the relative path for request.r
z{%s}z{+%s}zMissing path parameter %sz%Request missing required parameter %sN�utf_8z3Error setting required parameter %s to value %s: %s)�
relative_path� path_params�_RESERVED_URI_CHARS�replacer �InvalidUserInputError�KeyErrorr( r" r) �str� urllib_parse�quote�encode� TypeError)
�
method_config�paramsr4 �path�paramZparam_templateZreserved_charsZreserved_template�value�er r r �ExpandRelativePathi sJ
� �
� ���� rE �< c C s2 d| }|d }|t �| |�7 }tdt||��S )a� Calculates 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 @� )�random�uniform�max�min)Z
retry_attemptZmax_waitZ wait_timeZ
max_jitterr r r �CalculateWaitForRetry� s rM c s\ d�vrt �d� ��dd� | D �}|r<t �dd�|� ��dd� � t� �fd d
�| D ��S ) a� Return 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"c S s g | ]}d |v r|�qS )�;r �r$ �pr r r r& � r' z&AcceptableMimeType.<locals>.<listcomp>z.MIME patterns with parameter unsupported: "%s"z, c S s0 | dkr d} t dd� t| �d�|�d��D ��S )z4Return True iff mime_type is acceptable for pattern.�*z*/*c s s | ]\}}|d |fv V qdS )rR Nr )r$ �acceptZprovidedr r r � <genexpr>� r' z>AcceptableMimeType.<locals>.MimeTypeMatches.<locals>.<genexpr>rN )�all�zipr+ )�pattern� mime_typer r r �MimeTypeMatches� s
�z+AcceptableMimeType.<locals>.MimeTypeMatchesc 3 s | ]}� |��V qd S )Nr )r$ rW �rY rX r r rT � s �z%AcceptableMimeType.<locals>.<genexpr>)r r8 �GeneratedClientError�join�any)Zaccept_patternsrX Zunsupported_patternsr rZ r �AcceptableMimeType� s ����r^ c s � fdd�| D �S )z2Reverse parameter remappings for URL construction.c s g | ]}t j� |d �p|�qS ))� json_name)�encoding�GetCustomJsonFieldMappingrP �� request_typer r r& � s �z!MapParamNames.<locals>.<listcomp>r )r@ rc r rb r �
MapParamNames� s
�rd c C st t | �}| �� D ]^\}}tj||d�}|dur@|�|�||< |}t|tj�rtjt |�t
|�d�pht
|�||< q|S )a� Perform 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) �dict�itemsr` ra �popr( r �Enum�GetCustomJsonEnumMappingr- r: )r@ rc �
new_params�
param_namerC Zfield_remappingr r r �MapRequestParams� s � ��rm )N)N)rF )&�__doc__r rI r" � six.movesr Zsix.moves.urllib.error�moves�urllib�errorr Zsix.moves.urllib.parse�parser; Zsix.moves.urllib.request�requestr �apitools.base.protorpcliter �apitools.base.pyr r` r �PY3�collections.abcr � collections�__all__r6 r r r. r2 rE rM r^ rd rm r r r r �<module> s4 �
'
$