dashboard/accounts/decorators.py

44 lines
1.5 KiB
Python

# 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