在User模型中添加一个额外的字段。可以通过使用Django自带的用户模型来实现。首先,在models.py中导入User模型并定义一个新字段,例如“age”:
from django.contrib.auth.models import User
class UserProfile(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
age = models.IntegerField(blank=True, null=True)
这里定义了一个名为UserProfile的新模型,它包含了一个与User模型一对一关联的字段user和一个新的age字段。
接下来,可以在forms.py中定义一个新的表单,允许用户输入新的字段:
from django.contrib.auth.forms import UserCreationForm
class CustomUserCreationForm(UserCreationForm):
age = forms.IntegerField()
class Meta(UserCreationForm.Meta):
fields = UserCreationForm.Meta.fields + ('age',)
此代码使用Django自带的UserCreationForm,并将自定义的age字段添加到表单中。
最后,在views.py中,使用CustomUserCreationForm替换默认的UserCreationForm:
from django.contrib.auth import login, authenticate
from django.shortcuts import render, redirect
from myapp.forms import CustomUserCreationForm
def signup(request):
if request.method == 'POST':
form = CustomUserCreationForm(request.POST)
if form.is_valid():
user = form.save()
age = form.cleaned_data.get('age')
user_profile = UserProfile.objects.create(user=user, age=age)
login(request, user)
return redirect('home')
else:
form = CustomUserCreationForm()
return render(request, 'signup.html', {'form': form})
这里定义了一个新的视图函数signup,当用户提交表单时,会使用CustomUserCreationForm验证表单。如果表单有效,则创建用户并将用户与UserProfile模型关联。然后自动登录用户并重定向到主页。