�
!V1fi � �� � d dl Z d dlmZ d dlmZ d dlmZmZmZm Z d dl
m Z m Z m
Z
mZmZ d dlmZ e j$ ej& e� � Zddd d
d d �Ze
j. j0 e
j. j2 e
j. j4 d
�Z G d� de� Z dee fd�Z ed�� ed� deee ee f fd�� � Z ddee de j@ fd�Z!y)� N)�Enum)� lru_cache)�Dict�Optional�Tuple�Type)�clouds�
exceptions�messages�system�util)� apply_config_settings_override�AWSz AWS ChinazAWS Gov�Azure�GCP��awsz aws-chinazaws-gov�azure�gce)r r r c � � e Zd ZdZdZy)�NoCloudTypeReasonr � N)�__name__�
__module__� __qualname__�NO_CLOUD_DETECTED�CLOUD_ID_ERROR� � �:/usr/lib/python3/dist-packages/uaclient/clouds/identity.pyr r s
� ����Nr r �returnc � � t j g d�� \ } }| j � S # t j $ r Y nw xY wt
j
d� y)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�_errs r �get_instance_idr, sN � �
��K�K� F�G� ��T��y�y�{��� � +� +�
� �
���K�K�9�:� s �), �A�A)�maxsize�
cloud_typec �J � t j d� r, t j dg� \ } }| j � d fS d t j fS # t j
$ r;}t j dt |� � d t j fcY d }~S d }~ww xY w)Nzcloud-idzerror running cloud-id: %s) r �whichr% r& r
r' r( �debug�strr r r )r* r+ �excs r �get_cloud_typer4 + s� � � �|�|�J� � <� � � �Z�L�1�I�C���I�I�K��&� &�
�#�5�5� 6�6�� �/�/� <� �I�I�2�C��H� =��+�:�:�;� ;�� <�s �*A �B"�'0B �B"� B"�cloud_overridec �~ � ddl m}m}m} |j |j |j |j
|j d�}| �| }n
t � \ }}|st j � �|j |� }|st j |�� � |� }|j st j � �|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 �gcpr )r. )�uaclient.cloudsr r r7 �UAAutoAttachAWSInstance�UAAutoAttachAzureInstance�UAAutoAttachGCPInstancer4 r
�CloudFactoryNoCloudError�get�NonAutoAttachImageError� is_viable� CloudFactoryNonViableCloudError) r5 r r r7 �cloud_instance_mapr. �_�cls�instances r �cloud_instance_factoryrE : s� � � 0�/� �*�*��0�0��.�.��0�0��*�*� �� �!�#�
�&�(�
�
�A� ��1�1�3�3�
�
�
��
,�C� ��0�0�J�G�G��u�H� � � ��8�8�:�:� �Or )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 r2 r, r4 �AutoAttachCloudInstancerE r r r �<module>rU s �� � � � .� .� ?� ?� :��g���:��:�:�8�D�E�� � ��
� � � � �=�=� -� -�
�]�]�
1�
1� �=�=� -� -����� �
��#� � � �4�� � �-�
7��h�s�m�X�6G�-H� H�I�
7� .� �
7� %)�#� �S�M�#� �#�#�#r |