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 editcompanyprofile(request): try: company_obj=User.objects.get(user=request.user) except: company_obj=None if request.method=="POST": profile_form = CompanyProfileForm(data=request.POST, instance = request.user.companyprofile) user = request.user if profile_form.is_valid(): profile = profile_form.save(commit=False) profile.user=request.user profile.save() edit = True return HttpResponseRedirect('/') else: print profile_form.errors else: profile_form=CompanyProfileForm() return render(request, 'userpro/companyedit.html', {'profile_form':profile_form})