File: //usr/lib/python3/dist-packages/uaclient/clouds/__pycache__/azure.cpython-38.pyc
U
��Jhc � @ s� d dl Z d dlZd dlmZmZ d dlmZmZmZm Z m
Z
d dl m Z e �
e
�e��ZdZdZed e ed e d �Zd
Zd Zd Zd
ZG dd� de �ZdS )� N)�Any�Dict)�
exceptions�http�secret_manager�system�util)� PublicCloudAutoAttachInstancez http://169.254.169.254/metadata/z
2020-09-01z attested/document?api-version=z instance/compute?api-version=)�pkcs7�computez#/sys/class/dmi/id/chassis_asset_tagz%/var/lib/cloud/seed/azure/ovf-env.xmlz 7783-7084-3265-9085-8269-3286-77Z
UBUNTU_PROc @ s~ e Zd Zeejejdddgd�ee e
f d�dd���Z ee d�dd��Z ee
d�dd ��Ze
d�d
d �Ze
e
d �d
d�ZdS )�AzureAutoAttachInstance� )Z retry_sleeps)�returnc C s� i }t t�� �D ]j\}}tj|ddidd�}|jdkrHtj|j|jd��|dkrp|j d }|||< t
j � |� q|j ||< q|S ) N�Metadata�truer
)�headersZtimeout�� )�code�bodyr
� signature)
�sorted� IMDS_URLS�itemsr �readurlr r �CloudMetadataErrorr � json_dictr ZsecretsZ
add_secret)�selfZ responses�key�url�responser � r �7/usr/lib/python3/dist-packages/uaclient/clouds/azure.py� identity_doc s$ �
�
z$AzureAutoAttachInstance.identity_docc C s dS )NZazurer �r r r r! �
cloud_type/ s z"AzureAutoAttachInstance.cloud_typec C s2 t j�t�r&t�t�}t|�� kr&dS t j�t�S )z&This machine is a viable AzureInstanceT) �os�path�exists�DMI_CHASSIS_ASSET_TAGr Z load_file�AZURE_CHASSIS_ASSET_TAG�strip�AZURE_OVF_ENV_FILE)r Zchassis_asset_tagr r r! � is_viable3 s
z!AzureAutoAttachInstance.is_viablec C s dS )NTr r# r r r! �should_poll_for_pro_license< s z3AzureAutoAttachInstance.should_poll_for_pro_license)�wait_for_changer c
C s� |r t �� �t�dd�}ztj|ddid�}W n4 tk
rb } zt�|� t �� �W 5 d }~X Y nX |j dkr�t�|j
� t �� �|j �d�t kS )Nr � r r )r r Z licenseType)
r ZCancelProLicensePollingr �getr r �OSError�LOG�errorr r r �AZURE_PRO_LICENSE_TYPE)r r. r r �er r r! �is_pro_license_present@ s
z.AzureAutoAttachInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__�propertyr Zretryr r r �strr r" r$ �boolr, r- r6 r r r r! r s r )Zloggingr% �typingr r Zuaclientr r r r r Zuaclient.cloudsr Z getLoggerZ replace_top_level_logger_namer7 r2 Z
IMDS_BASE_URLZ API_VERSIONr r( r+ r) r4 r r r r r! �<module> s
�