1# Copyright (c) 2012 Amazon.com, Inc. or its affiliates.  All Rights Reserved
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
23class AccountAttribute(object):
24    def __init__(self, connection=None):
25        self.connection = connection
26        self.attribute_name = None
27        self.attribute_values = None
28
29    def startElement(self, name, attrs, connection):
30        if name == 'attributeValueSet':
31            self.attribute_values = AttributeValues()
32            return self.attribute_values
33
34    def endElement(self, name, value, connection):
35        if name == 'attributeName':
36            self.attribute_name = value
37
38
39class AttributeValues(list):
40    def startElement(self, name, attrs, connection):
41        pass
42
43    def endElement(self, name, value, connection):
44        if name == 'attributeValue':
45            self.append(value)
46
47
48class VPCAttribute(object):
49    def __init__(self, connection=None):
50        self.connection = connection
51        self.vpc_id = None
52        self.enable_dns_hostnames = None
53        self.enable_dns_support = None
54        self._current_attr = None
55
56    def startElement(self, name, attrs, connection):
57        if name in ('enableDnsHostnames', 'enableDnsSupport'):
58            self._current_attr = name
59
60    def endElement(self, name, value, connection):
61        if name == 'vpcId':
62            self.vpc_id = value
63        elif name == 'value':
64            if value == 'true':
65                value = True
66            else:
67                value = False
68            if self._current_attr == 'enableDnsHostnames':
69                self.enable_dns_hostnames = value
70            elif self._current_attr == 'enableDnsSupport':
71                self.enable_dns_support = value
72