/
lib
/
python3
/
dist-packages
/
uaclient
/
clouds
/
Upload File
HOME
from urllib.error import HTTPError try: from typing import Any, Dict # noqa: F401 except ImportError: # typing isn't available on trusty, so ignore its absence pass from uaclient.clouds import AutoAttachCloudInstance from uaclient import util IMDS_URL = "http://169.254.169.254/latest/dynamic/instance-identity/pkcs7" SYS_HYPERVISOR_PRODUCT_UUID = "/sys/hypervisor/uuid" DMI_PRODUCT_SERIAL = "/sys/class/dmi/id/product_serial" DMI_PRODUCT_UUID = "/sys/class/dmi/id/product_uuid" class UAAutoAttachAWSInstance(AutoAttachCloudInstance): # mypy does not handle @property around inner decorators # https://github.com/python/mypy/issues/1362 @property # type: ignore @util.retry(HTTPError, retry_sleeps=[1, 2, 5]) def identity_doc(self) -> "Dict[str, Any]": response, _headers = util.readurl(IMDS_URL) return {"pkcs7": response} @property def cloud_type(self) -> str: return "aws" @property def is_viable(self) -> bool: """This machine is a viable AWSInstance""" try: hypervisor_uuid = util.load_file(SYS_HYPERVISOR_PRODUCT_UUID) if "ec2" == hypervisor_uuid[0:3]: return True except FileNotFoundError: # SYS_HYPERVISOR_PRODUCT_UUID isn't present on all EC2 instance # types, fall through pass # Both DMI product_uuid and product_serial start with 'ec2' dmi_uuid = util.load_file(DMI_PRODUCT_UUID).lower() dmi_serial = util.load_file(DMI_PRODUCT_SERIAL).lower() if "ec2" == dmi_uuid[0:3] == dmi_serial[0:3]: return True return False