4

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

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'

在那之前一切都很好——迁移在数据库中创建了所需的表，我可以在管理员中修改自定义用户模型。现在我想通过数据迁移将所有数据复制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是。我已签入数据库 - 这不应为空。进一步调查表明，以下代码检索所需的对象：
User._meta.get_field('groups').rel.through.objects.filter(user_id=user.id)

和它一样user_permissions。我可以遍历上述查询中的对象列表并访问该Group对象以在我的迁移中使用它。这一切都很好并且有效。

问题是：为什么标准方式不起作用？我是否以某种不好的方式访问对象？我是否错过了数据迁移中的某些内容？我应该导入（？！）一些特定的依赖关系还是以某种特殊的方式检索Permission和Group模型？

4

1 回答 1

0

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

./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()
于 2015-02-26T13:54:00.747 回答