23 lines
783 B
Python
23 lines
783 B
Python
|
from django.contrib import messages
|
||
|
|
from django.contrib.auth.views import LoginView
|
||
|
|
from django.utils.decorators import method_decorator
|
||
|
|
|
||
|
|
from django.views.decorators.csrf import csrf_protect
|
||
|
|
|
||
|
|
|
||
|
|
@method_decorator(csrf_protect, name='dispatch')
|
||
|
|
class CustomLoginView(LoginView):
|
||
|
|
template_name = 'accounts/login.html'
|
||
|
|
|
||
|
|
def form_valid(self, form):
|
||
|
|
remember_me = form.cleaned_data.get('remember_me')
|
||
|
|
if remember_me:
|
||
|
|
self.request.session.set_expiry(1209600) # 2 weeks
|
||
|
|
else:
|
||
|
|
self.request.session.set_expiry(0) # Browser close
|
||
|
|
return super(CustomLoginView, self).form_valid(form)
|
||
|
|
|
||
|
|
def form_invalid(self, form):
|
||
|
|
messages.error(self.request, '用户名或密码错误。')
|
||
|
|
return super().form_invalid(form)
|