import logging

from maintenance_mode.core import get_maintenance_mode


class RequireNotMaintenanceMode503(logging.Filter):
    """
    Filters out 503 errors if maintenance mode is activated.
    """

    def filter(self, record):
        """
        Return False if maintenance mode is on and
        the given record has a status code of 503.
        """
        status_code = getattr(record, "status_code", None)
        if get_maintenance_mode() and status_code == 503:
            return False
        return True