File: //lib/python3/dist-packages/uaclient/clouds/__pycache__/gcp.cpython-38.pyc
U
��Jh( � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZ e�e�e��ZdZdZdZdZd Zd
Zd gd d
gddgddgddgd�ZG dd� de�Z dS )� N)�Any�Dict�List�Optional)�
exceptions�http�secret_manager�system�util)� PublicCloudAutoAttachInstancez�http://metadata/computeMetadata/v1/instance/service-accounts/default/identity?audience=contracts.canonical.com&format=full&licenses=TRUEzThttp://metadata.google.internal/computeMetadata/v1/instance/licenses/?recursive=truez&wait_for_change=truez&last_etag={etag}z /sys/class/dmi/id/product_namezGoogle Compute EngineZ8045211386737108299Z6022427724719891830Z7427070211152946628Z599959289349842382Z7818678586103571931Z2592866803419978320Z160084055690847662Z2176054482269786025Z4168323874790319525)ZxenialZbionicZfocalZjammyZnoblec @ s� e Zd Zdd� Zeejejdddgd�e e
e f d�dd���Z ee
d�d d
��Z
eed�d d ��Zee
d�d
d�Zed�dd�Zeed�dd�ZdS )�GCPAutoAttachInstancec C s
d | _ d S )N��etag��self� r �5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__% s z GCPAutoAttachInstance.__init__g �?� )Z retry_sleeps)�returnc C sx t jtddidd�}|jdkr6tj�|j� d|jiS |j� d�}|rbd|�
� krbt j |j|d ��t j
|j|jd
��d S ) N�Metadata-Flavor�Googler )�headersZtimeout�� �
identityTokenZerror_descriptionzservice account)Z status_codeZ error_msg)�code�body)r �readurl� TOKEN_URLr r ZsecretsZ
add_secretr Z json_dict�get�lowerr ZGCPServiceAccountError�CloudMetadataError)r �responseZ
error_descr r r � identity_doc, s$ �
� �z"GCPAutoAttachInstance.identity_docc C s dS )NZgcpr r r r r �
cloud_type? s z GCPAutoAttachInstance.cloud_typec C s* t j�t�r&t�t�}t|�� kr&dS dS )z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr Z load_file�GCP_PRODUCT_NAME�strip)r Z product_namer r r � is_viableC s
z GCPAutoAttachInstance.is_viablec C sN | j d }t�|�d�d d �}t�|�d��}|�di ��di ��dg �S ) z�Get a list of licenses from the GCP metadata.
Instance identity token (jwt) carries a list of licenses
associated with the instance itself.
Returns an empty list if licenses are not present in the metadata.
r �.r z===zutf-8ZgoogleZcompute_engineZ
license_id)r# �base64Zurlsafe_b64decode�split�json�loads�decoder )r �tokenZidentityZ
identity_dictr r r �get_licenses_from_identityM s
� ��z0GCPAutoAttachInstance.get_licenses_from_identityc C s$ t �� j}|tkr t�d� dS dS )Nz0This series isn't supported for GCP auto-attach.FT)r �get_release_info�series� GCP_LICENSES�LOG�info)r r5 r r r �should_poll_for_pro_license^ s
z1GCPAutoAttachInstance.should_poll_for_pro_license)�wait_for_changer c C s� t }|r(|t7 }| jr(|tj| jd�7 }tj|ddid�}|jdkr�dd� |jD �}|j �
d�| _t � � j
}t�
|g �D ]}||krx d S qxd
S t�|j� |jd kr�t�� �nt�� �d S ) Nr
r r )r r c S s g | ] }|d �qS )�idr )�.0�licenser r r �
<listcomp>o s z@GCPAutoAttachInstance.is_pro_license_present.<locals>.<listcomp>r TFi� )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAG�formatr r r Z json_listr r r r4 r5 r6 r7 �errorr r ZCancelProLicensePollingZDelayProLicensePolling)r r: Zurlr" Z license_idsr5 Zlicr r r �is_pro_license_presente s$
z,GCPAutoAttachInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__r �propertyr
Zretryr r! r �strr r# r$ �boolr+ r r3 r9 rD r r r r r $ s r ) r- r/ Zloggingr% �typingr r r r Zuaclientr r r r r
Zuaclient.cloudsr Z getLoggerZ replace_top_level_logger_namerE r7 r r? r@ rA r( r) r6 r r r r r �<module> s, ���