Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor NatGateways methods to use _get_param()

Browse Source
This commit is contained in:
Hugo Lopes Tavares 2016-03-01 13:57:41 -05:00
parent c689816337
commit 6222f335f3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
moto/ec2/responses/nat_gateways.py
View File

@ -6,14 +6,14 @@ from moto.ec2.utils import filters_from_querystring
class NatGateways(BaseResponse): class NatGateways(BaseResponse):
def create_nat_gateway(self): def create_nat_gateway(self):
subnet_id = self.querystring.get('SubnetId')[0] subnet_id = self._get_param('SubnetId')
allocation_id = self.querystring.get('AllocationId')[0] allocation_id = self._get_param('AllocationId')
nat_gateway = self.ec2_backend.create_nat_gateway(subnet_id=subnet_id, allocation_id=allocation_id) nat_gateway = self.ec2_backend.create_nat_gateway(subnet_id=subnet_id, allocation_id=allocation_id)
template = self.response_template(CREATE_NAT_GATEWAY) template = self.response_template(CREATE_NAT_GATEWAY)
return template.render(nat_gateway=nat_gateway) return template.render(nat_gateway=nat_gateway)
def delete_nat_gateway(self): def delete_nat_gateway(self):
nat_gateway_id = self.querystring.get('NatGatewayId')[0] nat_gateway_id = self._get_param('NatGatewayId')
nat_gateway = self.ec2_backend.delete_nat_gateway(nat_gateway_id) nat_gateway = self.ec2_backend.delete_nat_gateway(nat_gateway_id)
template = self.response_template(DELETE_NAT_GATEWAY_RESPONSE) template = self.response_template(DELETE_NAT_GATEWAY_RESPONSE)
return template.render(nat_gateway=nat_gateway) return template.render(nat_gateway=nat_gateway)