U lHJe @sddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z mZmZddlmZeeeZddd d d d Ze jje jje jjd ZGdddeZeedddZeddedeeeeefdddZe j dddZ!dS)N)Enum) lru_cache)DictOptionalTupleType)clouds exceptionsmessagessystemutil)apply_config_settings_overrideZAWSz AWS ChinazAWS GovZAzureZGCPawsz aws-chinazaws-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 rrdebugstrrrr)r r!excrrrget_cloud_type+s  r(cCsvddlm}m}m}|j|j|j|j|jd}t\}}|sDt | |}|s^tj |d|}|j srt |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)rrgcpr)r$)Zuaclient.cloudsrrr)ZUAAutoAttachAWSInstanceZUAAutoAttachAzureInstanceZUAAutoAttachGCPInstancer(r ZCloudFactoryNoCloudErrorgetZNonAutoAttachImageErrorZ is_viableZCloudFactoryNonViableCloudError)rrr)Zcloud_instance_mapr$_clsinstancerrrcloud_instance_factory:s"    r.)"Zloggingenumr functoolsrtypingrrrrZuaclientrr r r r Zuaclient.configr Z getLoggerZreplace_top_level_logger_namerrZCLOUD_TYPE_TO_TITLEZurlsZPRO_ON_AWS_HOME_PAGEZPRO_ON_AZURE_HOME_PAGEZPRO_ON_GCP_HOME_PAGEZPRO_CLOUD_URLSrr&r"r(ZAutoAttachCloudInstancer.rrrrs,     "