使用Django 1.7创建部分索引

Question

Django 1.7提及类的文档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支持在支持它们的数据库(PostgreSQL和SQLite)上的声明性部分唯一索引.所以你可以这样做:

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

class Post(Model):
    ...
    class Meta:
        constraints = [
            UniqueConstraint(fields=["title", "blog", "category"], condition=Q(category__isnull=False)),
            UniqueConstraint(fields=["title", "blog"], 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.