Handle database integrity error during upload

photologue_custom/views.py

@@ -8,15 +8,18 @@ from io import BytesIO
 from django.contrib import messages
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.core.mail import mail_managers
+from django.db import IntegrityError
 from django.http import HttpResponse
 from django.urls import reverse_lazy
+from django.utils.text import slugify
 from django.views.generic.detail import DetailView
 from django.views.generic.edit import FormView
-from photologue.models import Gallery
+from photologue.models import Gallery, Photo
 from photologue.views import GalleryArchiveIndexView, GalleryYearArchiveView
 from taggit.models import Tag
 
 from .forms import UploadForm
+from .models import PhotoExtended
 
 
 class TagDetail(LoginRequiredMixin, DetailView):
@@ -109,10 +112,27 @@ class GalleryUpload(FormView):
         # Upload photos
         # We take files from the request to support multiple upload
         files = self.request.FILES.getlist('file_field')
-        form.save(self.request.user, files)
+        gallery = form.get_or_create_gallery()
+        failed_upload = 0
+        for photo_file in files:
+            title = f"{gallery.title} - {photo_file.name}"
+            try:
+                photo = Photo(title=title, slug=slugify(title))
+                photo.image.save(photo_file.name, photo_file)
+                photo.save()
+                photo.galleries.set([gallery])
+                PhotoExtended.objects.create(photo=photo, owner=self.request.user)
+            except IntegrityError:
+                messages.error(self.request, f"{photo_file.name} was not uploaded. Maybe the photo was already uploaded.")
+                failed_upload += 1
 
         # Notify user then managers
-        messages.success(self.request, "Photos has been successfully uploaded.")
+        if not failed_upload:
+            messages.success(self.request, "All photos has been successfully uploaded.")
+        else:
+            n_success = len(files) - failed_upload
+            messages.warning(self.request, f"Only {n_success} photos were successfully uploaded !")
+
         gallery_title = form.cleaned_data['gallery'] or form.cleaned_data.get('new_gallery_title', '')
         photos = ", ".join(f.name for f in files)
         mail_managers(