File: //usr/lib/python3/dist-packages/uaclient/clouds/__pycache__/gcp.cpython-310.pyc
o
0xvct � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
d dl m Z m
Z
mZmZ d dlmZ e�d�ZdZdZd Zd
Zd Zd Zd
dddd�ZG dd� de�ZdS )� N)�Any�Dict�List�Optional)� HTTPError)�
exceptions�messages�system�util)�AutoAttachCloudInstancezpro.clouds.gcpz�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 s� zddi}t jt|dd�\}}W d
|iS typ } zQt|dd �}d }|rCz
tj|t jd�}W n ty: d }Y nw |rC|� dd �}|rG|n|j
}d } |r`d|� � v r`t j
jj|d �}t j
j} tj|| t|d
d �d ��d }~ww )N�Metadata-Flavor�Googler )�headers�timeout�body)�cls�error_descriptionzservice account)� error_msg�coder )�msg�msg_coder' �
identityToken)r
�readurl� TOKEN_URLr �getattr�json�loads�DatetimeAwareJSONDecoder�
ValueError�get�reason�lowerr �"GCP_SERVICE_ACCT_NOT_ENABLED_ERRORr( �format�namer �GCPProAccountError)
r r! � url_response�_headers�er# �
error_desc�
error_detailsr( r) r r r � identity_doc, s> ��
� � ����z$UAAutoAttachGCPInstance.identity_docc C s dS )N�gcpr r r r r �
cloud_typeL 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_viableP 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)r> �base64�urlsafe_b64decode�splitr. r/ �decoder2 )r �token�identity�
identity_dictr r r �get_licenses_from_identityZ s
�z2UAAutoAttachGCPInstance.get_licenses_from_identityc C s& t �� d }|tvrt�d� dS dS )N�seriesz0This series isn't supported for GCP auto-attach.FT)r �get_platform_info� GCP_LICENSES�LOG�info)r rV r r r �should_poll_for_pro_licensek s
z3UAAutoAttachGCPInstance.should_poll_for_pro_license�wait_for_changec
C s� t }|r|t7 }| jr|tj| jd�7 }z
tj|ddid�\}}W n ty? } zt� |� |j
dkr7t � � �t �
� �d }~ww dd� |D �}|�dd �| _t�� d }t�|�|v S )
Nr r r )r! i� c S s g | ]}|d �qS )�idr )�.0�licenser r r �
<listcomp>� s zBUAAutoAttachGCPInstance.is_pro_license_present.<locals>.<listcomp>�ETagrV )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAGr6 r
r+ r rY �errorr' r �CancelProLicensePolling�DelayProLicensePollingr2 r rW rX )r r\ �url�licensesr! r; � license_idsrV r r r �is_pro_license_presentr s( �
�� z.UAAutoAttachGCPInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__r �propertyr
�retryr r8 r �strr r> r@ �boolrI r rU r[ rk r r r r r $ s r ) rN r. �loggingrA �typingr r r r � urllib.errorr �uaclientr r r r
�uaclient.cloudsr � getLoggerrY r, rb rc rd rD rF rX r r r r r �<module> s.
���