Question

我有一个更新某些权限的数据迁移。我知道迁移中的权限存在一些已知问题，我可以通过自己的迁移创建权限（而不是使用模型中的元组快捷方式）来避免一些麻烦。

迁移：

from __future__ import unicode_literals
from django.db import migrations, models
from django.conf import settings

def create_feature_groups(apps, schema_editor):
    app = models.get_app('myauth')

    Group = apps.get_model("auth", "Group")
    pro = Group.objects.create(name='pro')

    Permission = apps.get_model("auth", "Permission")
    ContentType = apps.get_model("contenttypes", "ContentType")
    invitation_contenttype = ContentType.objects.get(name='Invitation')

    send_invitation = Permission.objects.create(
         codename='send_invitation',
         name='Can send Invitation',
         content_type=invitation_contenttype)

    pro.permissions.add(receive_invitation)    

class Migration(migrations.Migration):

    dependencies = [
        ('myauth', '0002_initial_data'),
    ]

    operations = [
            migrations.RunPython(create_feature_groups),
    ]

经过一些试验和错误，我能够使用manage.py migrate完成这项工作，但我在测试manage.py test中收到错误。

__fake__.DoesNotExist: ContentType matching query does not exist.

调试了一下，发现在测试中运行时迁移中此时没有ContentType（不确定原因）。按照此post中的建议，我尝试在迁移过程中手动更新内容类型。补充：

from django.contrib.contenttypes.management import update_contenttypes
update_contenttypes(app, models.get_models())

在获取Invitation模型的内容类型之前。出现以下错误

  File "C:\Python27\lib\site-packages\django-1.7-py2.7.egg\django\contrib\contenttypes\management.py", line 14, in update_contenttypes
    if not app_config.models_module:
AttributeError: 'module' object has no attribute 'models_module'

必须有某种方式以可测试的方式在数据迁移中创建/更新权限。

感谢。

修改

最后通过添加

使其工作

from django.contrib.contenttypes.management import update_all_contenttypes
update_all_contenttypes()

奇怪的是，这个还不够

update_contenttypes(apps.app_configs['contenttypes'])

我很想知道为什么所有这些都是必要的

Answer 1

在编写跨多个应用程序的数据迁移时遇到类似问题。事实证明，Django只会将这些模型加载到受迁移状态的“依赖关系”成员影响的应用程序注册表中：https://code.djangoproject.com/ticket/24303

基本上要为我使用的迁移依赖项添加一个条目，该条目与例如对当前正在迁移的应用程序的ForeignKey。

Answer 2

从那以后，我最终花了3-4个小时才加入我的解决方案。

当我同时运行多个迁移时，问题是ContentType和Permission对象未被创建。由于我在下次迁移时引用了这些内容类型和迁移，因此导致了问题。）

但是，如果我使用迁移号码逐个运行它们，它们的工作正常。（在未来的迁移中引用）

为了解决这个问题，我在其间添加了一个额外的迁移来创建ContentType和Permission对象。

# -*- coding: utf-8 -*-
# Generated by Django 1.10.6 on 2017-03-11 05:59
from __future__ import unicode_literals

from django.conf import settings
from django.db import migrations


def update_all_contenttypes(**kwargs):
    from django.apps import apps
    from django.contrib.contenttypes.management import update_contenttypes

    for app_config in apps.get_app_configs():
        update_contenttypes(app_config, **kwargs)


def create_all_permissions(**kwargs):
    from django.contrib.auth.management import create_permissions
    from django.apps import apps

    for app_config in apps.get_app_configs():
        create_permissions(app_config, **kwargs)


def forward(apps, schema_editor):
    update_all_contenttypes()
    create_all_permissions()


def backward(apps, schema_editor):
    pass


class Migration(migrations.Migration):
    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ('contenttypes', '0002_remove_content_type_name'),
        ('MY_APP', '0123_LAST_MIGRATION'),
    ]

    operations = [
        migrations.RunPython(forward, backward)
    ]

Answer 3

答案是：

apps.get_model（'contenttypes'，'ContentType'）:)今天我自己需要它。

Answer 4

update_contenttypes(apps.app_configs['contenttypes'])

将更新contenttypes应用的内容类型。

我相信你会想要这样做......

update_contenttypes(apps.app_configs['app_label'])

其中app_label是邀请模型所在的应用的应用标签。这将更新您的单个应用的内容类型，以便根据您的原始代码进行查询。

Answer 5

对于 Django 2.1 ，我必须从全局注册表中导入应用程序，因为传递到迁移中的应用程序是django.db.migrations.state.AppConfigStub的实例，没有填充的models_module属性。 create_contenttypes正在检查此属性。

from django.apps.registry import Apps, apps as global_apps
from django.contrib.contenttypes.management import create_contenttypes
from django.db import migrations


def add_permision(apps: Apps, schema_editor):
    gls_app_config = global_apps.get_app_config('my_app')
    create_contenttypes(gls_app_config)

    ...

Answer 6

from django.db import migrations
from django.db.migrations import RunPython
from django.apps.registry import Apps, apps as global_apps
from django.contrib.contenttypes.management import create_contenttypes


def add_content_type_records(apps: Apps, schema_editor):
    my_app_config = global_apps.get_app_config('my_1_app')
    my_app_config.models_module = True
    create_contenttypes(my_app_config)
    my_app_config.models_module = None
    my_app_config = global_apps.get_app_config('my_2_app')
    my_app_config.models_module = True
    create_contenttypes(my_app_config)
    my_app_config.models_module = None


def create_setup_data(apps, schema_editor):
    ...


def delete_setup_data(apps, schema_editor):
    ...


class Migration(migrations.Migration):

    dependencies = [
        ('my_1_app', '....'),
        ('my_2_app', '....'),
        ('contenttypes', '__latest__'),
    ]

    operations = [
        RunPython(add_content_type_records, RunPython.noop),
        RunPython(create_setup_data, delete_setup_data),
    ]

在迁移中获取模型ContentType - Django 1.7

6 个答案: