Расширение модели пользователя django с помощью отношений один к одному

Я очень новичок в dajngo, я пытаюсь получить некоторые дополнительные данные от пользователя и вставить их в таблицу профиля моя модель

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver

class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    bio = models.TextField(max_length=500, blank=True,null=True)
    location = models.CharField(max_length=30, blank=True,null=True)
    birth_date = models.DateField(null=True, blank=True)

@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()

моя форма

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


class UserRegistrationForm(UserCreationForm):
    first_name = forms.CharField(max_length=101)
    last_name = forms.CharField(max_length=101)
    email = forms.EmailField()

    class Meta:
        model = User
        fields = ['username', 'first_name', 'last_name', 'email', 'password1', 'password2']

class ProfileForm(forms.ModelForm):
    class Meta:
        model = Profile
        fields=['bio','location','birth_date']

и мое мнение

def register(request):
    if request.method == 'POST':
        user_form = UserRegistrationForm(request.POST)
        profile_form = ProfileForm(request.POST)
        if user_form.is_valid() and profile_form.is_valid():
            user_form.save()
            profile_form.save()
            messages.success(request, 'Your profile was successfully updated!')
            return redirect('home')
        else:

            messages.error(request,'Please correct the error below.')
    else:
        user_form = UserRegistrationForm()
        profile_form = ProfileForm()
    return render(request, 'users/register.html', {
        'user_form': user_form,
        'profile_form': profile_form
    })

Когда я пытаюсь сохранить форму, я получаю ошибку

django.db.utils.IntegrityError: NOT NULL constraint failed: users_profile.user_id

Я знаю, что это происходит потому, что я вызываю сохранение на profile_form и он не имеет представления об идентификаторе пользователя. Как я могу это исправить?

Вы можете использовать commit=False для редактирования объекта до его сохранения в базе данных. Затем вы можете прикрепить профиль к пользователю и сохранить его, например, так:

if user_form.is_valid() and profile_form.is_valid():
    user = user_form.save()
    profile = profile_form.save(commit=False)
    profile.user = user
    profile.save()

После того, как это будет исправлено, вы можете получить ошибку от сигнала post_save, так как профиль уже создан здесь и существует связь один-к-одному.

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

views.py

def register(request):
    if request.method == 'POST':
        user_form = UserRegistrationForm(request.POST)
        profile_form = ProfileForm(request.POST)
        if user_form.is_valid() and profile_form.is_valid():
            user = user_form.save()
            #user.refresh_from_db()
            user.profile.bio=profile_form.cleaned_data.get('bio')
            user.profile.location=profile_form.cleaned_data.get('location')
            user.save()
            messages.success(request, 'Your account successfully created!')
            return redirect('home')
        else:

            messages.error(request,'Please correct the error below.')
    else:
        user_form = UserRegistrationForm()
        profile_form = ProfileForm()
    return render(request, 'users/register.html', {
        'user_form': user_form,
        'profile_form': profile_form
    })

models.py

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver

class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    bio = models.TextField(max_length=500, blank=True,null=True)
    location = models.CharField(max_length=30, blank=True,null=True)
    birth_date = models.DateField(null=True, blank=True)


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

и никаких изменений над form.py .

Вернуться на верх