如何在 Django 中对 post_migrate 信号执行代码？

Question

我正在为我的项目做某种重构，我依赖于 django django.contrib.auth.models.Permission 模型。到目前为止，我使用 post_save 信号为每个新用户定义权限，因此在创建用户时，我使用 user.user_permissions.add(the_permission) 分配他们的权限，这非常有效。

现在我想使用django.contrib.auth.models.Group模型来分类用户应该拥有的权限。

这是我的代码：

from django.apps import AppConfig
from django.db.models.signals import post_migrate
from django.contrib.auth.models import Group, Permission


def create_group(name, permissions):
    group = Group.objects.create(name=name)
    [group.permissions.add(permission) for permission in permissions]


def define_company_groups(sender, **kwargs):
    permissions = [
        Permission.objects.get(codename='add_mymodel'),
        Permission.objects.get(codename='change_mymodel'),
    ]
    create_group('managers', permissions)


class MyAppConfig(AppConfig):
    name = 'players'
    verbose_name = 'The players app'

    def ready(self):
        post_migrate.connect(define_company_groups, sender=self)

定义此代码后，我希望在调用 ./manage.py migrate 后应触发此处理程序。但它并没有发生，我得到的只是：

Running post-migrate handlers for application players
Adding permission 'players | mymodel | Can add mymodel'
Adding permission 'companies | company | Can change mymodel'
Adding permission 'companies | company | Can delete company'

我找到这篇 https://groups.google.com/forum/#!topic/django-developers/8MdaWtJp4VQ 文章，他们说我应该在名为 management.py 的文件中定义我的 post_migrate 处理程序，但它对我不起作用。

最后，这是我的问题：我应该将自定义 post_migrate 信号的代码放在哪里？

Answer 1

Django docs 建议在应用配置的就绪方法中连接 post_migrate 信号。 Google 组 post 你 link 到的已经过时了，在文档更新之前。

您还需要在 INSTALLED_APPS 设置中 specify the app config。

INSTALLED_APPS = [
    'myapp.apps.MyAppConfig',
    # ...
]

另一种配置应用的方法是在应用的 __init__.py 中使用 default_app_config。参见 Configuring Applications。但另一种方式（AppConfig 的虚线路径）是首选。

Answer 2

信号post_migrate不同于其他信号。 './manage.py' 命令不会执行 apps.py 文件或 signals.py 文件中的代码要执行此信号，请将其放在 models.py 文件中。然后你会得到想要的结果

Answer 3

我之前为另一个问题做过post_migrate例子。我会写下它的解决方案。也许对你有帮助。

# in apps.py

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

def create_default_site_profile(sender, **kwargs):
    """after migrations"""
    from django.contrib.sites.models import Site
    from core.models import SiteProfile

    site = Site.objects.get(id=getattr(settings, 'SITE_ID', 1))

    if not SiteProfile.objects.exists():
        SiteProfile.objects.create(site=site)

class CoreConfig(AppConfig):
    name = 'core'

    def ready(self):
        post_migrate.connect(create_default_site_profile, sender=self)
        
        # if you have other signals e.g. post_save, you can include it 
        # like the one below.
        from .signals import (create_site_profile)

如何在 Django 中对 post_migrate 信号执行代码？

How to execute code on post_migrate signal in Django?

django

django-signals

django-migrations