dashboard/accounts/decorators.py

# Copyright © 2022 Aravinth Manivannan <realaravinth@batsense.net>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
from django.shortcuts import redirect
from django.urls import reverse

from .utils import ConfirmAccess


def confirm_access(function):
    def wrap(request, *args, **kwargs):
        return ConfirmAccess.validate_decorator(
            request=request, fn=function, *args, **kwargs
        )

    return wrap


def redirect_if_authenticated(fn):
    """
    Redirect authenticated users visiting sign in/sign up views
    """

    def wrap(request, *args, **kwargs):
        if request.user.is_authenticated:
            data = request.GET if request.method == "GET" else request.POST
            if "next" in data:
                return redirect(data["next"])
            return redirect(reverse("accounts.home"))
        return fn(request, *args, **kwargs)

    return wrap