表單（擴展）用戶

Question

我已經擴展了我的用戶模型並添加了一些額外的字段。我希望能夠通過網絡界面（而不是管理頁面）添加新用戶，因此需要某種形式。我如何在數據庫中創建這個新用戶以及表單的外觀如何？

class MyUser(models.Model): 
    user  = models.OneToOneField(User) 
    custcode = models.CharField(max_length=30) 

//這顯然是行不通的，但它是一個開始

from django.contrib.auth.forms import UserCreationForm 
from django.contrib.auth.models import User 

class UserCreateForm(UserCreationForm): 

    class Meta: 
     model = User 
     fields = ('username', 'custcode') 

Answer 1

既然你有兩個模型s，你需要兩種形式。

auth.User和yourapp.MyUser的一種形式。

在你看來，你會：

1. 實例化兩種形式
2. 檢查兩種形式都有效
3. 保存用戶的形式
4. 保存MYUSER形式，使用user.pk

這應該給那樣的東西

class MyUserCreationForm(forms.ModelForm): 
    class Meta: 
     model = User 
     exclude = ('user',) 
     # Exclude 'user' because we will set it manually 

class MyView(View): 

    def get(self, request): 
     user_form = UserCreationForm(prefix='user') 
     myuser_form = MyUserCreationForm(prefix='myuser') 
     # Use prefix to avoid name conflicts 
     return render(...) 

    def post(self, request): 
     user_form = UserCreationForm(request.POST, prefix='user') 
     myuser_form = MyUserCreationForm(request.POST, prefix='myuser') 
     user_form_is_valid = user_form.is_valid() 
     myuser_form_is_valid = myuser_form.is_valid() 
     # Do form.is_valid() here in order to always call both 
     if user_form_is_valid and myuser_form_is_valid: 
      user = user_form.save() 
      myuser = myuser_form.save(commit=False) 
      myuser.user = user 
      myuser.save() 
      ... 
     ... 
     return ... 

爲了保持一致性，表單保存也應包含在交易中。但那是另一個話題。

Answer 2

你可以做到這一點像這樣

class UserCreateForm(UserCreationForm): 
    custcode = form.CharField() 

    class Meta: 
     model = User 
     fields = ('username', 'custcode') 

，寫自定義保存功能對你形成像這

def save(self, commit=True): 
    user = super(UserCreateForm, self).save(commit=True) 
    myuser = MyUser.objects.get_or_create(user=user)[0] 
    myuser.save()