Quellcode durchsuchen

[videos] Put some views behind login

Colin Powell vor 1 Jahr
Ursprung
Commit
e00f2de4b1
2 geänderte Dateien mit 8 neuen und 8 gelöschten Zeilen
  1. 5 6
      vrobbler/apps/videos/views.py
  2. 3 2
      vrobbler/apps/webpages/views.py

+ 5 - 6
vrobbler/apps/videos/views.py

@@ -1,10 +1,9 @@
+from django.contrib.auth.mixins import LoginRequiredMixin
 from django.views import generic
 from videos.models import Series, Video
 
-# class VideoIndexView():
 
-
-class MovieListView(generic.ListView):
+class MovieListView(LoginRequiredMixin, generic.ListView):
     model = Video
     template_name = "videos/movie_list.html"
 
@@ -12,11 +11,11 @@ class MovieListView(generic.ListView):
         return Video.objects.filter(video_type=Video.VideoType.MOVIE)
 
 
-class SeriesListView(generic.ListView):
+class SeriesListView(LoginRequiredMixin, generic.ListView):
     model = Series
 
 
-class SeriesDetailView(generic.DetailView):
+class SeriesDetailView(LoginRequiredMixin, generic.DetailView):
     model = Series
     slug_field = "uuid"
 
@@ -34,6 +33,6 @@ class SeriesDetailView(generic.DetailView):
         return context_data
 
 
-class VideoDetailView(generic.DetailView):
+class VideoDetailView(LoginRequiredMixin, generic.DetailView):
     model = Video
     slug_field = "uuid"

+ 3 - 2
vrobbler/apps/webpages/views.py

@@ -1,9 +1,10 @@
+from django.contrib.auth.mixins import LoginRequiredMixin
 from django.db.models import Count
 from django.views import generic
 from webpages.models import WebPage
 
 
-class WebPageListView(generic.ListView):
+class WebPageListView(LoginRequiredMixin, generic.ListView):
     model = WebPage
     paginate_by = 20
 
@@ -16,7 +17,7 @@ class WebPageListView(generic.ListView):
         )
 
 
-class WebPageDetailView(generic.DetailView):
+class WebPageDetailView(LoginRequiredMixin, generic.DetailView):
     model = WebPage
     slug_field = "uuid"