def login(request):
remove_session(request)
login_form = LoginForm(request.POST or None)
if request.method == 'POST':
if login_form.is_valid():
user = User.return_valid_user(
email_address=request.POST['email'],
raw_password=request.POST['password'])
if user != None:
request.session[SESSION_KEY] = user.id
return redirect('home')
errors = login_form._errors.setdefault("email", ErrorList())
errors.append("Login Error!")
return render(request, 'login.html', {
'login_form': login_form,
'registration_form': RegisterForm()
})
else:
registration_form = RegisterForm()
return render(
request, 'login.html', {
'login_form': login_form,
'registration_form': registration_form,
'user': None
})
def login(request):
remove_session(request)
login_form = LoginForm(request.POST or None)
if request.method == 'POST':
if login_form.is_valid():
user = User.return_valid_user(email_address=request.POST['email'], raw_password=request.POST['password'])
if user != None:
request.session[SESSION_KEY] = user.id
return redirect('home')
errors = login_form._errors.setdefault("email", ErrorList())
errors.append("Login Error!")
return render(request, 'login.html', {'login_form': login_form, 'registration_form' : RegisterForm()})
else:
registration_form = RegisterForm()
return render(request, 'login.html', {'login_form': login_form, 'registration_form' : registration_form, 'user' : None})
def test_registration_page(self):
#setup
response = self.client.post('/register',
data={
'first_name': self.user.first_name,
'second_name': self.user.second_name,
'email': self.user.email,
'password': self.user.password
})
test_user = User.return_valid_user(self.user.email, self.user.password)
#Test
self.assertEqual(response.status_code, 200)
self.assertContains(response, 'Registration success')
self.assertTrue(
test_helper.obj_compare(self.user, test_user, ['id', '_state']))
