refactored to use dict invocation for calling client peerings functions

pull/18777/head
Mike Mochan 9 years ago committed by Matt Clay
parent 64f947a327
commit 5963fc4641

@ -182,12 +182,6 @@ def is_pending(peering_conn):
return peering_conn['Status']['Code'] == 'pending-acceptance' return peering_conn['Status']['Code'] == 'pending-acceptance'
def peer_status(resource, module):
peer_id = module.params.get('peering_id')
vpc_peering_connection = resource.VpcPeeringConnection(peer_id)
return vpc_peering_connection.status['Message']
def create_peer_connection(client, module): def create_peer_connection(client, module):
changed = False changed = False
params = dict() params = dict()
@ -203,12 +197,13 @@ def create_peer_connection(client, module):
peering_conns = describe_peering_connections(vpc_id, peer_vpc_id, client) peering_conns = describe_peering_connections(vpc_id, peer_vpc_id, client)
for peering_conn in peering_conns['VpcPeeringConnections']: for peering_conn in peering_conns['VpcPeeringConnections']:
if is_active(peering_conn): if is_active(peering_conn):
return (False, peering_conn['VpcPeeringConnectionId']) return (changed, peering_conn['VpcPeeringConnectionId'])
if is_pending(peering_conn): if is_pending(peering_conn):
return (False, peering_conn['VpcPeeringConnectionId']) return (changed, peering_conn['VpcPeeringConnectionId'])
try: try:
peering_conn = client.create_vpc_peering_connection(**params) peering_conn = client.create_vpc_peering_connection(**params)
return (True, peering_conn['VpcPeeringConnection']['VpcPeeringConnectionId']) changed = True
return (changed, peering_conn['VpcPeeringConnection']['VpcPeeringConnectionId'])
except botocore.exceptions.ClientError as e: except botocore.exceptions.ClientError as e:
module.fail_json(msg=str(e)) module.fail_json(msg=str(e))
@ -218,32 +213,18 @@ def accept_reject_delete(state, client, resource, module):
params = dict() params = dict()
params['VpcPeeringConnectionId'] = module.params.get('peering_id') params['VpcPeeringConnectionId'] = module.params.get('peering_id')
params['DryRun'] = module.check_mode params['DryRun'] = module.check_mode
invocations = {
'accept': client.accept_vpc_peering_connection,
'reject': client.reject_vpc_peering_connection,
'absent': client.delete_vpc_peering_connection
}
try:
invocations[state](**params)
changed = True
except botocore.exceptions.ClientError as e:
module.fail_json(msg=str(e))
peer_id = module.params.get('peering_id') return changed, params['VpcPeeringConnectionId']
if state == "accept":
if peer_status(resource, module) == "Active":
return (False, peer_id)
try:
client.accept_vpc_peering_connection(**params)
return (True, peer_id)
except botocore.exceptions.ClientError as e:
module.fail_json(msg=str(e))
if state == "reject":
if peer_status(resource, module) != "Active":
try:
client.reject_vpc_peering_connection(**params)
return (True, peer_id)
except botocore.exceptions.ClientError as e:
module.fail_json(msg=str(e))
else:
return (False, peer_id)
if state == "absent":
try:
client.delete_vpc_peering_connection(**params)
return (True, peer_id)
except botocore.exceptions.ClientError as e:
module.fail_json(msg=str(e))
return (changed, "")
def main(): def main():

Loading…
Cancel
Save