U Jh @sddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z mZmZddlmZeeeZddd d d d Ze jje jje jjd ZGdddeZeedddZeddedeeeeefdddZdeee j dddZ!dS)N)Enum) lru_cache)DictOptionalTupleType)clouds exceptionsmessagessystemutil)apply_config_settings_overrideZAWSz AWS ChinazAWS GovZAzureZGCP)aws aws-chinaaws-govazuregce)rrrc@seZdZdZdZdS)NoCloudTypeReasonrN)__name__ __module__ __qualname__NO_CLOUD_DETECTEDCLOUD_ID_ERRORrr:/usr/lib/python3/dist-packages/uaclient/clouds/identity.pyrsr)returncCsDztdddg\}}|WStjk r4YnXtddS)z%Query cloud instance-id from cmdline.z cloud-initZqueryZ instance_idz'Unable to determine current instance-idN)r subpstripr ProcessExecutionErrorLOGZwarning)out_errrrrget_instance_ids  r#)maxsize cloud_typec Csvtdrlztdg\}}|dfWStjk rj}z"tdt|dt j fWYSd}~XYnXdt j fS)Nzcloud-idzerror running cloud-id: %s) r Zwhichrrr rr debugstrrrr)r!r"excrrrget_cloud_type+s  r))cloud_overriderc Csddlm}m}m}m}|j|j|j|j|j|jd}|dk rD|}n t \}}|sZt | |}|stt j |d|} | jst | 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)rrgcplxd)rrrrrr,N)r%)Zuaclient.cloudsrrr+r,ZAWSAutoAttachInstanceZAzureAutoAttachInstanceZGCPAutoAttachInstanceZLXDAutoAttachInstancer)r ZCloudFactoryNoCloudErrorgetZNonAutoAttachImageErrorZ is_viableZCloudFactoryNonViableCloudError) r*rrr+r,Zcloud_instance_mapr%_clsinstancerrrcloud_instance_factory:s(     r1)N)"Zloggingenumr functoolsrtypingrrrrZuaclientrr r r r Zuaclient.configr Z getLoggerZreplace_top_level_logger_namerr ZCLOUD_TYPE_TO_TITLEZurlsZPRO_ON_AWS_HOME_PAGEZPRO_ON_AZURE_HOME_PAGEZPRO_ON_GCP_HOME_PAGEZPRO_CLOUD_URLSrr'r#r)ZAutoAttachInstancer1rrrrs4     "