26 lines
637 B
Python
26 lines
637 B
Python
from flask import jsonify
|
|
|
|
|
|
class ApiException(Exception):
|
|
status_code = 500
|
|
|
|
def __init__(self, status_code: int, message) -> None:
|
|
super().__init__()
|
|
self.status_code = status_code
|
|
self.message = message
|
|
|
|
def to_response(self):
|
|
rv = jsonify(self.message)
|
|
rv.status_code = self.status_code
|
|
return rv
|
|
|
|
|
|
class NotFoundException(ApiException):
|
|
def __init__(self, entity_name) -> None:
|
|
super().__init__(404, f'{entity_name} could not be found')
|
|
|
|
|
|
class BadRequestException(ApiException):
|
|
def __init__(self, message) -> None:
|
|
super().__init__(400, message)
|