Advertisement
Guest User

Untitled

a guest
Sep 19th, 2016
133
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.58 KB | None | 0 0
  1. models.py:
  2. class UserProfile(models.Model):
  3. user = models.OneToOneField(User)
  4. first_name = models.CharField(max_length=100, blank=True)
  5. last_name = models.CharField(max_length=100, blank=True)
  6. email = models.EmailField(max_length=100, blank=True)
  7. company_name = models.CharField(max_length=100, blank=True, null=True)
  8. website = models.URLField(max_length=100, blank=True, null=True)
  9. phone_number = models.CharField(max_length=100, blank=True, null=True)
  10. fax_number = models.CharField(max_length=100, blank=True, null=True)
  11. country = models.CharField(max_length=100, blank=True, null=True)
  12. address = models.CharField(max_length=100, blank=True, null=True)
  13.  
  14. def __str__(self):
  15. return self.user.username
  16.  
  17. forms.py:
  18. class UserForm(forms.ModelForm):
  19. password = forms.CharField(widget=forms.PasswordInput())
  20.  
  21. class Meta:
  22. model = User
  23. fields = ('username', 'password')
  24.  
  25.  
  26. class UserProfileForm1(forms.ModelForm):
  27. class Meta:
  28. model = UserProfile
  29. fields = ('first_name', 'last_name', 'email')
  30.  
  31. class UserProfileForm2(forms.ModelForm):
  32. class Meta:
  33. model = UserProfile
  34. fields = ('company_name', 'website', 'phone_number', 'fax_number',
  35. 'country', 'address')
  36.  
  37. views.py
  38. def index(request):
  39. registered = False
  40.  
  41. if request.method == 'POST':
  42. user_form = UserForm(data=request.POST)
  43. profile_form = UserProfileForm1(data=request.POST)
  44.  
  45. if user_form.is_valid() and profile_form.is_valid():
  46. user = user_form.save()
  47. user.set_password(user.password)
  48. user.save()
  49.  
  50. profile = profile_form.save(commit=False)
  51. profile.user = user
  52. profile.save()
  53. registered = True
  54. else:
  55. print user_form.errors, profile_form.errors
  56. else:
  57. user_form = UserForm()
  58. profile_form = UserProfileForm1()
  59.  
  60. context = {'user_form': user_form, 'profile_form': profile_form, 'registered': registered}
  61. return render(request, '.../register.html', context)
  62.  
  63. def register_page_two(request):
  64. if request.method == 'POST':
  65. partners_form = UserProfileForm2(data=request.POST)
  66.  
  67. if partners_form.is_valid():
  68.  
  69. profile = partners_form.save(commit=False)
  70. profile.save()
  71. else:
  72. print partners_form.errors
  73. else:
  74. partners_form = UserProfileForm2()
  75.  
  76. context = {'partners_form': partners_form}
  77.  
  78. return render(request, '.../register_second.html', context)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement