diff --git a/vrobbler/apps/videos/views.py b/vrobbler/apps/videos/views.py index b20b982..ee9aa14 100644 --- a/vrobbler/apps/videos/views.py +++ b/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" diff --git a/vrobbler/apps/webpages/views.py b/vrobbler/apps/webpages/views.py index 9c33514..e05adc3 100644 --- a/vrobbler/apps/webpages/views.py +++ b/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"