88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
# Copyright (c) 2012-2013, Eucalyptus Systems, Inc.
|
|
#
|
|
# Permission to use, copy, modify, and/or distribute this software for
|
|
# any purpose with or without fee is hereby granted, provided that the
|
|
# above copyright notice and this permission notice appear in all copies.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
|
|
import io
|
|
import six
|
|
|
|
try:
|
|
from xml.etree import cElementTree as ElementTree
|
|
except ImportError:
|
|
from xml.etree import ElementTree
|
|
|
|
class ClientError(Exception):
|
|
'''
|
|
General client error (e.g. error accessing the server)
|
|
'''
|
|
pass
|
|
|
|
|
|
class AuthError(ClientError):
|
|
'''
|
|
Authentication handler failure
|
|
'''
|
|
pass
|
|
|
|
|
|
class ServiceInitError(ClientError):
|
|
'''
|
|
Failure to set up a service
|
|
'''
|
|
|
|
def __init__(self, reason=None):
|
|
ClientError.__init__(self, reason)
|
|
|
|
|
|
class ServerError(RuntimeError):
|
|
'''
|
|
An error response from the server
|
|
'''
|
|
|
|
def __init__(self, status, body=None, *args):
|
|
RuntimeError.__init__(self, *args)
|
|
self.status_code = status # HTTP status code
|
|
self.body = body or u''
|
|
self.code = None # API error code
|
|
self.message = None # Error message
|
|
|
|
if self.body:
|
|
try:
|
|
xml_stream = io.StringIO(six.text_type(self.body))
|
|
for event, elem in ElementTree.iterparse(xml_stream,
|
|
events=('end',)):
|
|
# Strip the namespace, if any, off the tag
|
|
if elem.tag.startswith('{'):
|
|
bare_tag = elem.tag.split('}', 1)[1]
|
|
else:
|
|
bare_tag = elem.tag
|
|
if bare_tag == u'Code':
|
|
self.code = elem.text
|
|
elif bare_tag == u'Message':
|
|
self.message = elem.text
|
|
except ElementTree.ParseError as err:
|
|
# Dump the unparseable message body so we don't include
|
|
# unusable garbage in the exception. Since Eucalyptus
|
|
# frequently returns plain text and/or broken XML, store it
|
|
# in case we need it later.
|
|
self.message = self.body
|
|
self.body = None
|
|
|
|
def __str__(self):
|
|
s_bits = [self.__class__.__name__ + ':', self.status_code]
|
|
if self.code:
|
|
s_bits.append(self.code)
|
|
if self.message:
|
|
s_bits.append(self.message)
|
|
return ' '.join(s_bits)
|