def companyregister(request):
registered =False
if request.method == 'POST':
user_form = UserForm(data=request.POST)
profile_form = CompanyProfileForm(data=request.POST)
if user_form.is_valid() and profile_form.is_valid():
user = user_form.save()
user.set_password(user.password)
user.save()
g=Group.objects.get(name="company")
g.user_set.add(user)
profile = profile_form.save(commit=False)
profile.user = user
profile.save()
registered = True
else:
print user_form.errors, profile_form.errors
else:
user_form = UserForm()
profile_form = CompanyProfileForm()
return render(request,
'userpro/companyregister.html',
{'user_form': user_form, 'profile_form': profile_form, 'registered':registered})
def register(request):
registered = False
if request.user.is_authenticated():
return HttpResponseRedirect('/')
if request.method == 'POST':
user_form = UserForm(request.POST)
profile_form = UserProfileForm(request.POST, request.FILES)
if user_form.is_valid() and profile_form.is_valid():
user = user_form.save()
user = user_form.save(commit=False)
user.save()
g=Group.objects.get(name="student")
g.user_set.add(user)
profile = profile_form.save(commit=False)
profile.user = user
profile.save()
registered = True
return HttpResponseRedirect('/')
else:
print user_form.errors, profile_form.errors
else:
user_form = UserForm()
profile_form = UserProfileForm()
return render(request,
'userpro/register.html',
{'user_form': user_form, 'profile_form': profile_form, 'registered':registered})
