使用 Django 1.7 创建部分索引

Question

Django 1.7 的 documentation 提到 RunSQL 类可用于在表上创建部分索引。我有一张桌子，我希望 title、blog 和 category 的组合是独一无二的。但是，如果没有提供类别，标题和博客的组合仍然应该是唯一的。

class Post(models.Model):
    title = models.CharField(max_length=200)
    blog = models.ForeignKey(Blog)
    category = models.ForeignKey(Category, null=True, blank=True)

我可以使用部分索引来实现这个约束（如下所示的 SQL）。如果我使用 Django 1.7 迁移，我应该在哪里添加此代码？

CREATE UNIQUE INDEX idx1 
  ON Post (title, blog_id, category_id) 
  WHERE category_id IS NOT NULL;

CREATE UNIQUE INDEX idx2 
  ON Post (title, blog_id)
  WHERE category_id IS NULL;

Answer 1

Django 2.2 及更高版本

从 2.2 版起，Django 在支持 declarative partial unique indexes 的数据库（PostgreSQL 和 SQLite）上支持它们。所以你可以这样做：

from django.db.models import Model, Q, UniqueConstraint

class Post(Model):
    ...
    class Meta:
        constraints = [
            UniqueConstraint(
                fields=["title", "blog", "category"],
                name="idx1",
                condition=Q(category__isnull=False)),
            UniqueConstraint(
                fields=["title", "blog"], 
                name="idx2",                    
                condition=Q(category__isnull=True)),
        ]

Django 2.1 及更早版本

在旧版本中，您需要通过迁移来执行此操作。首先创建一个新的空迁移文件：

python manage.py makemigrations --empty yourappname

然后，为每个索引添加一个适当的RunSQL 行：

operations = [
    migrations.RunSQL("CREATE UNIQUE INDEX..."),
    migrations.RunSQL("CREATE UNIQUE INDEX..."),
]

最后，运行migrate。

Answer 2

您可以像这样提供unique_together：

class Post(models.Model):
    title = models.CharField(max_length=200)
    blog = models.ForeignKey(Blog)
    category = models.ForeignKey(Category, null=True, blank=True)

class Meta:
    unique_together = ("title", "blog", "category")

类别的 NULL 将按照您的意愿工作，如果未设置，则标题/博客必须是唯一的。

https://docs.djangoproject.com/en/1.8/ref/models/options/#unique-together