我正在将我的应用程序从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
模型?
答案 0 :(得分:0)
我没有足够的声誉来评论,但如果是一次性迁移,我建议您对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()