ec2_vpc - VPCException -> AnsibleVPCException

This commit is contained in:
Robert Estelle 2014-12-01 13:41:03 -05:00 committed by Matt Clay
parent bded55a84d
commit 504e1773ef

View file

@ -156,11 +156,11 @@ except ImportError:
sys.exit(1) sys.exit(1)
class RouteTableException(Exception): class AnsibleRouteTableException(Exception):
pass pass
class TagCreationException(RouteTableException): class AnsibleTagCreationException(AnsibleRouteTableException):
pass pass
@ -191,8 +191,8 @@ def ensure_tags(vpc_conn, resource_id, tags, add_only, dry_run):
latest_tags = get_resource_tags(vpc_conn, resource_id) latest_tags = get_resource_tags(vpc_conn, resource_id)
return {'changed': True, 'tags': latest_tags} return {'changed': True, 'tags': latest_tags}
except EC2ResponseError as e: except EC2ResponseError as e:
raise TagCreationException('Unable to update tags for {0}, error: {1}' raise AnsibleTagCreationException(
.format(resource_id, e)) 'Unable to update tags for {0}, error: {1}'.format(resource_id, e))
def get_route_table_by_id(vpc_conn, vpc_id, route_table_id): def get_route_table_by_id(vpc_conn, vpc_id, route_table_id):
@ -265,7 +265,7 @@ def get_subnet_by_cidr(vpc_conn, vpc_id, cidr):
subnets = vpc_conn.get_all_subnets( subnets = vpc_conn.get_all_subnets(
filters={'cidr': cidr, 'vpc_id': vpc_id}) filters={'cidr': cidr, 'vpc_id': vpc_id})
if len(subnets) != 1: if len(subnets) != 1:
raise RouteTableException( raise AnsibleRouteTableException(
'Subnet with CIDR {0} has {1} matches'.format(cidr, len(subnets)) 'Subnet with CIDR {0} has {1} matches'.format(cidr, len(subnets))
) )
return subnets[0] return subnets[0]
@ -274,8 +274,9 @@ def get_subnet_by_cidr(vpc_conn, vpc_id, cidr):
def get_subnet_by_id(vpc_conn, vpc_id, subnet_id): def get_subnet_by_id(vpc_conn, vpc_id, subnet_id):
subnets = vpc_conn.get_all_subnets(filters={'subnet-id': subnet_id}) subnets = vpc_conn.get_all_subnets(filters={'subnet-id': subnet_id})
if len(subnets) != 1: if len(subnets) != 1:
raise RouteTableException( raise AnsibleRouteTableException(
'Subnet with ID {0} has {1} matches'.format(subnet_id, len(subnets)) 'Subnet with ID {0} has {1} matches'.format(
subnet_id, len(subnets))
) )
return subnets[0] return subnets[0]
@ -333,7 +334,7 @@ def ensure_route_table_absent(vpc_conn, vpc_id, route_table_id, resource_tags,
elif resource_tags: elif resource_tags:
route_table = get_route_table_by_tags(vpc_conn, vpc_id, resource_tags) route_table = get_route_table_by_tags(vpc_conn, vpc_id, resource_tags)
else: else:
raise RouteTableException( raise AnsibleRouteTableException(
'must provide route_table_id or resource_tags') 'must provide route_table_id or resource_tags')
if route_table is None: if route_table is None:
@ -356,7 +357,7 @@ def ensure_route_table_present(vpc_conn, vpc_id, route_table_id, resource_tags,
route_table = get_route_table_by_tags(vpc_conn, vpc_id, resource_tags) route_table = get_route_table_by_tags(vpc_conn, vpc_id, resource_tags)
tags_valid = route_table is not None tags_valid = route_table is not None
else: else:
raise RouteTableException( raise AnsibleRouteTableException(
'must provide route_table_id or resource_tags') 'must provide route_table_id or resource_tags')
if check_mode and route_table is None: if check_mode and route_table is None:
@ -366,7 +367,7 @@ def ensure_route_table_present(vpc_conn, vpc_id, route_table_id, resource_tags,
try: try:
route_table = vpc_conn.create_route_table(vpc_id) route_table = vpc_conn.create_route_table(vpc_id)
except EC2ResponseError as e: except EC2ResponseError as e:
raise RouteTableException( raise AnsibleRouteTableException(
'Unable to create route table {0}, error: {1}' 'Unable to create route table {0}, error: {1}'
.format(route_table_id or resource_tags, e) .format(route_table_id or resource_tags, e)
) )
@ -381,7 +382,7 @@ def ensure_route_table_present(vpc_conn, vpc_id, route_table_id, resource_tags,
result = ensure_routes(vpc_conn, route_table, routes, check_mode) result = ensure_routes(vpc_conn, route_table, routes, check_mode)
changed = changed or result['changed'] changed = changed or result['changed']
except EC2ResponseError as e: except EC2ResponseError as e:
raise RouteTableException( raise AnsibleRouteTableException(
'Unable to ensure routes for route table {0}, error: {1}' 'Unable to ensure routes for route table {0}, error: {1}'
.format(route_table, e) .format(route_table, e)
) )
@ -396,7 +397,7 @@ def ensure_route_table_present(vpc_conn, vpc_id, route_table_id, resource_tags,
subnet = get_subnet_by_id(vpc_conn, vpc_id, subnet_name) subnet = get_subnet_by_id(vpc_conn, vpc_id, subnet_name)
associated_subnets.append(subnet) associated_subnets.append(subnet)
except EC2ResponseError as e: except EC2ResponseError as e:
raise RouteTableException( raise AnsibleRouteTableException(
'Unable to find subnets for route table {0}, error: {1}' 'Unable to find subnets for route table {0}, error: {1}'
.format(route_table, e) .format(route_table, e)
) )
@ -406,7 +407,7 @@ def ensure_route_table_present(vpc_conn, vpc_id, route_table_id, resource_tags,
vpc_conn, vpc_id, route_table, associated_subnets, check_mode) vpc_conn, vpc_id, route_table, associated_subnets, check_mode)
changed = changed or result['changed'] changed = changed or result['changed']
except EC2ResponseError as e: except EC2ResponseError as e:
raise RouteTableException( raise AnsibleRouteTableException(
'Unable to associate subnets for route table {0}, error: {1}' 'Unable to associate subnets for route table {0}, error: {1}'
.format(route_table, e) .format(route_table, e)
) )
@ -467,7 +468,7 @@ def main():
vpc_conn, vpc_id, route_table_id, resource_tags, vpc_conn, vpc_id, route_table_id, resource_tags,
module.check_mode module.check_mode
) )
except RouteTableException as e: except AnsibleRouteTableException as e:
module.fail_json(msg=str(e)) module.fail_json(msg=str(e))
module.exit_json(**result) module.exit_json(**result)