Проверить ссылку на профиль пользователя Django один на один

Question

Я добавляю профиль в мое приложение Django, чтобы хранить больше информации о пользователе. Подписан на https://simpleisbetterthancomplex.com/tutorial/2016/07/22/how-to-extend-django-user-model.html#onetoone

Однако при тестировании я получаю сообщение об ошибке. Я использую модель пользователя по умолчанию от django.contrib.auth.

models.py

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


class Profile(models.Model):
    user = models.OneToOneField(get_user_model(), on_delete=models.CASCADE)

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

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

Тесты / test_model.py

from django.test import TestCase
from app.models import Profile


class ProfileModelTests(TestCase):
    @classmethod
    def setUpTestData(cls):
        Profile.objects.create()

    def test_can_create_profile(self):
        pass

ошибка

/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/psycopg2/__init__.py:144: UserWarning: The psycopg2 wheel package will be renamed from release 2.8; in order to keep installing from binary please use "pip install psycopg2-binary" instead. For details see: <http://initd.org/psycopg/docs/install.html#binary-install-from-pypi>.
  """)
..........E............
======================================================================
ERROR: setUpClass (project-api.polls.tests.test_models.ProfileModelTests)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/backends/utils.py", line 85, in _execute
    return self.cursor.execute(sql, params)
psycopg2.IntegrityError: null value in column "user_id" violates not-null constraint
DETAIL:  Failing row contains (158, null).


The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/test/testcases.py", line 1002, in setUpClass
    cls.setUpTestData()
  File "/Users/user/programming/gatsby/project-api/polls/tests/test_models.py", line 86, in setUpTestData
    Profile.objects.create()
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/manager.py", line 82, in manager_method
    return getattr(self.get_queryset(), name)(*args, **kwargs)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/query.py", line 417, in create
    obj.save(force_insert=True, using=self.db)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/base.py", line 729, in save
    force_update=force_update, update_fields=update_fields)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/base.py", line 759, in save_base
    updated = self._save_table(raw, cls, force_insert, force_update, using, update_fields)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/base.py", line 842, in _save_table
    result = self._do_insert(cls._base_manager, using, fields, update_pk, raw)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/base.py", line 880, in _do_insert
    using=using, raw=raw)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/manager.py", line 82, in manager_method
    return getattr(self.get_queryset(), name)(*args, **kwargs)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/query.py", line 1125, in _insert
    return query.get_compiler(using=using).execute_sql(return_id)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/models/sql/compiler.py", line 1283, in execute_sql
    cursor.execute(sql, params)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/backends/utils.py", line 68, in execute
    return self._execute_with_wrappers(sql, params, many=False, executor=self._execute)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/backends/utils.py", line 77, in _execute_with_wrappers
    return executor(sql, params, many, context)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/backends/utils.py", line 85, in _execute
    return self.cursor.execute(sql, params)
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/utils.py", line 89, in __exit__
    raise dj_exc_value.with_traceback(traceback) from exc_value
  File "/Users/user/.local/share/virtualenvs/project-api-HgI9cQzm/lib/python3.6/site-packages/django/db/backends/utils.py", line 85, in _execute
    return self.cursor.execute(sql, params)
django.db.utils.IntegrityError: null value in column "user_id" violates not-null constraint
DETAIL:  Failing row contains (158, null).


----------------------------------------------------------------------
Ran 22 tests in 2.198s

FAILED (errors=1)
Using existing test database for alias 'default'...
System check identified no issues (0 silenced).
Preserving test database for alias 'default'...

Answer 1

@receiver(post_save, sender=get_user_model())

С помощью этого кода вы слушаете метод сохранения модели User, чтобы создать связанную запись Profile как раз вовремя , если ее еще нет. Таким образом, вы можете просто продолжать создавать новых пользователей, не беспокоясь о том, чтобы каждый раз создавать и связывать профили вручную.

Обратите внимание, что обратное не рассматривается: если вы хотите создать запись профиля напрямую, вы должны передать ему существующий объект User, иначе вы получите IntegrityError.

Решение: предпочтительно придерживаться одного основного способа создания пар пользователь + профиль; в вашем случае просто создайте пользователя и позвольте сигналу автоматически вызвать создание профиля.

class ProfileModelTests(TestCase):                                                                                        
    @classmethod                                                                                                          
    def setUpTestData(cls):                                                                                               
        get_user_model().objects.create()                                                                                                                                                                                   

    def test_profile(self):                                                                                    
        profile = get_user_model().objects.last().profile

Answer 2

Идея заключается в Profile.objects.create(user=instance), если пользователь еще не существует, и будет выдано исключение IntegrityError, так как user_id не будет найден. Сигналы будут прослушивать, когда пользователь будет создан не наоборот. В вашем тестовом примере Profile.objects.create() попытается создать отношение «один к одному» с несуществующим пользователем.