СвязанныйObjectDoesNotExist при использовании пользовательской модели пользователя в Django

Question

У меня есть приложение для учетной записи, в котором я создал модель профиля, расширив пользовательскую модель пользователя. Я создал представление, которое позволяет пользователю редактировать информацию своего профиля, а также иметь соответствующие UserEditForm и ProfileEditForm. На данный момент у пользователя нет профиля, поэтому, когда я открываю форму редактирования, я получаю сообщение об ошибке: «RelatedObjectDoesNotExist at/account/edit /». «Пользователь не имеет профиля» Я попытался создать профиль с помощью администратора, после чего ошибка исчезнет. Как я могу исправить это в файле views.py.

views.py

@login_required 
def edit(request): 
    if request.method =='POST': 
     user_form = UserEditForm(instance=request.user,data=request.POST) 

     profile_form =  ProfileEditForm(instance=request.user.profile,data=request.POST,files=request.FILES) 

     if user_form.is_valid() and profile_form.is_valid(): 
      user_form.save() 
      profile_form.save() 
      messages.success(request,'Profile updated successfully') 

     else: 
      messages.error(request,'Error updating your profile') 

    else: 
     user_form = UserEditForm(instance=request.user) 

     profile_form = ProfileEditForm(instance=request.user.profile) 


    context = { 
     'user_form':user_form, 
     'profile_form': profile_form 
    } 
    return render(request,'account/edit.html',context) 

models.py

CATEGORY_CHOICES = (
    ('SA','School Admin'), 
    ('T','Teacher'), 
    ('S','Student'), 
    ('P','Parent'), 
) 


class Profile(models.Model): 
    eduser = models.OneToOneField(settings.AUTH_USER_MODEL) 
    photo = models.ImageField(upload_to='users/%Y/%m/%d',blank=True) 
    about_me = models.TextField(max_length=200,blank=True) 
    category = models.CharField(max_length=1,choices=CATEGORY_CHOICES,blank=True) 
    date_of_birth = models.DateField(blank=True,null=True) 

    def __str__(self): 
     return 'Profile for user {}'.format(self.eduser.username) 

forms.py

class UserEditForm(forms.ModelForm): 
    class Meta: 
     model = User 
     fields = ('first_name','last_name','email') 

class ProfileEditForm(forms.ModelForm): 

    class Meta: 
     model = Profile 
     fields = ('category','date_of_birth','about_me','photo') 

Answer 1

Вы должны поймать ошибку. Вы можете сделать это в верхней части функции:

try: 
    profile = request.user.profile 
except ObjectDoesNotExist: 
    profile = Profile(user=request.user) 

и передать эту profile в ProfileEditForm в обоих, если ветви.