145 lines
3.7 KiB
Python
145 lines
3.7 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import json
|
|
import sys
|
|
|
|
|
|
class BaseException(Exception):
|
|
"""An error occurred."""
|
|
def __init__(self, message=None):
|
|
self.message = message
|
|
|
|
def __str__(self):
|
|
return self.message or self.__class__.__doc__
|
|
|
|
|
|
class CommandError(BaseException):
|
|
"""Invalid usage of CLI."""
|
|
|
|
|
|
class InvalidEndpoint(BaseException):
|
|
"""The provided endpoint is invalid."""
|
|
|
|
|
|
class CommunicationError(BaseException):
|
|
"""Unable to communicate with server."""
|
|
|
|
|
|
class HTTPException(BaseException):
|
|
"""Base exception for all HTTP-derived exceptions."""
|
|
code = 'N/A'
|
|
|
|
def __init__(self, details=None):
|
|
self.details = details
|
|
|
|
def __str__(self):
|
|
message = ""
|
|
if self.details:
|
|
message = self.details
|
|
try:
|
|
data = json.loads(self.details)
|
|
message = data.get("error_message", "")
|
|
if isinstance(message, dict) and "faultstring" in message:
|
|
message = "ERROR %s" % message["faultstring"]
|
|
except (ValueError, TypeError, AttributeError):
|
|
pass
|
|
|
|
if message:
|
|
message = " %s" % message
|
|
return "%s (HTTP %s)%s" % (self.__class__.__name__, self.code,
|
|
message)
|
|
|
|
|
|
class HTTPMultipleChoices(HTTPException):
|
|
code = 300
|
|
|
|
def __str__(self):
|
|
self.details = ("Requested version of OpenStack Images API is not"
|
|
"available.")
|
|
return "%s (HTTP %s) %s" % (self.__class__.__name__, self.code,
|
|
self.details)
|
|
|
|
|
|
class HTTPBadRequest(HTTPException):
|
|
code = 400
|
|
|
|
|
|
class HTTPUnauthorized(HTTPException):
|
|
code = 401
|
|
|
|
|
|
class HTTPForbidden(HTTPException):
|
|
code = 403
|
|
|
|
|
|
class HTTPNotFound(HTTPException):
|
|
code = 404
|
|
|
|
|
|
class HTTPMethodNotAllowed(HTTPException):
|
|
code = 405
|
|
|
|
|
|
class HTTPConflict(HTTPException):
|
|
code = 409
|
|
|
|
|
|
class HTTPOverLimit(HTTPException):
|
|
code = 413
|
|
|
|
|
|
class HTTPInternalServerError(HTTPException):
|
|
code = 500
|
|
|
|
|
|
class HTTPNotImplemented(HTTPException):
|
|
code = 501
|
|
|
|
|
|
class HTTPBadGateway(HTTPException):
|
|
code = 502
|
|
|
|
|
|
class HTTPServiceUnavailable(HTTPException):
|
|
code = 503
|
|
|
|
|
|
# NOTE(bcwaldon): Build a mapping of HTTP codes to corresponding exception
|
|
# classes
|
|
_code_map = {}
|
|
for obj_name in dir(sys.modules[__name__]):
|
|
if obj_name.startswith('HTTP'):
|
|
obj = getattr(sys.modules[__name__], obj_name)
|
|
_code_map[obj.code] = obj
|
|
|
|
|
|
def from_response(response, details=None):
|
|
"""Return an instance of an HTTPException based on http response."""
|
|
if hasattr(response, "status"):
|
|
# it is response from HTTPClient (httplib)
|
|
code = response.status
|
|
elif hasattr(response, "status_code"):
|
|
# it is response from SessionClient (requests)
|
|
code = response.status_code
|
|
else:
|
|
# it is something unexpected
|
|
raise TypeError("Function 'from_response' expects only response object"
|
|
" from httplib or requests libraries.")
|
|
cls = _code_map.get(code)
|
|
if cls is None:
|
|
exc = HTTPException(details)
|
|
exc.code = code
|
|
return exc
|
|
else:
|
|
return cls(details)
|