diff --git a/photologue_custom/views.py b/photologue_custom/views.py index f95ea07..5355a28 100644 --- a/photologue_custom/views.py +++ b/photologue_custom/views.py @@ -6,7 +6,8 @@ import zipfile from io import BytesIO from django.contrib import messages -from django.contrib.auth.mixins import LoginRequiredMixin +from django.contrib.auth.mixins import (LoginRequiredMixin, + PermissionRequiredMixin) from django.core.mail import mail_managers from django.db import IntegrityError from django.http import HttpResponse @@ -16,8 +17,8 @@ from django.views.generic.detail import DetailView from django.views.generic.edit import FormView from photologue.models import Gallery, Photo from photologue.views import GalleryArchiveIndexView, GalleryYearArchiveView -from taggit.models import Tag from PIL import Image +from taggit.models import Tag from .forms import UploadForm from .models import PhotoExtended @@ -101,13 +102,14 @@ class GalleryDownload(LoginRequiredMixin, DetailView): return response -class GalleryUpload(FormView): +class GalleryUpload(PermissionRequiredMixin, FormView): """ Form to upload new photos in a gallery """ form_class = UploadForm template_name = "photologue/upload.html" success_url = reverse_lazy("gallery-upload") + permission_required = 'photologue.add_gallery' def form_valid(self, form): # Upload photos