diff --git a/shiftregister/app/views.py b/shiftregister/app/views.py index c577d48..da806dd 100644 --- a/shiftregister/app/views.py +++ b/shiftregister/app/views.py @@ -1,5 +1,6 @@ from django.shortcuts import render, redirect, get_object_or_404 from .models import Shift, LoginToken, Helper, ShiftRegistration +from django.db import transaction from django.db.models import F, Count, Q, ExpressionWrapper from .forms import RegisterForm, EmptyForm from django.db.models.fields import DateTimeField @@ -118,8 +119,10 @@ def register(request): helper = Helper( name=form.cleaned_data["name"], phone=form.cleaned_data["phone"] ) - helper.save() - token = helper.send_confirmation() + with transaction.atomic(): + helper.save() + token = helper.send_confirmation() + request.session["token"] = token.pk if settings.DEBUG: messages.add_message(