Move response from AzureError to HttpResponseError (Azure#5591)

* Move response from AzureError to HttpResponseError * pylint fixes * expose HttpRequest * add BearerTokenCredentialPolicy for import
pjquirk · Jun 1, 2019 · 2f39cf1 · 2f39cf1
1 parent bd7206d
commit 2f39cf1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/sdk/core/azure-core/README.md b/sdk/core/azure-core/README.md
@@ -21,6 +21,7 @@ from azure.core.pipeline.policies import (
     HeadersPolicy,
     RetryPolicy,
     RedirectPolicy,
+    BearerTokenCredentialPolicy,
     ContentDecodePolicy
 )
 

diff --git a/sdk/core/azure-core/azure/core/__init__.py b/sdk/core/azure-core/azure/core/__init__.py
@@ -36,6 +36,7 @@
     "Configuration",
     "PipelineClient",
     "HttpResponseError",
+    "HttpRequest",
 ]
 
 try:

diff --git a/sdk/core/azure-core/azure/core/exceptions.py b/sdk/core/azure-core/azure/core/exceptions.py
@@ -67,7 +67,6 @@ class AzureError(Exception):
 
     def __init__(self, message, *args, **kwargs):
         self.inner_exception = kwargs.get('error')
-        self.response = kwargs.get('response')
         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
@@ -102,6 +101,7 @@ class HttpResponseError(AzureError):
 
     def __init__(self, message=None, response=None, **kwargs):
         self.reason = None
+        self.response = response
         if response:
             self.reason = response.reason
         message = "Operation returned an invalid status code '{}'".format(self.reason)
@@ -116,7 +116,7 @@ def __init__(self, message=None, response=None, **kwargs):
                     message = self.error.message #pylint: disable=no-member
         except AttributeError:
             pass
-        super(HttpResponseError, self).__init__(message=message, response=response, **kwargs)
+        super(HttpResponseError, self).__init__(message=message, **kwargs)
 
 
 class DecodeError(HttpResponseError):