python - django get_or_create 取决于自定义管理器方法的结果

Question

我正在尝试实现一种创建模型实例的安全方法 Bar 。我有一个自定义管理器方法来搜索任何附近现有的 Bar基于 x, y 的实例坐标对:Bar.objects.circle_search(x, y, radius)

这些x, y值通常来自 Foo 。创建 Foo 后然后，它应该通过外键与现有的 Bar 关联。实例如果 Foo.x, Foo.y在radius内的Bar.x, Bar.y 。如果没有Bar找到实例然后创建一个。每个Bar实例有 x, y Foo 定义的列创造了它。

我有一些简单的东西，例如:

bar = Bar.objects.circle_search(foo.x, foo.y, radius)
if bar is None:
    # fill some kwargs
    bar, created = Bar.objects.get_or_create(**kwargs)

但是这里有一个明显的竞争，其中两个附近 foo可以创建单独的Bar彼此接近的实例(彼此之间的 Bar 内不应有任何 radius 实例)。

是否有办法链接/插入 circle_search方法用作 get部分get_or_create ？即用简单的英语 - 获取 Bar与 circle_search 匹配，如果没有则创建一个。

# something like?
Bar.object.get_or_create(circle_search(foo.x, foo.y, radius), **kwargs)

问题是我不能依赖使用 bar.x, bar.y (从 foo.x, foo.y 填充)由于非常接近(在 x, y 空间) Foo ，因此是唯一的实例将被允许创建 Bar实例，但它们不应该，因为我需要收集所有 Foo在radius内与相同的 Bar 相关联.

(我的一部分想知道即使使用 get_or_create 这一切是否安全，因为即使我们最终在另一个 IntegrityError 内创建 Bar 也不会引发 radius - 在这种情况下是吗是否可以在某处编写一些自定义逻辑，这将导致在任何实例创建时抛出此错误？)

抱歉，如果这很难理解，我可以尝试在评论中进行澄清。

<小时/>

为了完整性，最小、简单的示例模型:

class BarQuerySet(models.QuerySet):
    def circle_search(self, x, y, radius):
        # custom SQL to filter on a distance
        return qs

class Bar(models.Model):
    x = models.FloatField()
    y = models.FloatField()
    objects = BarQuerySet.as_manager()

class Foo(models.Model):
    x = models.FloatField()
    y = models.FloatField()
    bar = models.ForeignKey(Bar)

Answer 1

您的查询不会引发 IntegrityError，因此您无法使用依赖于此的 get_or_create()。你可以写一个Constraint这样，当约束中存在另一行时，就会引发 IntegrityError 。但是您的查找可能太复杂，无法传递给 get_or_create()，在这种情况下，您应该查看 Django 代码库中的 get_or_create() 并使用您自己的查找自行复制它。