Миграция Django не может назначить экземпляр пользователя в поле пользователя - PullRequest
Новая
       36

Миграция Django не может назначить экземпляр пользователя в поле пользователя

0 голосов
/ 03 октября 2019

Проблема

Недавно я создал новую миграцию для модели SubscriptionAccount, которая заменяет поле email ссылкой на модель User. В этой миграции я также выполняю небольшую миграцию данных, которая выбирает пользователей с электронной почтой и устанавливает для нее значение SubscriptionAccount.user. Однако по какой-то причине миграция не хочет назначать пользователя только что созданному полю user. Жалуется, что он не может назначить User экземпляр муравья, который ожидает экземпляр UserЧто странно, потому что именно это я и делаю.

Ошибка

Возникает следующая ошибка: 
Traceback (most recent call last):
  File "./manage.py", line 13, in <module>
    execute_from_command_line(sys.argv)
  File "/usr/local/lib/python3.6/site-packages/django/core/management/__init__.py", line 364, in execute_from_command_line
    utility.execute()
  File "/usr/local/lib/python3.6/site-packages/django/core/management/__init__.py", line 356, in execute
    self.fetch_command(subcommand).run_from_argv(self.argv)
  File "/usr/local/lib/python3.6/site-packages/django/core/management/base.py", line 283, in run_from_argv
    self.execute(*args, **cmd_options)
  File "/usr/local/lib/python3.6/site-packages/django/core/management/base.py", line 330, in execute
    output = self.handle(*args, **options)
  File "/usr/local/lib/python3.6/site-packages/django/core/management/commands/migrate.py", line 204, in handle
    fake_initial=fake_initial,
  File "/usr/local/lib/python3.6/site-packages/django/db/migrations/executor.py", line 115, in migrate
    state = self._migrate_all_forwards(state, plan, full_plan, fake=fake, fake_initial=fake_initial)
  File "/usr/local/lib/python3.6/site-packages/django/db/migrations/executor.py", line 145, in _migrate_all_forwards
    state = self.apply_migration(state, migration, fake=fake, fake_initial=fake_initial)
  File "/usr/local/lib/python3.6/site-packages/django/db/migrations/executor.py", line 244, in apply_migration
    state = migration.apply(state, schema_editor)
  File "/usr/local/lib/python3.6/site-packages/django/db/migrations/migration.py", line 126, in apply
    operation.database_forwards(self.app_label, schema_editor, old_state, project_state)
  File "/usr/local/lib/python3.6/site-packages/django/db/migrations/operations/special.py", line 193, in database_forwards
    self.code(from_state.apps, schema_editor)
  File "/project/hypernodedb/migrations/0069_link_subscription_account_to_user.py", line 26, in migrate_email_to_user
    subscription_account.user = user
  File "/usr/local/lib/python3.6/site-packages/django/db/models/fields/related_descriptors.py", line 220, in __set__
    self.field.remote_field.model._meta.object_name,
ValueError: Cannot assign "<User: admin@example.com>": "SubscriptionAccount.user" must be a "User" instance.

Что я пробовал

Я попытался привести выбранного пользователя к реальному User экземпляру, но безрезультатно. Я сделал много отладок и вижу, что есть некоторые различия в модулях, видя, что в какой-то момент модуль пользователя меняется на __fake__. Однако это только внутри Django, так что я не уверен, о чем это. Я проверил все типы в миграции, и все, кажется, в порядке, за исключением модуля __fake__.

Я также попытался обновить до последних версий django и DRF, но это не сработало.

Код 

def migrate_email_to_user(apps, schema_editor):
    SubscriptionAccount = apps.get_model('hypernodedb', 'SubscriptionAccount')
    User = get_user_model()

    for subscription_account in SubscriptionAccount.objects.all():
        user, _ = User.objects.get_or_create(username=subscription_account.email)
        subscription_account.user = user
        subscription_account.save()


def revert_migrate_email_to_user(apps, schema_editor):
    SubscriptionAccount = apps.get_model('hypernodedb', 'SubscriptionAccount')

    for subscription_account in SubscriptionAccount.objects.all():
        subscription_account.email = subscription_account.user.username
        subscription_account.save()


class Migration(migrations.Migration):

    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ('hypernodedb', '0068_change_app_name_validation_error'),
    ]

    operations = [
        migrations.AddField(
            model_name='subscriptionaccount',
            name='user',
            field=models.OneToOneField(null=True, default=None, on_delete=django.db.models.deletion.CASCADE,
                                       to=settings.AUTH_USER_MODEL),
            preserve_default=False,
        ),
        migrations.RunPython(
            code=migrate_email_to_user,
            reverse_code=revert_migrate_email_to_user,
        ),
        migrations.RemoveField(
            model_name='subscriptionaccount',
            name='email',
        ),
    ]

Ниже приведено SubscriptionAccount в том виде, в каком оно существует при миграции: 

class SubscriptionAccount(TimeStampedModel):
    user = models.OneToOneField(get_user_model(), null=True)
    external_id = models.CharField(max_length=255)

    def __str__(self):
        return '{} - {}'.format(self.user.username, self.external_id)

Это была модель до изменений: 

class SubscriptionAccount(TimeStampedModel):
    # TODO: link this to a user model in our auth system
    email = models.CharField(max_length=255, unique=True)
    external_id = models.CharField(max_length=255)

    def __str__(self):
        return '{} - {}'.format(self.user.username, self.external_id)

1 Ответ

1 голос
/ 03 октября 2019

Оказывается, что вызов get_user_model() неверен для использования при миграции. Что, если подумать, логично. Поскольку вы никогда не должны импортировать модели непосредственно в процессе миграции. Однако ошибка миграции оттолкнула меня, и я пожаловался, что ожидал User, а не User. Также тот факт, что использование функции get_user_model() немного отличается от ее непосредственного импорта (то есть, что если вы используете другую модель User).

Способ решить эту проблему - импортировать модель User из apps: 

User = apps.get_model('auth', 'User')

После этого миграция прошла нормально.

...