27 lines
1 KiB
Python
27 lines
1 KiB
Python
from functools import wraps
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth.views import redirect_to_login
|
|
from django.contrib.auth import REDIRECT_FIELD_NAME
|
|
|
|
from django.shortcuts import render, redirect, resolve_url
|
|
from django.conf import settings
|
|
|
|
from .models import Account
|
|
|
|
def setup_wanted(view_func, setup_url=None, redirect_field_name=REDIRECT_FIELD_NAME):
|
|
"""
|
|
Decorator for views that checks that the user has completed the setup
|
|
process, redirecting to settings.SETUP_URL if required
|
|
"""
|
|
@wraps(view_func)
|
|
def wrap(request, *args, **kwargs):
|
|
if not request.user.is_authenticated or request.user.account.has_completed_setup:
|
|
return view_func(request, *args, **kwargs)
|
|
else:
|
|
resolved_setup_url = resolve_url(setup_url or settings.SETUP_URL)
|
|
path = request.get_full_path()
|
|
return redirect_to_login(
|
|
path, resolved_setup_url, redirect_field_name)
|
|
return wrap
|
|
|
|
setup_required = login_required(setup_wanted)
|