1
0
mirror of https://github.com/fspc/workstand.git synced 2025-02-23 09:13:23 -05:00

Protect view.

This commit is contained in:
Drew Larson 2016-06-23 17:37:44 -06:00
parent fb57443b76
commit c1e92efd2f

View File

@ -5,6 +5,8 @@ from django.core.urlresolvers import reverse
from django.http import HttpResponseRedirect from django.http import HttpResponseRedirect
from django.template.response import TemplateResponse from django.template.response import TemplateResponse
from django.views.generic import TemplateView, View from django.views.generic import TemplateView, View
from django.utils.decorators import method_decorator
from django.contrib.auth.decorators import login_required
from registration.models import Member from registration.models import Member
@ -13,11 +15,12 @@ from .forms import MembershipForm, PaymentForm
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@method_decorator(login_required, name='dispatch')
class DashboardView(View): class DashboardView(View):
def get(self, request): def get(self, request):
return TemplateResponse(request, 'dashboard.html') return TemplateResponse(request, 'dashboard.html')
@method_decorator(login_required, name='dispatch')
class NewMembershipView(TemplateView): class NewMembershipView(TemplateView):
template_name = 'membership_form.html' template_name = 'membership_form.html'