Как правильно расширить модель django User

Question

Здесь я пытаюсь добавить поля модели профиля вместе с полями модели пользователя. Но это не работает, но не добавляет данные моделей профиля. Но after adding когда я обновляю пользователя, я могу обновить обе модели user и profile вместе. Но при регистрации пользователя я не могу добавить profile данные моделей. Можно ли одновременно добавлять данные модели профиля вместе с моделью пользователя?

models.py

class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    first_name = models.CharField(max_length=100,blank=True,null=True)
    last_name = models.CharField(max_length=100,blank=True, null=True)
    location = models.CharField(max_length=100,blank=True, null=True)
    phone = models.CharField(max_length=20,blank=True,null=True)
    image = models.ImageField(upload_to='user image',default='default.jpg')


    @receiver(post_save, sender=User)
    def create_user_profile(sender, instance, created, **kwargs):
        if created:
            Profile.objects.create(user=instance)

    @receiver(post_save, sender=User)
    def save_user_profile(sender, instance, **kwargs):
        instance.profile.save()

forms.py

class RegisterForm(UserCreationForm):
    first_name = forms.CharField(max_length=100, required=False)
    last_name = forms.CharField(max_length=100, required=False)
    location = forms.CharField(max_length=100, required=False)
    phone = forms.CharField(max_length=20, required=False)
    image = forms.FileField(required=False, initial='default.jpg')

    def clean_email(self):
        email = self.cleaned_data['email']
        if User.objects.filter(email=email).exists():
            raise ValidationError('Email Already Exists')

        return email

    class Meta:
        model = User
        fields = ['username', "email", "password1", "password2",'first_name','last_name','location','phone','image','is_superuser', 'is_staff', 'is_active']

views.py

def add_user(request):
    if request.method == "POST":
        form = RegisterForm(request.POST or None,request.FILES or None)
        if form.is_valid():
            user = form.save(commit=False)
            user.save()
            messages.success(request, 'user created with username {}'.format(user.username))
            return redirect('list_user')
    else:
        form = RegisterForm()
    return render(request, 'add_user.html', {'form': form})

def update_user(request, id): #the problem while  updating is if i don't select the new image then it saves the default image.But other works fine
    user = get_object_or_404(User, id=id)
    if request.method == 'POST':

        form = EditRegisterForm(request.POST or None,request.FILES or None,instance=user)
        if form.is_valid():
            phone = form.cleaned_data['phone']
            location = form.cleaned_data['location']
            image = form.cleaned_data['image']
            first_name = form.cleaned_data['first_name']
            last_name = form.cleaned_data['last_name']
            user = form.save(commit=False)
            user.profile.phone = phone
            user.profile.location = location
            user.profile.first_name = first_name
            user.profile.last_name = last_name
            user.profile.image = image
            user.save()
            messages.success(request, ' {}  updated.'.format(user.username))
            return redirect('list_user')

add_user.html

 <form action="" method="post" enctype="multipart/form-data">
                            {% csrf_token %}
                             {% bootstrap_form form %}
                            <div>
                                <button type="submit" class="btn btn-info">Add</button>
                            </div>
                        </form>

Answer 1

Хм, вы уже создаете профиль с помощью сигналов. Так почему бы вам не использовать это так:

def update_user(request, id):
    # Some code
    if form.is_valid():
        phone = form.cleaned_data['phone']
        location = form.cleaned_data['location']
        image = form.cleaned_data['image']
        first_name = form.cleaned_data['first_name']
        last_name = form.cleaned_data['last_name']
        <b>user = form.save()
        user.refresh_from_db()</b>
        user.profile.phone = phone
        user.profile.location = location
        user.profile.first_name = first_name
        user.profile.last_name = last_name
        user.profile.image = image
        user.profile.save()
        messages.success(request, ' {}  updated.'.format(user.username))
        return redirect('list_user')

Я использую refresh_from_db() для перезагрузки объекта. Поскольку профиль был создан с помощью сигналов, обновление объекта User должно привести к перезагрузке модели профиля.

Answer 2

Вместо того, чтобы пытаться сделать это с вашей собственной логикой, рассмотрите возможность использования декоратора @action.atomic по вашему мнению.

Ничто не будет сохранено в вашей БД, если что-либо вызовет исключение Документ здесь: https://docs.djangoproject.com/en/2.2/topics/db/transactions/#django.db.transaction.atomic

Вы можете просто сделать это:

# remove your signals first, they're not required if you manually create the profile here

@transaction.atomic
def add_user(request):
    if request.method == "POST":
        form = RegisterForm(request.POST or None,request.FILES or None)
        if form.is_valid():
            user = form.save(commit=False)
            user.save()
            Profile.objects.create(
                user=user,
                first_name=form.cleaned_data['first_name'],
                #...and other attributes
            )
            messages.success(request, 'user created with username {}'.format(user.username))
            return redirect('list_user')
    else:
        form = RegisterForm()
    return render(request, 'add_user.html', {'form': form})```