extended UserCreationForm to RegistrationForm, adding fields for email/firstname/lastname

signup/forms.py

@@ -0,0 +1,12 @@
+from django import forms
+from django.contrib.auth.forms import UserCreationForm
+from django.contrib.auth.models import User
+
+class RegistrationForm(UserCreationForm):
+    email = forms.EmailField(label = "Email",widget=forms.TextInput(), required=True)
+    first_name = forms.CharField(label="Prénom", widget=forms.TextInput(),required=True)
+    last_name = forms.CharField(label="Nom", widget=forms.TextInput(),required=True)
+
+    class Meta:
+        model = User
+        fields = ["username", "password1", "password2", "email", "first_name", "last_name"]

signup/views.py

@@ -4,16 +4,18 @@ from django.http.response import HttpResponse
 
 
 from django.shortcuts import redirect, render
-from django.contrib.auth.forms import UserCreationForm
+from .forms import RegistrationForm
-from django.contrib.auth.models import User
 def signup(request):
     if request.method == 'POST':
-        form = UserCreationForm(request.POST)
+        form = RegistrationForm(request.POST)
         if form.is_valid():
             user = form.save()
+            user.first_name = form.cleaned_data.get('first_name')
+            user.last_name = form.cleaned_data.get('last_name')
+            user.email = form.cleaned_data.get('email')
             login(request, user)
             return redirect('/')
         return render(request,'signup.html', {'form':form})
     else:
-        form = UserCreationForm()
+        form = RegistrationForm()
     return render(request,'signup.html', {'form':form})