def get(self, request, *args, **kwargs):
if request.user.is_authenticated():
return HttpResponseRedirect('/')
else:
user_form = UserForm()
context = {'user_form': user_form}
return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
user_form = UserForm(data=request.POST)
player_form = PlayerForm(data=request.POST)
if user_form.is_valid() and player_form.is_valid():
user = user_form.save()
user.set_password(user.password)
user.save()
player = Player(user=user)
player.save()
user = authenticate(username=user_form.cleaned_data['username'],
password=user_form.cleaned_data['password'])
login(request, user)
return HttpResponseRedirect('/')
else:
context = {'user_form': user_form}
messages.error(request, 'Something went wrong...')
return render(request, self.template_name, context)
