# --------------------------------------------------------------------------
#
# Copyright (c) Microsoft Corporation. All rights reserved.
#
# The MIT License (MIT)
#
# 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, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following 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 MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS 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.
#
# --------------------------------------------------------------------------
import json
import logging
import sys
from typing import Callable, Any, Dict, Optional, List, Union, TYPE_CHECKING
_LOGGER = logging.getLogger(__name__)
if TYPE_CHECKING:
from azure.core.pipeline.transport._base import _HttpResponseBase
__all__ = [
'AzureError',
'ServiceRequestError',
'ServiceResponseError',
'HttpResponseError',
'DecodeError',
'ResourceExistsError',
'ResourceNotFoundError',
'ClientAuthenticationError',
'ResourceModifiedError',
'ResourceNotModifiedError',
'TooManyRedirectsError',
'ODataV4Format',
'ODataV4Error',
]
def raise_with_traceback(exception, *args, **kwargs):
# type: (Callable, Any, Any) -> None
"""Raise exception with a specified traceback.
This MUST be called inside a "except" clause.
:param Exception exception: Error type to be raised.
:param args: Any additional args to be included with exception.
:keyword str message: Message to be associated with the exception. If omitted, defaults to an empty string.
"""
message = kwargs.pop('message', '')
exc_type, exc_value, exc_traceback = sys.exc_info()
# If not called inside a "except", exc_type will be None. Assume it will not happen
exc_msg = "{}, {}: {}".format(message, exc_type.__name__, exc_value) # type: ignore
error = exception(exc_msg, *args, **kwargs)
try:
raise error.with_traceback(exc_traceback)
except AttributeError:
error.__traceback__ = exc_traceback
raise error
def map_error(status_code, response, error_map):
if not error_map:
return
error_type = error_map.get(status_code)
if not error_type:
return
error = error_type(response=response)
raise error
class AzureError(Exception):
"""Base exception for all errors."""
def __init__(self, message, *args, **kwargs):
self.inner_exception = kwargs.get('error')
self.exc_type, self.exc_value, self.exc_traceback = sys.exc_info()
self.exc_type = self.exc_type.__name__ if self.exc_type else type(self.inner_exception)
self.exc_msg = "{}, {}: {}".format(message, self.exc_type, self.exc_value) # type: ignore
self.message = str(message)
super(AzureError, self).__init__(self.message, *args)
def raise_with_traceback(self):
try:
raise super(AzureError, self).with_traceback(self.exc_traceback)
except AttributeError:
self.__traceback__ = self.exc_traceback
raise self
class ServiceRequestError(AzureError):
"""An error occurred while attempt to make a request to the service.
No request was sent.
"""
class ServiceResponseError(AzureError):
"""The request was sent, but the client failed to understand the response.
The connection may have timed out. These errors can be retried for idempotent or
safe operations"""
class HttpResponseError(AzureError):
"""A request was made, and a non-success status code was received from the service.
:param status_code: HttpResponse's status code
:type status_code: int
:param response: The response that triggered the exception.
:type response: ~azure.core.pipeline.transport.HttpResponse or ~azure.core.pipeline.transport.AsyncHttpResponse
"""
def __init__(self, message=None, response=None, **kwargs):
self.reason = None
self.status_code = None
self.response = response
if response:
self.reason = response.reason
self.status_code = response.status_code
message = message or "Operation returned an invalid status '{}'".format(self.reason)
try:
try:
if self.error.error.code or self.error.error.message:
message = "({}) {}".format(
self.error.error.code,
self.error.error.message)
except AttributeError:
if self.error.message: #pylint: disable=no-member
message = self.error.message #pylint: disable=no-member
except AttributeError:
# Exception will only have an error if it has been deserialized from
# generated code. We should add the empty attribute if it's not present.
if not hasattr(self, 'error'):
self.error = None
super(HttpResponseError, self).__init__(message=message, **kwargs)
class DecodeError(HttpResponseError):
"""Error raised during response deserialization."""
class ResourceExistsError(HttpResponseError):
"""An error response with status code 4xx.
This will not be raised directly by the Azure core pipeline."""
class ResourceNotFoundError(HttpResponseError):
""" An error response, typically triggered by a 412 response (for update) or 404 (for get/post)
"""
class ClientAuthenticationError(HttpResponseError):
"""An error response with status code 4xx.
This will not be raised directly by the Azure core pipeline."""
class ResourceModifiedError(HttpResponseError):
"""An error response with status code 4xx, typically 412 Conflict.
This will not be raised directly by the Azure core pipeline."""
class ResourceNotModifiedError(HttpResponseError):
"""An error response with status code 304.
This will not be raised directly by the Azure core pipeline."""
class TooManyRedirectsError(HttpResponseError):
"""Reached the maximum number of redirect attempts."""
def __init__(self, history, *args, **kwargs):
self.history = history
message = "Reached maximum redirect attempts."
super(TooManyRedirectsError, self).__init__(message, *args, **kwargs)
class ODataV4Format(object):
"""Class to describe OData V4 error format.
http://docs.oasis-open.org/odata/odata-json-format/v4.0/os/odata-json-format-v4.0-os.html#_Toc372793091
:param dict json_object: A Python dict representing a ODataV4 JSON
:ivar str ~.code: Its value is a service-defined error code.
This code serves as a sub-status for the HTTP error code specified in the response.
:ivar str message: Human-readable, language-dependent representation of the error.
:ivar str target: The target of the particular error (for example, the name of the property in error).
This field is optional and may be None.
:ivar list[ODataV4Format] details: Array of ODataV4Format instances that MUST contain name/value pairs
for code and message, and MAY contain a name/value pair for target, as described above.
:ivar dict innererror: An object. The contents of this object are service-defined.
Usually this object contains information that will help debug the service.
"""
def __init__(self, json_object):
# Required fields, but assume they could be missing still to be robust
self.code = json_object.get("code") # type: Optional[str]
self.message = json_object.get("message") # type: Optional[str]
# Optional fields
self.target = json_object.get("target", None) # type: Optional[str]
# details is recursive of this very format
self.details = [
self.__class__(detail_node)
for detail_node in json_object.get("details", [])
] # type: List[ODataV4Format]
self.innererror = json_object.get("innererror", {}) # type: Dict[str, Any]
def __str__(self):
error_str = "Code: {}".format(self.code)
error_str += "\nMessage: {}".format(self.message)
if self.target:
error_str += "\nTarget: {}".format(self.target)
if self.details:
error_str += "\nException Details:"
for error_obj in self.details:
# Indent for visibility
error_str += "\n".join("\t" + s for s in str(error_obj).splitlines())
if self.innererror:
error_str += "\nInner error: {}".format(
json.dumps(self.innererror, indent=4)
)
return error_str
class ODataV4Error(HttpResponseError):
"""An HTTP response error where the JSON is decoded as OData V4 error format.
http://docs.oasis-open.org/odata/odata-json-format/v4.0/os/odata-json-format-v4.0-os.html#_Toc372793091
:ivar dict odata_json: The parsed JSON body as attribute for convenience.
:ivar str ~.code: Its value is a service-defined error code.
This code serves as a sub-status for the HTTP error code specified in the response.
:ivar str message: Human-readable, language-dependent representation of the error.
:ivar str target: The target of the particular error (for example, the name of the property in error).
This field is optional and may be None.
:ivar list[ODataV4Format] details: Array of ODataV4Format instances that MUST contain name/value pairs
for code and message, and MAY contain a name/value pair for target, as described above.
:ivar dict innererror: An object. The contents of this object are service-defined.
Usually this object contains information that will help debug the service.
"""
_ERROR_FORMAT = ODataV4Format
def __init__(self, response, **kwargs):
# type: (_HttpResponseBase, Any) -> None
# Ensure field are declared, whatever can happen afterwards
self.odata_json = None # type: Optional[Dict[str, Any]]
try:
self.odata_json = json.loads(response.text())
odata_message = self.odata_json.setdefault("error", {}).get("message")
except Exception: #pylint: disable=broad-except
# If the body is not JSON valid, just stop now
odata_message = None
self.code = None # type: Optional[str]
self.message = kwargs.get("message", odata_message) # type: Optional[str]
self.target = None # type: Optional[str]
self.details = [] # type: Optional[List[Any]]
self.innererror = {} # type: Optional[Dict[str, Any]]
if self.message and "message" not in kwargs:
kwargs["message"] = self.message
super(ODataV4Error, self).__init__(response=response, **kwargs)
self._error_format = None # type: Optional[Union[str, ODataV4Format]]
if self.odata_json:
try:
error_node = self.odata_json["error"]
self._error_format = self._ERROR_FORMAT(error_node)
self.__dict__.update(
{
k: v
for k, v in self._error_format.__dict__.items()
if v is not None
}
)
except Exception: #pylint: disable=broad-except
_LOGGER.info("Received error message was not valid OdataV4 format.")
self._error_format = "JSON was invalid for format " + str(self._ERROR_FORMAT)
def __str__(self):
if self._error_format:
return str(self._error_format)
return super(ODataV4Error, self).__str__()