3b3eff70ca
We were retrying requests with 401 so that we can use the response headers to determine type of authentication. Instead, always send request with auth if we already have credentials (from core.dcos_acs_token). This also fixes multi-part requests that currently break if we have to retry the request (since the stream gets opened by the initial request).
92 lines
2.1 KiB
Python
92 lines
2.1 KiB
Python
import abc
|
|
|
|
|
|
class DCOSException(Exception):
|
|
pass
|
|
|
|
|
|
class DCOSHTTPException(DCOSException):
|
|
""" A wrapper around Response objects for HTTP error codes.
|
|
|
|
:param response: requests Response object
|
|
:type response: Response
|
|
"""
|
|
def __init__(self, response):
|
|
self.response = response
|
|
|
|
def status(self):
|
|
return self.response.status_code
|
|
|
|
def __str__(self):
|
|
return 'Error while fetching [{0}]: HTTP {1}: {2}'.format(
|
|
self.response.request.url,
|
|
self.response.status_code,
|
|
self.response.reason)
|
|
|
|
|
|
class DCOSAuthenticationException(DCOSHTTPException):
|
|
"""A wrapper around Response objects for HTTP Authentication errors (401).
|
|
|
|
:param response: requests Response object
|
|
:type response: Response
|
|
"""
|
|
def __init__(self, response):
|
|
self.response = response
|
|
|
|
def __str__(self):
|
|
return "Authentication failed. Please run `dcos auth login`"
|
|
|
|
|
|
class DCOSAuthorizationException(DCOSHTTPException):
|
|
"""A wrapper around Response objects for HTTP Authorization errors (403).
|
|
|
|
:param response: requests Response object
|
|
:type response: Response
|
|
"""
|
|
def __init__(self, response):
|
|
self.response = response
|
|
|
|
def __str__(self):
|
|
return "You are not authorized to perform this operation"
|
|
|
|
|
|
class DCOSBadRequest(DCOSHTTPException):
|
|
"""A wrapper around Response objects for HTTP Bad Request (400).
|
|
|
|
:param response: requests Response object
|
|
:type response: Response
|
|
"""
|
|
def __init__(self, response):
|
|
self.response = response
|
|
|
|
def __str__(self):
|
|
return "Bad request"
|
|
|
|
|
|
class Error(object):
|
|
"""Abstract class for describing errors."""
|
|
|
|
@abc.abstractmethod
|
|
def error(self):
|
|
"""Creates an error message
|
|
|
|
:returns: The error message
|
|
:rtype: str
|
|
"""
|
|
|
|
raise NotImplementedError
|
|
|
|
|
|
class DefaultError(Error):
|
|
"""Construct a basic Error class based on a string
|
|
|
|
:param message: String to use for the error message
|
|
:type message: str
|
|
"""
|
|
|
|
def __init__(self, message):
|
|
self._message = message
|
|
|
|
def error(self):
|
|
return self._message
|