def test_find_latest_image(self):
def get(url):
if "raring" in url:
return MockResponse(self.data)
else:
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), "data", "releases.txt")) as f:
return MockResponse(f.read())
self.assertEqual("ami-7d317314", find_image("us-east-1", _session=MockSession(get)).ami_id)
def main():
parser = argparse.ArgumentParser(
"ubuntufinder", description="An utility tool to find the latest Ubuntu EC2 AMI IDs"
)
parser.add_argument("region", help="AWS Region to return an image ID for.")
parser.add_argument("-r", "--release", default=LATEST, help="Release codename. Defaults to the latest release.")
parser.add_argument(
"-a", "--architecture", default="amd64", help="Architecture. `amd64` or `i386`. Defauts to `amd64`."
)
parser.add_argument(
"-i", "--image-type", default="ebs", help="AWS image type. `ebs` or `instance-store`. Defaults to `ebs`."
)
parser.add_argument(
"-v",
"--virtualization",
default="paravirtual",
help="Virtualization type. `paravirtual` or `hvm`. Defaults to `paravirtual`.",
)
parser.add_argument("--instance-type") # Legacy argument name
args = parser.parse_args()
# Legacy argument handling
if args.instance_type is not None:
print("WARNING: `--instance-type` is deprecated. Use `--image-type` instead", file=sys.stderr)
image_type = args.instance_type
else:
image_type = args.image_type
# Actual image location and exit logic
try:
image = find_image(args.region, args.release, args.architecture, image_type, args.virtualization)
except ImageNotFound:
# 1-10: user error
print("ERROR: No image found!", file=sys.stderr)
sys.exit(1)
except LatestReleaseNotFound:
# 11-20: program error
print("ERROR: Latest release could not be identified", file=sys.stderr)
sys.exit(11)
except ServiceUnavailable:
# 21-30: temporary failure
print("ERROR: Ununtu cloud images are currently unavailable", file=sys.stderr)
sys.exit(21)
except:
# 31-40: internal error
traceback.print_exc(file=sys.stderr)
sys.exit(31)
print(image.ami_id)
def test_find_latest_image(self):
def get(url):
if "raring" in url:
return MockResponse(self.data)
else:
with open(
os.path.join(
os.path.dirname(os.path.abspath(__file__)), "data",
"releases.txt")) as f:
return MockResponse(f.read())
self.assertEqual(
"ami-7d317314",
find_image("us-east-1", _session=MockSession(get)).ami_id)
def main():
parser = argparse.ArgumentParser(
"ubuntufinder",
description="An utility tool to find the latest Ubuntu EC2 AMI IDs")
parser.add_argument("region", help="AWS Region to return an image ID for.")
parser.add_argument(
"-r",
"--release",
default=LATEST,
help="Release codename. Defaults to the latest release.")
parser.add_argument(
"-a",
"--architecture",
default="amd64",
help="Architecture. `amd64` or `i386`. Defauts to `amd64`.")
parser.add_argument(
"-i",
"--image-type",
default="ebs",
help="AWS image type. `ebs` or `instance-store`. Defaults to `ebs`.")
parser.add_argument(
"-v",
"--virtualization",
default="paravirtual",
help=
"Virtualization type. `paravirtual` or `hvm`. Defaults to `paravirtual`."
)
parser.add_argument('--instance-type') # Legacy argument name
args = parser.parse_args()
# Legacy argument handling
if args.instance_type is not None:
print(
"WARNING: `--instance-type` is deprecated. Use `--image-type` instead",
file=sys.stderr)
image_type = args.instance_type
else:
image_type = args.image_type
# Actual image location and exit logic
try:
image = find_image(args.region, args.release, args.architecture,
image_type, args.virtualization)
except ImageNotFound:
# 1-10: user error
print("ERROR: No image found!", file=sys.stderr)
sys.exit(1)
except LatestReleaseNotFound:
# 11-20: program error
print("ERROR: Latest release could not be identified", file=sys.stderr)
sys.exit(11)
except ServiceUnavailable:
# 21-30: temporary failure
print("ERROR: Ununtu cloud images are currently unavailable",
file=sys.stderr)
sys.exit(21)
except:
# 31-40: internal error
traceback.print_exc(file=sys.stderr)
sys.exit(31)
print(image.ami_id)