o =Wbv @sddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z ddlmZddd d Zd d d dddZgdZGdddeZdeefddZeddeddeeeeeffddZde jfddZdS)N)Enum) lru_cache)DictOptionalTupleType)clouds exceptionsutil)apply_config_settings_overrideazureawsgcp)ZazurenetZec2gceZAWSz AWS ChinazAWS GovZAzureZGCP)r aws-chinaaws-govr rr r rc@seZdZdZdZdS)NoCloudTypeReasonrN)__name__ __module__ __qualname__NO_CLOUD_DETECTEDCLOUD_ID_ERRORrr:/usr/lib/python3/dist-packages/uaclient/clouds/identity.pyrsrreturncCs@ztgd\}}|WStjyYnwtddS)z%Query cloud instance-id from cmdline.)z cloud-initZqueryZ instance_idz'Unable to determine current instance-idN)r subpstripr ProcessExecutionErrorloggingZwarningoutZ_errrrrget_instance_ids  r#)maxsize cloud_typecCsRtdr$ztdg\}}|dfWStjy#dtjfYSwdtjfS)Nzcloud-id) r whichrrr rrrrr!rrrget_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 rZUAAutoAttachAWSInstanceZUAAutoAttachAzureInstanceZUAAutoAttachGCPInstancer'r ZCloudFactoryNoCloudErrorgetZ!CloudFactoryUnsupportedCloudErrorZ is_viableZCloudFactoryNonViableCloudError)r r rZcloud_instance_mapr%_clsinstancerrrcloud_instance_factory6s"      r,)r enumr functoolsrtypingrrrrZuaclientrr r Zuaclient.configr ZDATASOURCE_TO_CLOUD_IDZCLOUD_TYPE_TO_TITLEZ PRO_CLOUDSrstrr#r'ZAutoAttachCloudInstancer,rrrrs(     "