from functools import wraps

from maintenance_mode.http import get_maintenance_response


def force_maintenance_mode_off(view_func):
    @wraps(view_func)
    def wrapper(request, *args, **kwargs):
        return view_func(request, *args, **kwargs)

    wrapper.__dict__["force_maintenance_mode_off"] = True
    return wrapper


def force_maintenance_mode_on(view_func):
    @wraps(view_func)
    def wrapper(request, *args, **kwargs):
        return get_maintenance_response(request)

    wrapper.__dict__["force_maintenance_mode_on"] = True
    return wrapper