CasperSecurity
o
��Jh� � @ s
d dl Z d dlmZ d dlmZ d dlmZmZmZm Z d dl
m Z m Z m
Z
mZmZ d dlmZ e �e�e��Zddd d
d d �Ze
jje
jje
jjd
�ZG dd� de�Z dee fdd�Z edd�ed�deee ee f fdd���Z ddee de j fdd�Z!dS )� N)�Enum)� lru_cache)�Dict�Optional�Tuple�Type)�clouds�
exceptions�messages�system�util)� apply_config_settings_override�AWSz AWS ChinazAWS Gov�Azure�GCP)�aws� aws-china�aws-gov�azure�gce)r r r c @ s e Zd ZdZdZdS )�NoCloudTypeReasonr � N)�__name__�
__module__� __qualname__�NO_CLOUD_DETECTED�CLOUD_ID_ERROR� r r �:/usr/lib/python3/dist-packages/uaclient/clouds/identity.pyr s r �returnc C s@ zt �g d��\} }| �� W S tjy Y nw t�d� dS )z%Query cloud instance-id from cmdline.)z
cloud-init�query� instance_idz'Unable to determine current instance-idN)r �subp�stripr �ProcessExecutionError�LOG�warning)�out�_errr r r �get_instance_id s
�
r) )�maxsize�
cloud_typec
C st t �d�r5zt �dg�\} }| �� d fW S tjy4 } zt�dt|�� d t j
fW Y d }~S d }~ww d t j fS )Nzcloud-idzerror running cloud-id: %s) r �whichr" r# r r$ r% �debug�strr r r )r' r( �excr r r �get_cloud_type+ s
��
r0 �cloud_overridec
C s� ddl m}m}m}m} |j|j|j|j|j|jd�}| dur"| }nt � \}}|s-t
� � �|� |�}|s:t
j
|d��|� } | jsDt
�� �| S )a
:raises CloudFactoryError: if no cloud instance object can be constructed
:raises CloudFactoryNoCloudError: if no cloud instance object can be
constructed because we are not on a cloud
:raises CloudFactoryUnsupportedCloudError: if no cloud instance object can
be constructed because we don't have a class for the cloud we're on
:raises CloudFactoryNonViableCloudError: if no cloud instance object can be
constructed because we explicitly do not support the cloud we're on
r )r r �gcp�lxd)r r r r r r3 N)r+ )�uaclient.cloudsr r r2 r3 �AWSAutoAttachInstance�AzureAutoAttachInstance�GCPAutoAttachInstance�LXDAutoAttachInstancer0 r �CloudFactoryNoCloudError�get�NonAutoAttachImageError� is_viable� CloudFactoryNonViableCloudError)
r1 r r r2 r3 �cloud_instance_mapr+ �_�cls�instancer r r �cloud_instance_factory: s( �
rB )N)"�logging�enumr � functoolsr �typingr r r r �uaclientr r r
r r �uaclient.configr
� getLogger� replace_top_level_logger_namer r% �CLOUD_TYPE_TO_TITLE�urls�PRO_ON_AWS_HOME_PAGE�PRO_ON_AZURE_HOME_PAGE�PRO_ON_GCP_HOME_PAGE�PRO_CLOUD_URLSr r. r) r0 �AutoAttachInstancerB r r r r �<module> s8 � � "���