Name	Size	Mode	Actions
aws.cpython-310.pyc	5377	0644	editdlrm
azure.cpython-310.pyc	2888	0644	editdlrm
gcp.cpython-310.pyc	4542	0644	editdlrm
identity.cpython-310.pyc	3123	0644	editdlrm
lxd.cpython-310.pyc	2545	0644	editdlrm
__init__.cpython-310.pyc	2582	0644	editdlrm

o �K�i(�@s�dd
lZdd
l
Z
dd
lZdd
lZddlmZmZmZmZ
ddl m Z mZmZm Z mZ
ddlmZ
e�e�e�
�
ZdZdZdZdZd Zd Zdg
dd gddgddgddgd�ZGdd�de�Zd
S)�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 Engine�8045211386737108299�6022427724719891830�7427070211152946628�599959289349842382�7818678586103571931�2592866803419978320�160084055690847662�2176054482269786025�4168323874790319525)�xenial�bionic�focal�jammy�noblec@s�eZ
dZd
d�Zeejejgd�
d�de e effdd��
�
Zede fdd ��
Z edefd d��
Zdee fdd �Zdefdd�Zdedefdd�ZdS)�GCPAutoAttachInstancec

Cs d|_dS)
N�
�etag�
�self�r�5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__%s zGCPAutoAttachInstance.__init__)g�?�
r")
�retry_sleeps�returnc
Csttj
td
di
dd�}
|
jdkrtj�|
j�

d|
ji
S|
j� d�
}|r1d|� �vr1tj|
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�secrets� add_secretr/� json_dict�get�lowerr�GCPServiceAccountError�CloudMetadataError)r�response� error_descrrr �identity_doc,s
� 




�
�z"GCPAutoAttachInstance.identity_docc


Csd
S)N�gcprrrrr � cloud_type?sz GCPAutoAttachInstance.cloud_typec
Cs*tj
�t�
rt�t�
}
t|
��krd
SdS)z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr � load_file�GCP_PRODUCT_NAME�strip)r�product_namerrr � is_viableCs 


zGCPAutoAttachInstance.is_viablec
CsN|jd
}
t
�|
�d�
dd�
}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�split�json�loads�decoder5)r�token�identity� identity_dictrrr �get_licenses_from_identityMs 



�z0GCPAutoAttachInstance.get_licenses_from_identityc
Cs$t�
�j}
|
tv
rt�d
�

dSdS)Nz0This series isn't supported for GCP auto-attach.FT)r �get_release_info�series�GCP_LICENSES�LOG�info)rrVrrr �should_poll_for_pro_license^s




z1GCPAutoAttachInstance.should_poll_for_pro_license�wait_for_changec

Cs�t}|
r|t
7}|jr|tj|jd
�
7}tj|ddi
d�}|jdkrHdd�|jD�
}|j � d�
|_t��j }t� |g�D] }||vrE
d Sqosz@GCPAutoAttachInstance.is_pro_license_present..rTFi�
)�LICENSES_URL�WAIT_FOR_CHANGEr� LAST_ETAG�formatrr0r.� json_listr'r5r rUrVrWrX�errorr/r�CancelProLicensePolling�DelayProLicensePolling)rr[�urlr9�license_idsrV�licrrr �is_pro_license_presentes&










�

z,GCPAutoAttachInstance.is_pro_license_presentN)�__name__� __module__�__qualname__r!�propertyr �retryrr8r�strrr;r=�boolrFrrTrZrkrrrr r$s




 r)rKrN�loggingr>�typingrrrr�uaclientrrrr r �uaclient.cloudsr� getLogger�replace_top_level_logger_namerlrXr1r`rarbrArCrWrrrrr �
s.




��





�