diff --git a/PDMI/store/views.py b/PDMI/store/views.py index 26dec6a..4fa8d57 100644 --- a/PDMI/store/views.py +++ b/PDMI/store/views.py @@ -1,3 +1,4 @@ +from django.contrib.auth.mixins import LoginRequiredMixin from django.shortcuts import render from django.urls import reverse_lazy from bootstrap_modal_forms.generic import BSModalCreateView, BSModalLoginView @@ -8,19 +9,22 @@ import os import PDMI.settings as settings from .response import response_mimetype, JsonResponse + class SignUpView(BSModalCreateView): form_class = CustomUserCreationForm template_name = 'store/signup.html' success_message = 'Success: Sign up succeeded. You can now Log in.' success_url = reverse_lazy('store_front_page') + class CustomLoginView(BSModalLoginView): authentication_form = CustomAuthenticationForm template_name = 'store/login.html' success_message = 'Success: You were successfully logged in.' extra_content = dict(success_url=reverse_lazy('index')) -class UploadView(FormView): + +class UploadView(LoginRequiredMixin, FormView): form_class = FileFieldForm template_name = 'store/upload.html'