1# Copyright (c) 2009-2010 Mitch Garnaat http://garnaat.org/
2#
3# Permission is hereby granted, free of charge, to any person obtaining a
4# copy of this software and associated documentation files (the
5# "Software"), to deal in the Software without restriction, including
6# without limitation the rights to use, copy, modify, merge, publish, dis-
7# tribute, sublicense, and/or sell copies of the Software, and to permit
8# persons to whom the Software is furnished to do so, subject to the fol-
9# lowing conditions:
10#
11# The above copyright notice and this permission notice shall be included
12# in all copies or substantial portions of the Software.
13#
14# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
16# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
17# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
18# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
20# IN THE SOFTWARE.
21
22"""
23Represents a Vpn Gateway
24"""
25
26from boto.ec2.ec2object import TaggedEC2Object
27
28class Attachment(object):
29
30    def __init__(self, connection=None):
31        self.vpc_id = None
32        self.state = None
33
34    def startElement(self, name, attrs, connection):
35        pass
36
37    def endElement(self, name, value, connection):
38        if name == 'vpcId':
39            self.vpc_id = value
40        elif name == 'state':
41            self.state = value
42        else:
43            setattr(self, name, value)
44
45class VpnGateway(TaggedEC2Object):
46
47    def __init__(self, connection=None):
48        super(VpnGateway, self).__init__(connection)
49        self.id = None
50        self.type = None
51        self.state = None
52        self.availability_zone = None
53        self.attachments = []
54
55    def __repr__(self):
56        return 'VpnGateway:%s' % self.id
57
58    def startElement(self, name, attrs, connection):
59        retval = super(VpnGateway, self).startElement(name, attrs, connection)
60        if retval is not None:
61            return retval
62        if name == 'item':
63            att = Attachment()
64            self.attachments.append(att)
65            return att
66
67    def endElement(self, name, value, connection):
68        if name == 'vpnGatewayId':
69            self.id = value
70        elif name == 'type':
71            self.type = value
72        elif name == 'state':
73            self.state = value
74        elif name == 'availabilityZone':
75            self.availability_zone = value
76        elif name == 'attachments':
77            pass
78        else:
79            setattr(self, name, value)
80
81    def attach(self, vpc_id, dry_run=False):
82        return self.connection.attach_vpn_gateway(
83            self.id,
84            vpc_id,
85            dry_run=dry_run
86        )
87
88