File: //lib/python3/dist-packages/uaclient/clouds/__pycache__/gcp.cpython-310.pyc
o
nHJe� � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
d dlmZ e�e
�e��ZdZdZdZdZd Zd
Zd d d
dd�ZG dd� de�ZdS )� N)�Any�Dict�List�Optional)�
exceptions�http�system�util)�AutoAttachCloudInstancez�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 Engine�8045211386737108299�6022427724719891830�599959289349842382�2592866803419978320)�xenial�bionic�focal�jammyc @ s� e Zd Zdd� Zeejejg d�d�de e
e f fdd���Z ede
fdd ��Z
edefd
d ��Zdee
fd d
�Zdefdd�Zdedefdd�ZdS )�UAAutoAttachGCPInstancec C s
d | _ d S )N��etag��self� r �5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__$ s
z UAAutoAttachGCPInstance.__init__)g �?� r )� retry_sleeps�returnc C sf t jtddidd�}|jdkrd|jiS |j�d�}|r*d|�� v r*tj |j|d ��tj
|j|jd
��) N�Metadata-Flavor�Googler )�headers�timeout�� �
identityToken�error_descriptionzservice account)� status_code� error_msg)�code�body) r �readurl� TOKEN_URLr' r( � json_dict�get�lowerr �GCPServiceAccountError�CloudMetadataError)r �response�
error_descr r r � identity_doc+ s
�
��z$UAAutoAttachGCPInstance.identity_docc C s dS )N�gcpr r r r r �
cloud_type= s z"UAAutoAttachGCPInstance.cloud_typec C s* t j�t�rt�t�}t|�� krdS dS )z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr � load_file�GCP_PRODUCT_NAME�strip)r � product_namer r r � is_viableA s
z!UAAutoAttachGCPInstance.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-8�google�compute_engine�
license_id)r2 �base64�urlsafe_b64decode�split�json�loads�decoder, )r �token�identity�
identity_dictr r r �get_licenses_from_identityK s
�z2UAAutoAttachGCPInstance.get_licenses_from_identityc C s$ t �� j}|tvrt�d� dS dS )Nz0This series isn't supported for GCP auto-attach.FT)r �get_release_info�series� GCP_LICENSES�LOG�info)r rM r r r �should_poll_for_pro_license\ s
z3UAAutoAttachGCPInstance.should_poll_for_pro_license�wait_for_changec C s� t }|r|t7 }| jr|tj| jd�7 }tj|ddid�}|jdkr=dd� |jD �}|j �
d�| _t � � j
}t�
|�|v S t�|j� |jd krLt�� �t�� �)
Nr r r )r r" c S s g | ]}|d �qS )�idr )�.0�licenser r r �
<listcomp>m s zBUAAutoAttachGCPInstance.is_pro_license_present.<locals>.<listcomp>r i� )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAG�formatr r) r' � json_listr r, r rL rM rN rO �errorr( r �CancelProLicensePolling�DelayProLicensePolling)r rR �urlr0 � license_idsrM r r r �is_pro_license_presentc s
z.UAAutoAttachGCPInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__r �propertyr �retryr r/ r �strr r2 r4 �boolr= r rK rQ ra r r r r r # s r ) rB rE �loggingr5 �typingr r r r �uaclientr r r r �uaclient.cloudsr
� getLogger� replace_top_level_logger_namerb rO r* rW rX rY r8 r: rN r r r r r �<module> s, ���