o ¡K&b= ã@sÌddlZddlmZddlmZmZmZmZddlm Z m Z m Z ddl m Z dddd œZd d d d ddœZgd¢ZGdd„deƒZdeefdd„Ze dƒdeeeeeffdd„ƒZde jfdd„ZdS)éN)ÚEnum)ÚDictÚOptionalÚTupleÚType)ÚcloudsÚ exceptionsÚutil)Úapply_config_settings_overrideÚazureÚawsÚgcp)ZazurenetZec2ÚgceZAWSz AWS ChinazAWS GovZAzureZGCP)r ú aws-chinaúaws-govr r ©r r r c@seZdZdZdZdS)ÚNoCloudTypeReasonréN)Ú__name__Ú __module__Ú __qualname__ÚNO_CLOUD_DETECTEDÚCLOUD_ID_ERROR©rrú:/usr/lib/python3/dist-packages/uaclient/clouds/identity.pyrsrÚreturncCs@zt gd¢¡\}}| ¡WStjyYnwt d¡dS)z%Query cloud instance-id from cmdline.)z cloud-initZqueryZ instance_idz'Unable to determine current instance-idN)r ÚsubpÚstriprÚProcessExecutionErrorÚloggingZwarning©ÚoutZ_errrrrÚget_instance_ids ÿ r"Ú cloud_typecCsRt d¡r$zt dg¡\}}| ¡dfWStjy#dtjfYSwdtjfS)Nzcloud-id) r Úwhichrrrrrrrr rrrÚget_cloud_type's ÿ r%cCsxddlm}m}m}|j|j|j|j|jdœ}tƒ\}}|s#t  |¡‚|  |¡}|s/t  |¡‚|ƒ}|j s:t  |¡‚|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 rr)r rrr r)Zuaclient.cloudsr r r ZUAAutoAttachAWSInstanceZUAAutoAttachAzureInstanceZUAAutoAttachGCPInstancer%rZCloudFactoryNoCloudErrorÚgetZ!CloudFactoryUnsupportedCloudErrorZ is_viableZCloudFactoryNonViableCloudError)r r r Zcloud_instance_mapr#Ú_ÚclsÚinstancerrrÚcloud_instance_factory4s" û     r*)rÚenumrÚtypingrrrrZuaclientrrr Zuaclient.configr ZDATASOURCE_TO_CLOUD_IDZCLOUD_TYPE_TO_TITLEZ PRO_CLOUDSrÚstrr"r%ZAutoAttachCloudInstancer*rrrrÚs$   û