Add delete photo view

photologue/views.py

@@ -16,7 +16,7 @@ from django.urls import reverse_lazy
 from django.utils.text import slugify
 from django.views.generic.dates import ArchiveIndexView, YearArchiveView
 from django.views.generic.detail import DetailView
-from django.views.generic.edit import FormView
+from django.views.generic.edit import FormView, DeleteView
 from PIL import Image
 
 from .forms import UploadForm
@@ -56,6 +56,18 @@ class PhotoDetailView(LoginRequiredMixin, DetailView):
             return qs.filter(is_public=True)
 
 
+class PhotoDeleteView(PermissionRequiredMixin, DeleteView):
+    model = Photo
+    permission_required = 'photologue.delete_photo'
+
+    def get_success_url(self):
+        galleries = self.object.galleries.all()
+        if not galleries:
+            return reverse_lazy('photologue:pl-gallery-archive')
+        slug = galleries[0].slug
+        return reverse_lazy('photologue:pl-gallery', args=[slug])
+
+
 class TagDetail(LoginRequiredMixin, DetailView):
     model = Tag