diff --git a/src/InSillyCloWeb/pre_basetheme_bootstrap/views.py b/src/InSillyCloWeb/pre_basetheme_bootstrap/views.py index 9bcea6add1baf3f0d474c4e1335e7e79d15f9851..9507cca322efb7d297bca4e92defa9afbb3772b7 100644 --- a/src/InSillyCloWeb/pre_basetheme_bootstrap/views.py +++ b/src/InSillyCloWeb/pre_basetheme_bootstrap/views.py @@ -1,5 +1,7 @@ from basetheme_bootstrap.forms import AuthenticationForm +from django.contrib import messages from django.contrib.auth import login +from django.contrib.messages.context_processors import messages from django.contrib.sessions.models import Session from django.shortcuts import render, redirect @@ -22,19 +24,22 @@ def loginView(request): if session_key: try: session = Session.objects.get(session_key=session_key) - models.SimulatorJob.objects.filter( - owning_session=session, - ).update( - owner=user, - owning_session=None, - ) - - models.Assembly.objects.filter( - owning_session=session, - ).update( - owner=user, - owning_session=None, - ) + for klass in [ + models.SimulatorJob, + models.Assembly, + ]: + instance_to_migrate = klass.objects.filter( + owning_session=session, + ) + if instance_to_migrate.exists(): + messages.success( + f"We successfully associated {instance_to_migrate.count()} " + f"{klass._meta.verbose_name_plural.title()} with your account." + ) + instance_to_migrate.update( + owner=user, + owning_session=None, + ) except Session.DoesNotExist: pass