如何防止灯具与 django post_save 信号代码冲突？

Question

在我的应用程序中，我想在新用户注册时在某些表中创建条目。例如，我想创建一个用户配置文件，然后该配置文件将引用他们的公司和他们的一些其他记录。我用 post_save 信号实现了这一点：

def callback_create_profile(sender, **kwargs):
    # check if we are creating a new User
    if kwargs.get('created', True):
        user = kwargs.get('instance')
        company = Company.objects.create(name="My Company")
        employee = Employee.objects.create(company=company, name_first=user.first_name, name_last=user.last_name)
        profile = UserProfile.objects.create(user=user, employee=employee, partner=partner)
# Register the callback
post_save.connect(callback_create_profile, sender=User, dispatch_uid="core.models")

这在运行时效果很好。我可以使用管理员创建一个新用户，其他三个表也可以获取合理的条目。 （除此之外，由于 user.first_name 和 user.last_name 在保存时没有在管理员表单中填写，因此我仍然不明白为什么这样做）

问题出现在我运行测试时套房。在此之前，我创建了一堆固定装置来在表中创建这些条目。现在我收到一个错误，指出：

IntegrityError: duplicate key value violates unique constraint "core_userprofile_user_id_key"

我认为这是因为我已经在 ID 为“1”的装置中创建了一家公司、员工和个人资料记录，现在 post_save 信号正在尝试重新创建它。

我的问题是：运行灯具时可以禁用此 post_save 信号吗？我可以检测到我正在作为测试套件的一部分运行而不创建这些记录吗？我现在应该从灯具中删除这些记录吗（尽管信号仅设置默认值而不是我想要测试的值）？为什么夹具加载代码不覆盖创建的记录？

人们如何做到这一点？

Answer 1

我想我找到了一种方法来做到这一点。与信号一起传入的 kwargs 中有一个“raw”参数，因此我可以用以下参数替换上面的测试：

if (kwargs.get('created', True) and not kwargs.get('raw', False)):

在 loaddata 运行时使用 Raw。这似乎可以解决问题。

这里提到： http://code.djangoproject.com/ticket/13299

将是如果有记录就好了： http:// /docs.djangoproject.com/en/stable/ref/signals/#django.db.models.signals.post_save

Answer 2

这是一个老问题，但我发现最简单的解决方案是使用由加载数据传递的“原始”参数，并装饰侦听器函数，例如：

from functools import wraps


def disable_for_loaddata(signal_handler):
    @wraps(signal_handler)
    def wrapper(*args, **kwargs):
        if kwargs['raw']:
            print "Skipping signal for %s %s" % (args, kwargs)
            return
        signal_handler(*args, **kwargs)
    return wrapper

然后

@disable_for_loaddata
def callback_create_profile(sender, **kwargs):
    # check if we are creating a new User
    ...

Answer 3

简单的解决方案，将其添加到 post_save 函数的开头：

if kwargs.get('raw', False):
    return False

这将导致该函数在加载夹具时退出。

请参阅： https://docs.djangoproject.com/en/dev/参考/信号/#post-save

Answer 4

我在我的一个项目中遇到了类似的问题。就我而言，信号也减慢了测试速度。我最终放弃了信号，转而重写 Model.save() 方法。

但就您而言，我认为通过重写任何 save() 方法来实现此目的没有意义。在这种情况下，您可能想尝试一下。警告，我只尝试过一次。它似乎有效，但尚未经过彻底测试。

1. 创建您自己的测试运行程序。
2. 在加载灯具之前，断开连接 来自 User 类的 post_save 信号的 callback_create_profile 函数。
3. 让夹具加载。
4. 将函数连接回信号。