| Current File : //usr/lib/python3/dist-packages/uaclient/clouds/__pycache__/gcp.cpython-312.pyc |
�
��g( � � � d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZ ej" ej$ e� � ZdZdZdZdZd Zd
Zd gd d
gddgddgddgd�Z G d� de� Z y)� 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 �� � e Zd Zd� Ze ej ej g d��� de e
e f fd�� � Z ede
fd�� Z
edefd�� Zdee
fd�Zdefd �Zd
edefd �Zy )
�GCPAutoAttachInstancec � � d | _ y )N��etag��selfs �5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__z GCPAutoAttachInstance.__init__% s � � �� � )g �?� r% )� retry_sleeps�returnc � � t j t ddid�� }|j dk( r7t j
j
|j � d|j iS |j j d� }|r3d|j � v r!t j |j |� � �t j |j |j �
� �) N�Metadata-Flavor�Googler% )�headers�timeout�� �
identityToken�error_descriptionzservice account)� status_code� error_msg)�code�body)r �readurl� TOKEN_URLr2 r �secrets�
add_secretr3 � json_dict�get�lowerr �GCPServiceAccountError�CloudMetadataError)r! �response�
error_descs r" � identity_docz"GCPAutoAttachInstance.identity_doc, s� � � �<�<� � 1�8� <�a�
�� �=�=�C� � � "� "� -� -�h�m�m� <�#�X�]�]�3� 3� �'�'�+�+�,?�@�
� �+�z�/?�/?�/A�A� �3�3� $�M�M�Z��
� �+�+�� � �X�]�]�
�
r$ c � � y)N�gcp� r s r" �
cloud_typez GCPAutoAttachInstance.cloud_type? s � �r$ c � � t j j t � r1t j
t � }t |j � k( ryy)z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr
� load_file�GCP_PRODUCT_NAME�strip)r! � product_names r" � is_viablez GCPAutoAttachInstance.is_viableC s= � � �7�7�>�>�*� +�!�+�+�,<�=�L� �<�#5�#5�#7�7��r$ c � � | j d }t j |j d� d dz � }t j
|j
d� � }|j di � j di � j 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�decoder9 )r! �token�identity�
identity_dicts r" �get_licenses_from_identityz0GCPAutoAttachInstance.get_licenses_from_identityM s| � � �!�!�/�2���+�+�E�K�K��,<�Q�,?�%�,G�H�� �
�
�8�?�?�7�#;�<�
� � � �h�� +�
�S�!�2�
&�
�S� �r�
"�
r$ c �| � t j � j }|t vrt j d� yy)Nz0This series isn't supported for GCP auto-attach.FT)r
�get_release_info�series� GCP_LICENSES�LOG�info)r! r_ s r" �should_poll_for_pro_licensez1GCPAutoAttachInstance.should_poll_for_pro_license^ s1 � ��(�(�*�1�1�� � � %� �H�H�G� H��r$ �wait_for_changec � � t }|r8|t z
}| j r#|t j | j �� z
}t j |ddi�� }|j dk( r||j D �cg c] }|d �� }}|j j d� | _ t j � j }t j |g � D ] }||v s� y y t j! |j" � |j d
k( rt% j& � �t% j( � �c c}w ) Nr r) r* )r+ r- �idr TFi� )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAG�formatr r4 r2 � json_listr+ r9 r
r^ r_ r` ra �errorr3 r �CancelProLicensePolling�DelayProLicensePolling)r! rd �urlr= �license� license_idsr_ �lics r" �is_pro_license_presentz,GCPAutoAttachInstance.is_pro_license_presente s � ��� � �?� "�C��y�y��y�'�'�T�Y�Y�'�7�7���<�<��.?��-J�K�� �=�=�C� �8@�8J�8J�K�W�7�4�=�K�K�K� �(�(�,�,�V�4�D�I��,�,�.�5�5�F�#�'�'���3�
���+�%� �
� � � � �(�-�-� � �=�=�C� � �4�4�6� 6� �3�3�5� 5�� Ls �7 D;N)�__name__�
__module__� __qualname__r# �propertyr �retryr r< r �strr r? rC �boolrM r r\ rc rs rB r$ r" r r $ s� � �� ��T�Z�Z�
�-�-�K�H�
�d�3��8�n�
� I� �
�" ��C� � �� ��4� � ��
�D��I�
�"�T� �6�� 6�$� 6r$ r ) rS rV �loggingrE �typingr r r r �uaclientr r r r
r �uaclient.cloudsr � getLogger� replace_top_level_logger_namert ra r5 rg rh ri rH rJ r` r rB r$ r" �<module>r� s� ��
� � � � ,� ,� C� C� 9��g���:��:�:�8�D�E�� �
� �
� *�� � �3� �*� � %�%�$�&;�<�"�$9�
:�#�%9�
:�#�%:�
;� � �W6� 9� W6r$