一尘不染

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

django

在我的应用程序中,我想在新用户注册时在某些表中创建条目。例如,我要创建一个用户个人资料,然后将参考他们的公司和一些其他记录。我用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")

运行时效果很好。我可以使用admin创建一个新用户,其他三个表也可以获取有意义的条目。(除非是雇员,因为保存时未在管理员表单中填写user.first_name和user.last_name。我仍然不明白为什么要这样做)

问题出在我运行测试套件时。在此之前,我创建了一堆夹具以在表中创建这些条目。现在我得到一个错误,指出:

IntegrityError: duplicate key value violates unique constraint "core_userprofile_user_id_key"

我认为这是因为我已经在设备中创建了ID为“ 1”的公司,员工和档案记录,现在post_save信号正在尝试重新创建它。

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

人们如何做到这一点?


阅读 250

收藏
2020-03-29

共1个答案

一尘不染

我想我想出了一种方法来做到这一点。与信号一起传递的kwarg中有一个“ raw”参数,因此我可以用以下参数替换上面的测试:

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

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

2020-03-29