File: //opt/gsutil/third_party/apitools/apitools/base/py/__pycache__/base_api.cpython-39.pyc
a
�z�g,u � @ s6 d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl
m Z ddl
mZ ddl
mZ dd l
mZ dd
l
mZ g d �Ze�d ��ej� d
ZG dd� de j�ZG dd� de j�ZdZdd� Zdd� Z dd� Z dd� Z G dd � d e �Z d d � Z!G d d � d e �Z"G d!d"� d"e �Z#dS )#z Base class for api services.� N)� http_client)�urllib)�
message_types)�messages)�encoding)�
exceptions)� http_wrapper)�util)�
ApiMethodInfo�
ApiUploadInfo�
BaseApiClient�BaseApiService�NormalizeApiEndpointzoauth2client.utili c @ sP e Zd ZdZejddd�Ze�d�Ze�d�Z e�
d�Z e�d�Z e�
d �Z
d
S ) r aT Media upload information for a method.
Fields:
accept: (repeated) MIME Media Ranges for acceptable media uploads
to this method.
max_size: (integer) Maximum size of a media upload, such as 3MB
or 1TB (converted to an integer).
resumable_path: Path to use for resumable uploads.
resumable_multipart: (boolean) Whether or not the resumable endpoint
supports multipart uploads.
simple_path: Path to use for simple uploads.
simple_multipart: (boolean) Whether or not the simple endpoint
supports multipart uploads.
� T��repeated� � � � � N)�__name__�
__module__� __qualname__�__doc__r � StringField�accept� IntegerField�max_sizeZresumable_path� BooleanFieldZresumable_multipartZ simple_pathZsimple_multipart� r r �=/opt/gsutil/third_party/apitools/apitools/base/py/base_api.pyr 5 s
r c @ s� e Zd ZdZe�d�Ze�d�Ze�d�Ze�d�Z ejddd�Z
ejd dd�Z ejd
dd�Z e�d �Z
e�d �Ze�d
�Zejddd�Ze�ed�Zejddd�Ze�d�ZdS )r
a� Configuration info for an API method.
All fields are strings unless noted otherwise.
Fields:
relative_path: Relative path for this method.
flat_path: Expanded version (if any) of relative_path.
method_id: ID for this method.
http_method: HTTP verb to use for this method.
path_params: (repeated) path parameters for this method.
query_params: (repeated) query parameters for this method.
ordered_params: (repeated) ordered list of parameters for
this method.
description: description of this method.
request_type_name: name of the request type.
response_type_name: name of the response type.
request_field: if not null, the field to pass as the body
of this POST request. may also be the REQUEST_IS_BODY
value below to indicate the whole message is the body.
upload_config: (ApiUploadInfo) Information about the upload
configuration supported by this method.
supports_download: (boolean) If True, this method supports
downloading the request via the `alt=media` query
parameter.
api_version_param: API version system parameter for this
method.
r r r r r Tr r � � � �
� � )�default� �
F� N)r r r r r r �
relative_pathZ flat_path� method_id� http_method� path_params� query_paramsZordered_params� description�request_type_name�response_type_name�
request_field� MessageFieldr �
upload_configr Zsupports_download�api_version_paramr r r r! r
M s
r
z <request>c C sH | � d�r$| �d�\}}}tt|�S d| vr6t|| �S t�d| ��d S )Nzmessage_types.�.zUnknown class %s)�
startswith� partition�getattrr r �GeneratedClientError)�name�messages_module�_� classnamer r r! �
_LoadClass~ s
rA c C sJ |D ]@}|� � }t| d| �r(t| |�sd|t| �jf }t�|��qd S )Nz%sz'No %s specified for object of class %s.)�upper�hasattrr; �typer r r< )�obj�attrs�attr� attr_name�msgr r r! �_RequireClassAttrs� s
�rJ c C s | � d�s| d7 } | S )N�/)�endswith)Z api_endpointr r r! r � s
r c C sZ |� d�s|� d�r"tj�| |�S | �d�r0| n| d }|� d�rN|dd� n|}|| S )z8Custom urljoin replacement supporting : before / in url.zhttp://zhttps://rK r N)r9 r �parse�urljoinrL )�base�url�new_base�new_urlr r r! �_urljoin� s
rS c @ sX e Zd ZdZddd�Zedd� �Zedd� �Zej d d� �Zed
d � �Z
ed d
� �Z dS )� _UrlBuilderz"Convenient container for url data.Nc C st t j�t||pd��}|jr,t�d|j ��t j�|jp:d�| _ |d urT| j �
|� |j | _ |j
| _|jpld| _d S )Nr' zUnexpected url fragment: %s)r rM �urlsplitrS �fragmentr �ConfigurationValueError�parse_qs�queryr0 �update�scheme�_UrlBuilder__scheme�netloc�_UrlBuilder__netloc�pathr, )�self�base_urlr, r0 �
componentsr r r! �__init__� s �� z_UrlBuilder.__init__c C sL t j�|�}t j�|j�}t j�|j|jdd d f�}|jp<d}| |||d�S )Nr' )r, r0 ) r rM rU rX rY �
urlunsplitr[ r] r_ )�clsrP �urlpartsr0 ra r, r r r! �FromUrl� s �
�z_UrlBuilder.FromUrlc C s t j�| j| jdddf�S )Nr' )r rM rd r\ r^ �r` r r r! ra � s �z_UrlBuilder.base_urlc C s@ t j�|�}|js |js |jr,t�d| ��|j| _ |j
| _ d S )NzInvalid base url: %s) r rM rU r_ rY rV r rW r[ r\ r] r^ )r` �valuerb r r r! ra � s �c C s t j�| jd�S )NT)r rM � urlencoder0 rh r r r! rY � s z_UrlBuilder.queryc C sB d| j v sd| j v r$t�d| j ��tj�| j| j| j | jdf�S )N�{�}z'Cannot create url with relative path %sr' ) r, r rW r rM rd r\ r^ rY rh r r r! rP � s ��z_UrlBuilder.url)NN) r r r r rc � classmethodrg �propertyra �setterrY rP r r r r! rT � s
rT c C s dS )z0Hook for skipping credentials. For internal use.Fr r r r r! �_SkipGetCredentials� s rp c @ sL e Zd ZdZdZdZdZdZdZg Z dZ
d:dd �Z d
d � Z e
d d
� �Zedd� �Zedd� �Zedd� �Ze
dd� �Zedd� �Zedd� �Zedd� �Zd d � Zed d � �Zejd d!� �Zed"d#� �Z ejd$d%� �Z ed&d'� �Z e j d(d'� �Z ed)d*� �Z e j d+d*� �Z ejd,d-� �Z!d.d/� Z"d0d1� Z#d2d3� Z$d4d5� Z%d6d7� Z&d8d9� Z'dS );r z Base class for client libraries.Nr' TFr �< c C s� t | d� | d ur t�| | j� | | _|| _|| _d| _d| _|| _ | | _
|| _ |oZt � }|r||s||
pji }
| j
f i |
�� t|�| _|p�t�� | _| j d ur�| j �| j�| _d | _| p�i | _|
| _|| _|| _d| _|}d| _d S )N)Z_packageZ_scopesr> r rq F�proto)rJ r � Typecheck� params_type�%_BaseApiClient__default_global_params� log_request� log_response�_BaseApiClient__num_retries� _BaseApiClient__max_retry_wait� num_retries�max_retry_wait� _credentialsrp �_SetCredentialsr �_urlr ZGetHttp�_http� authorize� _BaseApiClient__include_fields�additional_http_headers�check_response_func�
retry_func�response_encoding�(overwrite_transfer_urls_with_client_base�#_BaseApiClient__response_type_model)r` rP � credentials�get_credentials�http�modelrv rw rz r{ Zcredentials_argsZdefault_global_paramsr� r� r� r� r? r r r! rc � s6
zBaseApiClient.__init__c K sL | j | | j| j| j| j| jd�}|�|� ddlm} |j f i |��| _
dS )aF Fetch credentials, and set them for this client.
Note that we can't simply return credentials, since creating them
may involve side-effecting self.
Args:
**kwds: Additional keyword arguments are passed on to GetCredentials.
Returns:
None. Sets self._credentials.
)�api_key�client� client_id�
client_secret� package_name�scopes�
user_agentr )�credentials_libN) �_API_KEY�
_CLIENT_ID�_CLIENT_SECRET�_PACKAGE�_SCOPES� _USER_AGENTrZ �apitools.base.pyr� ZGetCredentialsr| )r` �kwds�argsr� r r r! r} s
�
z BaseApiClient._SetCredentialsc C s& | j | jd�tt�| j���| jd�S )N� )r� r� �scoper� )r� r� �join�sortedr ZNormalizeScopesr� r� �re r r r! �
ClientInfo= s
�zBaseApiClient.ClientInfoc C s d S �Nr rh r r r! �base_model_classF s z BaseApiClient.base_model_classc C s | j S r� )r rh r r r! r� J s zBaseApiClient.httpc C s | j S r� )r~ rh r r r! rP N s zBaseApiClient.urlc C s | j S r� )r� r� r r r! � GetScopesR s zBaseApiClient.GetScopesc C s t d| j�S )N�StandardQueryParameters)rA �MESSAGES_MODULErh r r r! rt V s zBaseApiClient.params_typec C s | j S r� )r� rh r r r! r� Z s zBaseApiClient.user_agentc C s | j d u r| �� | _ | j S r� )ru rt rh r r r! �_default_global_params^ s
z$BaseApiClient._default_global_paramsc C s | j }t|||� d S r� )r� �setattr)r` r= ri �paramsr r r! �AddGlobalParame s z BaseApiClient.AddGlobalParamc C s t �| j�S r� )r ZCopyProtoMessager� rh r r r! �
global_paramsi s zBaseApiClient.global_paramsc c s || _ d V d | _ d S r� )r� )r` �include_fieldsr r r! �
IncludeFieldsm s zBaseApiClient.IncludeFieldsc C s | j S r� )r� rh r r r! �response_type_models s z!BaseApiClient.response_type_modelc c s | j }d| _dV || _dS )z2In this context, return raw JSON instead of proto.�jsonN)r� r� )r` Z old_modelr r r! �JsonResponseModelw s z BaseApiClient.JsonResponseModelc C s | j S r� )rx rh r r r! rz s zBaseApiClient.num_retriesc C s* t �|tj� |dk r t�d��|| _d S )Nr z*Cannot have negative value for num_retries)r rs �six�
integer_typesr �InvalidDataErrorrx �r` ri r r r! rz � s �c C s | j S r� )ry rh r r r! r{ � s z BaseApiClient.max_retry_waitc C s* t �|tj� |dkr t�d��|| _d S )Nr z)max_retry_wait must be a postiive integer)r rs r� r� r r� ry r� r r r! r{ � s �c c s | j }|| _ d V || _ d S r� )rz )r` rz Zold_num_retriesr r r! � WithRetries� s zBaseApiClient.WithRetriesc C s | j rt�d|j|j|� |S )z$Hook for pre-processing of requests.z Calling method %s with %s: %s)rv �logging�infor- r2 )r` �
method_config�requestr r r! �ProcessRequest� s �z BaseApiClient.ProcessRequestc C sd |j �| j� | jr`t�d|j|j� t�dt� |j �� |j
rVt�d|j pP|j
� n
t�d� |S )z)Hook for pre-processing of http requests.zMaking http %s to %sz Headers: %szBody:
%sz Body: (none)) �headersrZ r� rv r� r� r. rP �pprint�pformat�bodyZ
loggable_body)r` � http_requestr r r! �ProcessHttpRequest� s �
�
z BaseApiClient.ProcessHttpRequestc C s | j rt�d|j|� |S )NzResponse of type %s: %s)rw r� r� r3 )r` r� �responser r r! �ProcessResponse� s
�z BaseApiClient.ProcessResponsec C s t j|| jd�S )N)r� )r Z
MessageToJsonr� )r` �messager r r! �SerializeMessage� s �z BaseApiClient.SerializeMessagec
C sZ zt �||�}W nD tjtjtfyT } z"t�d||j|f ��W Y d}~n
d}~0 0 |S )z:Deserialize the given data as method_config.response_type.z+Error decoding response "%s" as type %s: %sN)r Z
JsonToMessager ZInvalidDataFromServerErrorr �ValidationError�
ValueErrorr )r` �
response_type�datar� �er r r! �DeserializeMessage� s � ��z BaseApiClient.DeserializeMessagec C sF t �|�}t| jdd�r&| jj|jd< | jr@t �| j�}|j|_|j S )z?Modify the url for a given transfer, based on auth and version.�keyN)
rT rg r; r� r� r0 r� r~ ra rP )r` rP � url_builderZclient_url_builderr r r! �FinalizeTransferUrl� s
z!BaseApiClient.FinalizeTransferUrl)NTNNFFr rq NNNNNN)(r r r r r� r� r� r� r� r� r� rc r} rm r� rn r� r� rP r� rt r� r� r� r� �
contextlib�contextmanagerr� r� r� rz ro r{ r� r� r� r� r� r� r� r r r r! r � sr �
+
r c @ s� e Zd ZdZdd� Zedd� �Zedd� �Zdd � Ze d
d � �Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd-dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd.d'd(�Zd/d)d*�Zd0d+d,�ZdS )1r
z&Base class for generated API services.c C s || _ i | _i | _d S r� )�_BaseApiService__client�_method_configs�_upload_configs)r` r� r r r! rc � s zBaseApiService.__init__c C s | j S r� �r� rh r r r! �_client� s zBaseApiService._clientc C s | j S r� r� rh r r r! r� � s zBaseApiService.clientc C s` | j �|�}|r|S t| |d�}|du r0t|��t|dd�}|du rLt|��|� | j |< }|S )z6Returns service cached method config for given method.Nr� )r� �getr; �KeyError)r` �methodr� �func�configr r r! �GetMethodConfig� s z BaseApiService.GetMethodConfigc C s dd� t �| j�D �S )Nc S s g | ]}t |d d�r|j�qS )r� N)r; r )�.0�fr r r! �
<listcomp>� s �z1BaseApiService.GetMethodsList.<locals>.<listcomp>)r� �
itervalues�__dict__r� r r r! �GetMethodsList� s z BaseApiService.GetMethodsListc C s | j �|�S r� )r� r� )r` r� r r r! �GetUploadConfig� s z BaseApiService.GetUploadConfigc C s | � |�}t| jj|j�S r� )r� r; r� r� r2 �r` r� r� r r r! �GetRequestType� s
�z BaseApiService.GetRequestTypec C s | � |�}t| jj|j�S r� )r� r; r� r� r3 r� r r r! �GetResponseType s
�z BaseApiService.GetResponseTypec C s~ t �|td�| jjf� | j�� }|p.| j�� }|�� D ]@}|�|j�}|du r\|�|j�}|dg dfvr8t||j|� q8|S )z+Combine the given params with the defaults.Nr ) r rs rD r� rt �
all_fields�get_assigned_valuer= r� )r` r� Zdefault_params�result�fieldri r r r! Z__CombineGlobalParams s
z$BaseApiService.__CombineGlobalParamsc C s, |� dd�sd|d< |� dd�s(d|d< |S )N� prettyPrintTr �pp)�pop)r` �
query_infor r r! Z__EncodePrettyPrint s
z"BaseApiService.__EncodePrettyPrintc C sb t |tj�r |dur t�|�S t |tj�r4|�d�S t |tj�rJ|� d�S t |t
j
�r^|� � S |S )zAEncode value for the URL, using field to skip encoding for bytes.N�utf8) �
isinstancer �
BytesField�base64�urlsafe_b64encoder� � text_type�encode� binary_type�decode�datetime� isoformat)r` ri r� r r r! Z__FinalUrlValue s
z BaseApiService.__FinalUrlValuec s� �� � �jj�� t�dd� �jj�� D ��jj�}t� ��t� ��fdd�|D ��}t�|t���}t���|� ���fdd�|D �� tdd� |�
� D ��}�� |�}t� |t���}|S )z<Construct a dictionary of query parameters for this request.c S s g | ]
}|j �qS r )r= )r� �xr r r! r� 2 � z9BaseApiService.__ConstructQueryParams.<locals>.<listcomp>c 3 s* | ]"}|�� t� |�t�|��fV qd S r� �� _BaseApiService__FinalUrlValuer; �r� �param)r� �global_params_typer` r r! � <genexpr>5 s � ��z8BaseApiService.__ConstructQueryParams.<locals>.<genexpr>c 3 s, | ]$}|�� t� |d �t�|��fV qd S r� r� r� )r� � request_typer` r r! r = s ���c s s" | ]\}}|d ur||fV qd S r� r )r� �k�vr r r! r B s
�)
�$_BaseApiService__CombineGlobalParamsr� r� r �
MapParamNamesrt r� rD �dictrZ �items�"_BaseApiService__EncodePrettyPrint�MapRequestParams)r` r0 r� r� Zglobal_param_namesr� Zquery_param_namesr )r� r� r� r r` r! Z__ConstructQueryParams, s( ����
z%BaseApiService.__ConstructQueryParamsNc sH t �|jt� ��}t� fdd�|D ��}t �|t� ��}t j|||d�S )z(Determine the relative path for request.c s g | ]}|t � |d �f�qS r� )r; r� �r� r r! r� M s �z:BaseApiService.__ConstructRelativePath.<locals>.<listcomp>�r, )r r r/ rD r r ZExpandRelativePath)r` r� r� r, Zpython_param_namesr� r r
r! Z__ConstructRelativePathH s
� ��z&BaseApiService.__ConstructRelativePathc C sL |j dkr@t|j�tkr@d|_ d|jd< d|jd< |j|_i |_|j|_dS )z2Make any final general adjustments to the request.�GET�POSTzx-http-method-overridez!application/x-www-form-urlencoded� content-typeN)r. �lenrP �_MAX_URL_LENGTHr� rY r� r0 )r` r� r� r r r! Z__FinalizeRequestS s
�
��z BaseApiService.__FinalizeRequestc C s� |j tjtjtjfvr(tjj|||d��|j tjkrHtj |j
d|j d�}|j }| j
jrnt|t�rn|�| j
j�}| jjdkr~|S t|j| jj�}| j�||�S )z Process the given http response.)r� r� z{})r� �content� request_urlr� )� status_coder �OK�CREATED�
NO_CONTENTr � HttpError� FromResponser �Responser� r r r� r� r� �bytesr� r� r� rA r3 r� r� )r` r� �
http_responser� r r� r r r! Z__ProcessHttpResponse_ s* �� � �z$BaseApiService.__ProcessHttpResponsec C s, |j pd}||jd< d|jd< d|jd< dS )z*Fill in the basic headers on http_request.zapitools-client/1.0z
user-agent�application/jsonr z
gzip, deflatezaccept-encodingN)r� r� )r` r� r� r� r r r! Z__SetBaseHeadersw s
z BaseApiService.__SetBaseHeadersc C s |j r|j |jd< dS )z;Fill in the system parameters to always set for the method.zX-Goog-Api-VersionN)r7 r� )r` r� r� r r r! Z__SetBaseSystemParams� s �z$BaseApiService.__SetBaseSystemParamsc C s� |j s
dS t|j| jj�}|j tkr.|}|}n,t||j �}|�|j �}t� |t
j � |j }|pb|� }|rp|spdS t� ||� d|j
d< | j�|�|_dS )z!Fill in the body on http_request.Nr r )r4 rA r2 r� r� �REQUEST_IS_BODYr; �
field_by_namer rs r r5 rD r� r� r� ) r` r� r� r� �uploadr Z
body_valueZ body_typeZ
body_fieldr r r! Z __SetBody� s(
�
�
zBaseApiService.__SetBodyc
C s� t |j| jj�}t�||� | j�||�}tj|j d�}| �
|| j� | � ||� | � ||||� t
| jj|jd�} | �|j||�| _|dur�|�||| � |dur�|�|| � | j||| jd�| _| �|| � | j�|�S )z$Prepares an HTTP request to be sent.)r. r N)rA r2 r� r� r rs r� r �Requestr. � _BaseApiService__SetBaseHeaders�$_BaseApiService__SetBaseSystemParams�_BaseApiService__SetBodyrT rP r, �%_BaseApiService__ConstructQueryParamsr0 ZConfigureRequest�&_BaseApiService__ConstructRelativePath� _BaseApiService__FinalizeRequestr� )
r` r� r� r� r r6 �downloadr r� r� r r r! �PrepareHttpRequest� s4
� �
�� � z!BaseApiService.PrepareHttpRequestc C s� |dur|durt �d��| �||||||�}|durJ|j|| jd� dS d}|durf|j|| jd�}|du r�| jj} |r�|jr�|j} | jj | jj
d�}
| jj r�| jj |
d< | jj r�| jj |
d< t
j| |fi |
��}| �|||�S )z Call this method with request.Nz/Cannot yet use both upload and download at once)r� )Zretriesr{ r� r� )r ZNotYetImplementedErrorr( ZInitializeDownloadr� ZInitializeUploadr� r� Z
bytes_httprz r{ r� r� r Z MakeRequest�ProcessHttpResponse) r` r� r� r� r r6 r' r� r r� �optsr r r! �
_RunMethod� sB �
��
� ��zBaseApiService._RunMethodc C s | j �|| �|||��S )z6Convert an HTTP response to the expected message type.)r� r� �$_BaseApiService__ProcessHttpResponse)r` r� r r� r r r! r) � s �z"BaseApiService.ProcessHttpResponse)N)NNNN)NNNN)N)r r r r rc rn r� r� r� rm r� r� r� r� r r r� r$ r% r& r, r! r" r# r( r+ r) r r r r! r
� s<
�
�
�
,r
)$r r� r� r� r� r� r� � six.movesr r �apitools.base.protorpcliter r r� r r r r �__all__� getLogger�setLevel�ERRORr �Messager r
r rA rJ r rS �objectrT rp r r
r r r r! �<module> s:
.
: q