如何编写迁移以使用 ManyToManyField 更改模型的主键

Question

我有一个 UserProfile 模型，它使用 OneToOneField 引用我的 User 模型。我还使用post_save 信号在创建用户时自动创建UserProfile。除了通过管理员（我使用内联）创建用户时，当我收到有关重复配置文件的错误时，这非常有用。 This answer recommends setting the primary key to be the OneToOneField referring to user.

那么之前：

class UserProfile(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    # ...
    subjects = models.ManyToManyField(Subject, null=True, blank=True)

之后

class UserProfile(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL, primary_key=True)
    # ...
    subjects = models.ManyToManyField(Subject, null=True, blank=True)

我正在尝试使用 Django 1.7 中的迁移来做到这一点，但由于配置文件有多个 ManyToManyField，因此生活变得复杂 - 所以它们都引用了 UserProfile 的 id 字段模型。使用 makemigrations 创建迁移以使用户成为主键，并删除旧的 id 字段，但它忽略了 ManyToManyField。

我目前正陷入困境，在迁移中使用大量 RunSQL 语句来修改 ManyToManyField 的直通表。我刚刚遇到另一个错误，其中一个表中的约束名称与另一个表中的不同。

所以我的问题是：在 Django 迁移中是否有一种方法可以完成更改直通表的工作，使其引用新的主键，更新所有约束、键等？如果没有，处理这种情况的最佳方法是什么？

我将 Django 1.7 与 MySQL 一起使用。

Answer 1

所以我最终使用 SQL 来修复它。我的解决方案的核心如下 - 基本上我

• 在新配置文件的 user_id 上创建索引
  • 此索引必须存在才能将其作为外键引用
• 创建一个新的直通表
  • 我从SHOW CREATE TABLE userprofile_userprofile_subjects（特定于MySQL）的输出开始
  • 我稍微修改了键名和约束名
• 将所有数据复制到新的直通表中
• 通过表删除旧的
• 将新的直通表重命名为旧直通表的名称
• 终于完成了 django 迁移自动为我生成的操作

我希望这对其他人有所帮助。而且我仍然有兴趣了解更好的解决方案。

from django.db import migrations

class Migration(migrations.Migration):

    dependencies = [
        # ...
    ]

    operations = [
        migrations.RunSQL(
            'ALTER TABLE userprofile_userprofile '
            'ADD INDEX `userprofile_userprofile_1234abcd` (user_id)'
        ),
        migrations.RunSQL (
            'CREATE TABLE userprofile_temp_table ('
            '`id` int(11) NOT NULL AUTO_INCREMENT, '
            '`userprofile_id` int(11) NOT NULL, '
            '`subject_id` int(11) NOT NULL, '
            'PRIMARY KEY (`id`), '
            'UNIQUE KEY `userprofile_userprofile_subjects_userprofile_us_7ded3060_uniq` (`userprofile_id`,`subject_id`), '
            'KEY `userprofile_userprofile_subject_1be9924f` (`userprofile_id`), '
            'KEY `userprofile_userprofile_subject_e5a9504a` (`subject_id`), '
            'CONSTRAINT `subject_id_refs_id_69796996` FOREIGN KEY (`subject_id`) REFERENCES `otherapp_subject` (`id`), '
            'CONSTRAINT `userprofile_user_id_refs_user_id_1234abcd` FOREIGN KEY (`userprofile_id`) REFERENCES `userprofile_userprofile` (`user_id`) '
            ') ENGINE=InnoDB AUTO_INCREMENT=35500 DEFAULT CHARSET=utf8 '
        ),
        migrations.RunSQL (
            'INSERT INTO userprofile_temp_table '
            '(userprofile_id, subject_id) '
            '('
            '  SELECT userprofile_userprofile.user_id, userprofile_userprofile_subjects.subject_id'
            '    FROM userprofile_userprofile_subjects'
            '    INNER JOIN userprofile_userprofile'
            '    ON userprofile_userprofile_subjects.userprofile_id ='
            '        userprofile_userprofile.id'
            ')'
        ),
        migrations.RunSQL (
            'DROP TABLE `userprofile_userprofile_subjects`'
        ),
        migrations.RunSQL (
            'RENAME TABLE `userprofile_temp_table` TO `userprofile_userprofile_subjects`'
        ),
        migrations.RemoveField(
            model_name='userprofile',
            name='id',
        ),
        migrations.AlterField(
            model_name='userprofile',
            name='user',
            field=models.OneToOneField(
                primary_key=True, serialize=False, to=settings.AUTH_USER_MODEL
            ),
            preserve_default=True,
        ),
    ]