# This file is part of photo21 # Copyright (C) 2021-2022 Amicale des élèves de l'ENS Paris-Saclay # SPDX-License-Identifier: GPL-3.0-or-later import os from django.conf import settings from django.contrib.auth import get_user_model from django.contrib.auth.mixins import LoginRequiredMixin from django.http import FileResponse, Http404 from django.views.generic import ListView, View from photologue.models import Gallery class MediaAccess(View): def get(self, request, path): if not request.user.is_authenticated and not request.session.get('public_gallery_access'): from django.contrib.auth.views import redirect_to_login return redirect_to_login(request.get_full_path()) media_root = os.path.realpath(settings.MEDIA_ROOT) file_path = os.path.realpath(os.path.join(media_root, path)) if not file_path.startswith(media_root + os.sep): raise Http404 if not os.path.isfile(file_path): raise Http404 response = FileResponse(open(file_path, 'rb')) response['Cache-Control'] = 'max-age=2678400' return response class IndexView(LoginRequiredMixin, ListView): queryset = Gallery.objects.all() paginate_by = 4 template_name = "index.html" def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) # Get superusers user_model = get_user_model() superusers = user_model.objects.filter(is_superuser=True) context["superusers"] = superusers return context