Question

我创建了一个自定义用户模型。 auth数据库中已有用户。因此，我使用数据迁移将数据迁移到我的自定义用户模型。

这是我在自动迁移文件（我从here中找到）中进行数据迁移的方式：

已更新

from __future__ import unicode_literals

from django.db import models, migrations


class Migration(migrations.Migration):

    dependencies = [
        ('userauth', '0002_auto_20150721_0605'),
    ]

    operations = [
        migrations.RunSQL('INSERT INTO userauth_userauth SELECT * FROM auth_user'),
        migrations.RunSQL('INSERT INTO userauth_userauth_groups SELECT * FROM auth_user_groups'),
        migrations.RunSQL('INSERT INTO userauth_userauth_user_permissions SELECT * FROM auth_user_user_permissions'),
    ]

models.py

class UserManager(BaseUserManager):
        def _create_user(self, username, email, password, is_staff, is_superuser, **extra_fields):
            now = timezone.now()
            if not username:
              raise ValueError(_('The given username must be set'))
            email = self.normalize_email(email)
            user = self.model(username=username, email=email,
                     is_staff=is_staff, is_active=False,
                     is_superuser=is_superuser, last_login=now,
                     date_joined=now, **extra_fields)
            user.set_password(password)
            user.save(using=self._db)
            if not is_staff:
                group = Group.objects.get(name='normal')
                user.groups.add(group)
            return user

        def create_user(self, username, email=None, password=None, **extra_fields):
            return self._create_user(username, email, password, False, False,
                     **extra_fields)

        def create_superuser(self, username, email, password, **extra_fields):
            user=self._create_user(username, email, password, True, True,
                         **extra_fields)
            user.is_active=True
            user.save(using=self._db)
            return user


    class UserAuth(AbstractBaseUser, PermissionsMixin):
        #original fields
        username = models.CharField(_('username'), max_length=30, unique=True,
        help_text=_('Required. 30 characters or fewer. Letters, numbers and @/./+/-/_ characters'),
        validators=[
          validators.RegexValidator(re.compile('^[\w.@+-]+$'), _('Enter a valid username.'), _('invalid'))
        ])
        first_name = models.CharField(_('first name'), max_length=30, blank=True, null=True)
        last_name = models.CharField(_('last name'), max_length=30, blank=True, null=True)
        email = models.EmailField(_('email address'), max_length=255, unique=True)
        is_staff = models.BooleanField(_('staff status'), default=False,
        help_text=_('Designates whether the user can log into this admin site.'))
        is_active = models.BooleanField(_('active'), default=False,
        help_text=_('Designates whether this user should be treated as active. Unselect this instead of deleting accounts.'))
        date_joined = models.DateTimeField(_('date joined'), default=timezone.now)
      #additional fields
        full_name =  models.CharField(_('Full Name'), max_length=600, blank=True, null=True)
        profileImage = models.ImageField(upload_to="upload",blank=True,null=True,
        help_text = _("Please upload your picture"))
        BioUser =  models.TextField(blank=True,null=True)
        Social_link = models.URLField(blank=True,null=True)

        objects = UserManager()
        USERNAME_FIELD = 'username'
        REQUIRED_FIELDS = []

        class Meta:
            verbose_name = _('user')
            verbose_name_plural = _('users')

        def get_full_name(self):
            return self.full_name

        def get_short_name(self):
            return self.first_name

        def email_user(self, subject, message, from_email=None):
            send_mail(subject, message, from_email, [self.email])

迁移后的问题，我无法创建新用户。我收到这个错误：

重复键值违反了唯一约束＆＃34; userauth_userauth_pkey＆＃34; DETAIL：Key（id）=（3）已经存在。

似乎表格不同步。我该如何解决这个问题？

Answer 1

在Postgres上，Django处理为其数据库记录创建唯一主键的方式是使用database sequence从中获取新的主键。查看我自己的数据库，我看到如果我有一个名为 x 的表，那么Django会在名称 x _id_seq下创建序列。因此，userauth_userauth表格的顺序为userauth_userauth_id_seq。

现在，您进行迁移的方式是使用原始SQL语句。这完全绕过了Django的ORM，这意味着迁移不会触及新表的序列。执行此类原始迁移后应执行的操作是将主键序列设置为不会与数据库中已存在的数字冲突的数字。借用this answer，你应该发出：

select setval('userauth_userauth_id_seq', max(id)) 
       from userauth_userauth;

对其他表执行相同类型的操作：如果他们的id字段，则将他们自己的序列设置为最大值。（如果您想知道，将使用的下一个值将通过nextval获得，并且将等于调用nextval之前的序列值的一个。）

在评论中，您想知道为什么创建新用户最终有效。可能发生的事情是你试图创建新用户，它是这样的：

Django从适当的序列中获得了一个新的主键。这里序列增加了。
Django试图保存新用户，因为上一步中获得的数字并不是唯一的。

如果你这样做了足够多次，你的序列会在每次尝试时增加，因为无论事务，Postgres都不会回滚序列。 documentation说：

重要提示：由于序列是非事务性的，因此如果事务回滚，则setval所做的更改不会被撤消。

所以最终，序列增加超过表中已有的最大主键，从那时起它就可以了。

Answer 2

您使用的数据库是什么？听起来您的数据库有一个生成ID的主键的计数器。由于您使用主键创建了新行，您可能需要手动重置DB计数器。有关postgres的示例，请参阅How to reset postgres' primary key sequence when it falls out of sync?

在django 1.8中将数据从原始用户模型迁移到自定义用户模型

2 个答案: