Django一分钟：保护被引用的数据不被删除PROTECT和RESTRICT的区别

在Django中一个由Foreignkey引用的对象被删除时，Django将会按照on_delete参数所指定的SQL约束进行处理。

on_delete的可用配置如下：

• CASADE
• PROTECT
• RESTRIC
• SET_NULL
• SET_DEFAULT
• SET()

本文要介绍的是PROTECT和RESTRIC，两者都用于防止删除被引用对象。

场景如下：库存管理系统中你有一个名为Unit的表，另一张名为Material的通过外键引用了Unit，你希望保护被引用的unit数据不被删除，只有完全没有被任何数据引用的unit才能被删除。

这样的需求非常常见，因为我们常常需要保护数据的完整性。在Djangoz中有两种方式可以实现此需求：即配置Foreignkey的on_delete属性为RESTRIC或PROTECT。虽然两者配置都能实现此目的，但是它们还是有很大的不同的，接下来我们通过一些示例来介绍其中的区别。

RESTRIC

创建下面的模型：

class Artist(models.Model):
    name = models.CharField(max_length=10)


class Album(models.Model):
    artist = models.ForeignKey(Artist, on_delete=models.CASCADE)


class Song(models.Model):
    artist = models.ForeignKey(Artist, on_delete=models.CASCADE)
    album = models.ForeignKey(Album, on_delete=models.RESTRIC)

创建数据

>>> artist1 = Artist.objects.create(name="artist one")
>>> artist2 = Artist.objects.create(name="artist two")
>>> album1 = Album.objects.create(artist=artist1 )
>>> album2 = Album.objects.create(artist=artist2 )
>>> song1 = Song.objects.create(artist=artist1, album=album1 )
>>> song2 = Song.objects.create(artist=artist2, album=album2 )

这些数据的关系如下图所示:

接着我们进行测试：

>>> album1.delete()
# Raises RestrictedError.
>>> artist2.delete()
# Raises RestrictedError.
>>> artist1.delete()
(4, {'Song': 2, 'Album': 1, 'Artist': 1})

1. album1不能被删除，因为album1<---RESTRIC---song1
2. artist2不能被删除，因为album1<---CASCADE---album2---RESTRIC---song2;
3. album1可以被删除，因为虽然album1<---CASCADE---album1---RESTRIC---song1但是额外还有album1<---CASCADE---song1这样层关系在，因此可以删除，相关的数据也都会被级联删除。

PROTECT

修改代码:

# ...
class Song(models.Model):
    # ...      
    # 将RESTRIC改为PROTECT
    album = models.ForeignKey(Album, on_delete=models.PROTECT)

我们再以同样的方式创建数据：

>>> artist1 = Artist.objects.create(name="artist one")
>>> artist2 = Artist.objects.create(name="artist two")
>>> album1 = Album.objects.create(artist=artist1 )
>>> album2 = Album.objects.create(artist=artist2 )
>>> song1 = Song.objects.create(artist=artist1, album=album1 )
>>> song2 = Song.objects.create(artist=artist2, album=album2 )

此时数据的关系如下：

我们再进行测试发现无论如何都无法删除artist1了：

>>> album1.delete()
# Raises ProtectedError.
>>> artist2.delete()
# Raises ProtectedError.
>>> artist1.delete()
# Raises ProtectedError.

显然PROTECT更加严格，即使你存在一层额外的CASCADE关系在，但也无法删除被引用的数据。