ec2_vpc - VPCException -> AnsibleVPCException
This commit is contained in:
parent
bded55a84d
commit
504e1773ef
1 changed files with 15 additions and 14 deletions
|
@ -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)
|
||||||
|
|
Loading…
Reference in a new issue