class WpApiError(Exception):

def __init__(self, status_code, code='', message='', data={}):

super(WpApiError, self).__init__(message)

self.message = message

if status_code is not None:

self.status_code = status_code

self.data = data

@staticmethod

def factory(response):

status_code = response.status_code

payload = response.json()

code = payload.get('code')

message = payload.get('message')

data = payload.get('data')

print('status code', status_code)

if status_code == 400:

return BadRequestWpApiError(code, message, data)

elif status_code == 401:

return UnauthorizedWpApiError(code, message, data)

elif status_code == 403:

return ForbiddenWpApiError(code, message, data)

elif status_code == 404:

return NotFoundWpApiError(code, message, data)

elif status_code == 500:

return InternalErrorWpApiError(code, message, data)

else:

return WpApiError(status_code, code, message, data)

class BadRequestWpApiError(WpApiError):

def __init__(self, code='', message='', data={}):

super(BadRequestWpApiError, self).__init__(400, code, message, data)

class UnauthorizedWpApiError(WpApiError):

def __init__(self, code='', message='', data={}):

super(UnauthorizedWpApiError, self).__init__(401, code, message, data)

class ForbiddenWpApiError(WpApiError):

def __init__(self, code='', message='', data={}):

super(ForbiddenWpApiError, self).__init__(403, code, message, data)

class NotFoundWpApiError(WpApiError):

def __init__(self, code='', message='', data={}):

super(NotFoundWpApiError, self).__init__(404, code, message, data)

class InternalErrorWpApiError(WpApiError):

def __init__(self, code='', message='', data={}):

super(InternalErrorWpApiError, self).__init__(500, code, message, data)