rpms
/
python-boto
20
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add NAT gateway support

Browse Source 
https://github.com/boto/boto/pull/3472
f38
Garrett Holmstrom 9 years ago
parent 3a12803124
commit e892191bb8
2 changed files with 325 additions and 0 deletions
Show Stats Download Patch File Download Diff File

319
boto-2.40.0-nat-gateway.patch
Unescape View File

@ -0,0 +1,319 @@
Index: boto-2.40.0/boto/vpc/__init__.py
===================================================================
--- boto-2.40.0.orig/boto/vpc/__init__.py
+++ boto-2.40.0/boto/vpc/__init__.py
@@ -29,6 +29,7 @@ from boto.vpc.vpc import VPC
from boto.vpc.customergateway import CustomerGateway
from boto.vpc.networkacl import NetworkAcl
from boto.vpc.routetable import RouteTable
+from boto.vpc.natgateway import NatGateway
from boto.vpc.internetgateway import InternetGateway
from boto.vpc.vpngateway import VpnGateway, Attachment
from boto.vpc.dhcpoptions import DhcpOptions
@@ -783,6 +784,76 @@ class VPCConnection(EC2Connection):
return self.get_status('DeleteNetworkAclEntry', params)
+ # NAT Gateways
+
+ def get_all_nat_gateways(self, nat_gateway_ids=None, filters=None, dry_run=False):
+ """
+ Get a list of NAT gateways. You can filter results to return information
+ about only those gateways that you're interested in.
+
+ :type nat_gateway_ids: list
+ :param nat_gateway_ids: A list of strings with the desired gateway IDs.
+
+ :type filters: list of tuples or dict
+ :param filters: A list of tuples or dict containing filters. Each tuple
+ or dict item consists of a filter key and a filter value.
+
+ :type dry_run: bool
+ :param dry_run: Set to True if the operation should not actually run.
+
+ """
+ params = {}
+
+ if nat_gateway_ids:
+ self.build_list_params(params, nat_gateway_ids,
+ 'NatGatewayId')
+ if filters:
+ self.build_filter_params(params, filters)
+ if dry_run:
+ params['DryRun'] = 'true'
+ return self.get_list('DescribeNatGateways', params,
+ [('item', NatGateway)])
+
+ def create_nat_gateway(self, subnet_id, allocation_id, dry_run=False):
+ """
+ Creates a NAT gateway for VPC.
+
+ :type subnet_id: str
+ :param subnet_id: The subnet in which the NAT gateway should be launched.
+
+ :type allocation_id: str
+ :param allocation_id: The allocation ID of an elastic IP address for the public side of the gateway.
+
+ :type dry_run: bool
+ :param dry_run: Set to True if the operation should not actually run.
+
+ :rtype: Newly created nat gateway.
+ :return: `boto.vpc.natgateway.NATGateway`
+ """
+ params = {'SubnetId': subnet_id,
+ 'AllocationId': allocation_id}
+ if dry_run:
+ params['DryRun'] = 'true'
+ return self.get_object('CreateNatGateway', params, NatGateway)
+
+ def delete_nat_gateway(self, nat_gateway_id, dry_run=False):
+ """
+ Deletes a NAT gateway from the VPC.
+
+ :type nat_gateway_id: str
+ :param nat_gateway_id: The ID of the NAT gateway to delete.
+
+ :type dry_run: bool
+ :param dry_run: Set to True if the operation should not actually run.
+
+ :rtype: Bool
+ :return: True if successful
+ """
+ params = {'NatGatewayId': nat_gateway_id}
+ if dry_run:
+ params['DryRun'] = 'true'
+ return self.get_status('DeleteNatGateway', params)
+
# Internet Gateways
def get_all_internet_gateways(self, internet_gateway_ids=None,
Index: boto-2.40.0/boto/vpc/natgateway.py
===================================================================
--- /dev/null
+++ boto-2.40.0/boto/vpc/natgateway.py
@@ -0,0 +1,89 @@
+# Copyright (c) 2009-2010 Mitch Garnaat http://garnaat.org/
+#
+# Permission is hereby granted, free of charge, to any person obtaining a
+# copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish, dis-
+# tribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the fol-
+# lowing conditions:
+#
+# The above copyright notice and this permission notice shall be included
+# in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+
+"""
+Represents a NAT Gateway
+"""
+
+from boto.ec2.ec2object import TaggedEC2Object
+from boto.resultset import ResultSet
+
+
+class NatGateway(TaggedEC2Object):
+ def __init__(self, connection=None):
+ super(NatGateway, self).__init__(connection)
+ self.id = None
+ self.vpc_id = None
+ self.subnet_id = None
+ self.state = None
+ self.addresses = []
+
+ def __repr__(self):
+ return 'NatGateway:%s' % self.id
+
+ def startElement(self, name, attrs, connection):
+ result = super(NatGateway, self).startElement(name, attrs, connection)
+
+ if result is not None:
+ # Parent found an interested element, just return it
+ return result
+
+ if name == 'natGatewayAddressSet':
+ self.addresses = ResultSet([('item', NatGatewayAddress)])
+ return self.addresses
+ else:
+ return None
+
+ def endElement(self, name, value, connection):
+ if name == 'natGatewayId':
+ self.id = value
+ elif name == 'vpcId':
+ self.vpc_id = value
+ elif name == 'subnetId':
+ self.subnet_id = value
+ elif name == 'state':
+ self.state = value
+ else:
+ setattr(self, name, value)
+
+
+class NatGatewayAddress(object):
+ def __init__(self, connection=None):
+ self.interface_id = None
+ self.allocation_id = None
+ self.ip_public = None
+ self.ip_private = None
+
+ def __repr__(self):
+ return 'NatGatewayAddress:%s' % self.interface_id
+
+ def startElement(self, name, attrs, connection):
+ return None
+
+ def endElement(self, name, value, connection):
+ if name == 'networkInterfaceId':
+ self.interface_id = value
+ elif name == 'publicIp':
+ self.ip_public = value
+ elif name == 'allocationId':
+ self.allocation_id = value
+ elif name == 'privateIp':
+ self.ip_private = value
Index: boto-2.40.0/docs/source/ref/vpc.rst
===================================================================
--- boto-2.40.0.orig/docs/source/ref/vpc.rst
+++ boto-2.40.0/docs/source/ref/vpc.rst
@@ -32,6 +32,13 @@ boto.vpc.internetgateway
:members:
:undoc-members:
+boto.vpc.natgateway
+-------------------
+
+.. automodule:: boto.vpc.natgateway
+ :members:
+ :undoc-members:
+
boto.vpc.routetable
-------------------
Index: boto-2.40.0/tests/unit/vpc/test_natgateway.py
===================================================================
--- /dev/null
+++ boto-2.40.0/tests/unit/vpc/test_natgateway.py
@@ -0,0 +1,113 @@
+from tests.unit import unittest
+from tests.unit import AWSMockServiceTestCase
+
+from boto.vpc import VPCConnection, NatGateway
+
+
+class TestDescribeNatGateway(AWSMockServiceTestCase):
+
+ connection_class = VPCConnection
+
+ def default_body(self):
+ return b"""
+ <DescribeNatGatewaysResponse xmlns="http://ec2.amazonaws.com/doc/2015-10-01/">
+ <requestId>bfed02c6-dae9-47c0-86a2-example</requestId>
+ <natGatewaySet>
+ <item>
+ <subnetId>subnet-1a2a3a4a</subnetId>
+ <natGatewayAddressSet>
+ <item>
+ <networkInterfaceId>eni-00e37850</networkInterfaceId>
+ <publicIp>198.18.125.129</publicIp>
+ <allocationId>eipalloc-37fc1a52</allocationId>
+ <privateIp>10.0.2.147</privateIp>
+ </item>
+ </natGatewayAddressSet>
+ <createTime>2015-11-25T14:00:55.416Z</createTime>
+ <vpcId>vpc-4e20d42b</vpcId>
+ <natGatewayId>nat-04e77a5e9c34432f9</natGatewayId>
+ <state>available</state>
+ </item>
+ </natGatewaySet>
+ </DescribeNatGatewaysResponse>
+ """
+
+ def test_describe_nat_gateway(self):
+ self.set_http_response(status_code=200)
+ api_response = self.service_connection.get_all_nat_gateways(
+ 'nat-04e77a5e9c34432f9', filters=[('natGatewayAddress.allocationId', ['eipalloc-37fc1a52'])])
+ self.assert_request_parameters({
+ 'Action': 'DescribeNatGateways',
+ 'NatGatewayId.1': 'nat-04e77a5e9c34432f9',
+ 'Filter.1.Name': 'natGatewayAddress.allocationId',
+ 'Filter.1.Value.1': 'eipalloc-37fc1a52'},
+ ignore_params_values=['AWSAccessKeyId', 'SignatureMethod',
+ 'SignatureVersion', 'Timestamp',
+ 'Version'])
+ self.assertEquals(len(api_response), 1)
+ self.assertIsInstance(api_response[0], NatGateway)
+ self.assertEqual(api_response[0].id, 'nat-04e77a5e9c34432f9')
+
+
+class TestCreateNatGateway(AWSMockServiceTestCase):
+
+ connection_class = VPCConnection
+
+ def default_body(self):
+ return b"""
+ <CreateNatGatewayResponse xmlns="http://ec2.amazonaws.com/doc/2015-10-01/">
+ <requestId>1b74dc5c-bcda-403f-867d-example</requestId>
+ <natGateway>
+ <subnetId>subnet-1a2b3c4d</subnetId>
+ <natGatewayAddressSet>
+ <item>
+ <allocationId>eipalloc-37fc1a52</allocationId>
+ </item>
+ </natGatewayAddressSet>
+ <createTime>2015-11-25T14:00:55.416Z</createTime>
+ <vpcId>vpc-4e20d42b</vpcId>
+ <natGatewayId>nat-04e77a5e9c34432f9</natGatewayId>
+ <state>pending</state>
+ </natGateway>
+ </CreateNatGatewayResponse>
+ """
+
+ def test_create_nat_gateway(self):
+ self.set_http_response(status_code=200)
+ api_response = self.service_connection.create_nat_gateway('subnet-1a2b3c4d', 'eipalloc-37fc1a52')
+ self.assert_request_parameters({
+ 'Action': 'CreateNatGateway',
+ 'SubnetId': 'subnet-1a2b3c4d',
+ 'AllocationId': 'eipalloc-37fc1a52'},
+ ignore_params_values=['AWSAccessKeyId', 'SignatureMethod',
+ 'SignatureVersion', 'Timestamp',
+ 'Version'])
+ self.assertIsInstance(api_response, NatGateway)
+ self.assertEqual(api_response.id, 'nat-04e77a5e9c34432f9')
+
+
+class TestDeleteNatGateway(AWSMockServiceTestCase):
+
+ connection_class = VPCConnection
+
+ def default_body(self):
+ return b"""
+ <DeleteNatGatewayResponse xmlns="http://ec2.amazonaws.com/doc/2015-10-01/">
+ <requestId>741fc8ab-6ebe-452b-b92b-example</requestId>
+ <natGatewayId>nat-04ae55e711cec5680</natGatewayId>
+ </DeleteNatGatewayResponse>
+ """
+
+ def test_delete_nat_gateway(self):
+ self.set_http_response(status_code=200)
+ api_response = self.service_connection.delete_nat_gateway('nat-04ae55e711cec5680')
+ self.assert_request_parameters({
+ 'Action': 'DeleteNatGateway',
+ 'NatGatewayId': 'nat-04ae55e711cec5680'},
+ ignore_params_values=['AWSAccessKeyId', 'SignatureMethod',
+ 'SignatureVersion', 'Timestamp',
+ 'Version'])
+ self.assertEquals(api_response, True)
+
+if __name__ == '__main__':
+ unittest.main()

6
python-boto.spec
Unescape View File

@ -34,6 +34,10 @@ Source1: boto-mit-license.txt
# https://github.com/boto/boto/pull/3086
Patch1: boto-2.39.0-devendor.patch
# Add NAT gateway support
# https://github.com/boto/boto/pull/3472
Patch2: boto-2.40.0-nat-gateway.patch
BuildRequires: python2-devel
BuildRequires: python-setuptools
BuildRequires: python-six
@ -114,6 +118,7 @@ cloud systems like Eucalyptus, OpenStack and Open Nebula.
%prep
%setup -q -n boto-%{version}
%patch1 -p1
%patch2 -p1
rm -r boto/vendored
@ -174,6 +179,7 @@ mv $RPM_BUILD_ROOT/%{_bindir} examples
%changelog
* Tue Jun 21 2016 Garrett Holmstrom <gholms@fedoraproject.org> - 2.40.0-2
- Cleaned up spec file
- Added NAT gateway support
* Mon May 23 2016 Jon Ciesla <limburgher@gmail.com> - 2.40.0-1
- 2.40.0.

Write Preview
Loading…
Cancel
Save