Question

我正在将我的应用程序从Django 1.4迁移到Django 1.7。我已经升级了所有第三方库和Django本身。运行manage.py check或使用manage.py runserver时没有错误和警告。现在我想转向自定义用户模型。我创建了模型，它是来自django.contrib.auth.models.User的用户模型的直接副本，但有一点不同 - 我创建了自己的PermissionMixin基础，因为默认情况下，反向关系在默认用户模型和我的自定义用户之间发生冲突模型。

class PermissionsMixin(models.Model):
    is_superuser = models.BooleanField('superuser status', default=False)
    groups = models.ManyToManyField(
        Group, verbose_name='custom_groups', blank=True,
        related_name="custom_user_set", related_query_name="user")
    user_permissions = models.ManyToManyField(
        Permission, verbose_name='user permissions', blank=True,
        related_name="custom_user_set", related_query_name="user")

    class Meta:
        abstract = True

    def get_group_permissions(self, obj=None):
        permissions = set()
        for backend in auth.get_backends():
            if hasattr(backend, "get_group_permissions"):
                permissions.update(backend.get_group_permissions(self, obj))
        return permissions

    def get_all_permissions(self, obj=None):
        return _user_get_all_permissions(self, obj)

    def has_perm(self, perm, obj=None):

        # Active superusers have all permissions.
        if self.is_active and self.is_superuser:
            return True

        # Otherwise we need to check the backends.
        return _user_has_perm(self, perm, obj)

    def has_perms(self, perm_list, obj=None):
        for perm in perm_list:
            if not self.has_perm(perm, obj):
                return False
        return True

    def has_module_perms(self, app_label):
        # Active superusers have all permissions.
        if self.is_active and self.is_superuser:
            return True

        return _user_has_module_perms(self, app_label)


class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField('email address', max_length=254, unique=True)
    first_name = models.CharField('first name', max_length=30, blank=True)
    last_name = models.CharField('last name', max_length=30, blank=True)
    is_staff = models.BooleanField('staff status', default=False)
    is_active = models.BooleanField('active', default=True)
    date_joined = models.DateTimeField('date joined', default=timezone.now)

    USERNAME_FIELD = 'email'

在此之前一切正常 - 迁移会在数据库中创建所需的表，我可以在admin中修改自定义用户模型。现在，我想通过数据迁移将auth.User的所有数据复制到users.CustomUser模型。以下代码反映了我想要做的事情：

from django.db import migrations, transaction


@transaction.atomic
def copy_old_users(apps, schema_editor):
    User = apps.get_model("auth", "User")
    CustomUser = apps.get_model("users", "CustomUser")

    fields = ['id', 'username', 'email', 'first_name', 'last_name',
              'is_staff', 'is_active', 'date_joined', 'is_superuser',
              'last_login']

    for user in User.objects.all():
        custom_user = CustomUser()
        for field in fields:
            setattr(custom_user, field, getattr(user, field))

        custom_user.save()

        custom_user.groups.add(*user.groups.all())
        custom_user.user_permissions.add(*user.user_permissions.all())

@transaction.atomic
def remove_new_users(apps, schema_editor):
    CustomUser = apps.get_model("users", "CustomUser")
    CustomUser.objects.all().delete()


class Migration(migrations.Migration):

    dependencies = [
        ('auth', '0003_auto_20150128_1715'),
        ('users', '0001_initial'),
    ]

    operations = [
        migrations.RunPython(copy_old_users, remove_new_users),
    ]

我正在使用MySQL。这里的问题是user.user_permissions.all()，user.groups.all()始终为空。我在django shell中尝试过相同的 - 空。在管理员中，groups未正确附加且user_permissions为。{1}}。我已经检查了数据库 - 这不应该是空的。进一步调查表明，以下代码检索所需的对象：
User._meta.get_field('groups').rel.through.objects.filter(user_id=user.id)

与user_permissions相同。我可以从上面的查询中遍历对象列表并访问Group对象以在我的迁移中使用它。这一切都很顺利。

问题是：为什么标准方式不起作用？我是否以某种糟糕的方式访问对象？我是否遗漏了数据迁移方面的内容？我应该导入（?!）某些特定依赖项还是以某种特殊方式检索Permission和Group模型？

Answer 1

我没有足够的声誉来评论，但如果是一次性迁移，我建议您对auth.group和auth.permission型号进行json转储：

./manage.py dumpdata --indent=2 auth.group auth.permission > groups_and_perms.json

然后您可以使用以下命令再次加载它们：

./manage.py loaddata groups_and_perms.json

此外，群组和权限将是ManyToMany关系，它们不是吗？如果是这种情况，请尝试在.save_m2m()函数中使用copy_old_users()：

custom_user.save()

custom_user.groups.add(*user.groups.all())
custom_user.user_permissions.add(*user.user_permissions.all())

custom_user.save_m2m()

从auth.User迁移到自定义用户模型。组和权限始终为空

1 个答案: