4

我需要将现有项目更新到 Django 1.5 以利用其新可用的自定义用户模型。但是，我无法将包含具有外键的模型的可重用应用程序迁移给用户。目前，外键指向auth.User但使用自定义用户模型，它需要指向myapp.CustomUser. 因此，需要某种迁移。我不能简单地为它创建一个迁移文件，因为它是一个可重用的应用程序。这不会是未来的证明，因为每次更新应用程序时，我都需要记住再次创建该迁移（甚至可能存在迁移冲突），因此这不是一个合理的解决方案。

除了可能分叉每个项目，添加一个迁移文件，然后改用它之外，是否有解决这个问题的方法？

一些代码：

models.py在可重复使用的应用程序中

from django.conf import settings
from django.db import models

UserModel = getattr(settings, 'AUTH_USER_MODEL', 'auth.User')

class ModelA(models.Model):
    user = models.ForeignKey(UserModel)

models.py在我的项目中

from django.conf import settings
from django.contrib.auth.models import AbstractUser

class CustomUser(AbstractUser):
    ...

settings.py在我的项目中

AUTH_USER_MODEL = 'myapp.CustomUser'
4

1 回答 1

3

因此，如果可重用应用程序具有为用户创建外键的迁移，则可以执行以下操作来支持 Django 1.5 的自定义用户模型。

try:
    from django.contrib.auth import get_user_model
except ImportError: # django < 1.5
    from django.contrib.auth.models import User
else:
    User = get_user_model()

class Migration(SchemaMigration):

    def forwards(self, orm):
        db.create_table('reusableapp.modela', (
            ('user', self.gf('django...ForeignKey')(to=orm["%s.%s" % (User._meta.app_label, User._meta.object_name)])

    models = {
        ...
        # this should replace "auth.user"
        "%s.%s" % (User._meta.app_label, User._meta.module_name): {
        'Meta': {'object_name': User.__name__},
        }
        "reusableapp.modela": {
             'user': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['%s.%s']"% (User._meta.app_label, User._meta.object_name)})
        }
    }

我不确定这是否是最好的解决方案，但它正在 django-reversion 等应用程序中使用。

auth.User但是，如果您最初从 开始然后更改为，则此解决方案仍然可能会带来问题，这myapp.customuser仅仅是因为 south 是荣誉 AUTH_USER_MODEL 但尚未创建自定义用户模型的迁移。这可能在测试期间发生。南票 #1179 解决了这个问题 ( http://south.aeracode.org/ticket/1179 )。

于 2013-03-20T01:14:26.450 回答