fediseer/overseer/exceptions.py

from werkzeug import exceptions as wze
from loguru import logger

class BadRequest(wze.BadRequest):
    def __init__(self, message, log=None):
        self.specific = message
        self.log = log

class Forbidden(wze.Forbidden):
    def __init__(self, message, log=None):
        self.specific = message
        self.log = log

class Unauthorized(wze.Unauthorized):
    def __init__(self, message, log=None):
        self.specific = message
        self.log = log

class NotFound(wze.NotFound):
    def __init__(self, message, log=None):
        self.specific = message
        self.log = log

class Locked(wze.Locked):
    def __init__(self, message, log=None):
        self.specific = message
        self.log = log

def handle_bad_requests(error):
    '''Namespace error handler'''
    if error.log:
        logger.warning(error.log)
    return({'message': error.specific}, error.code)