Question

我正在Django的图书馆管理系统工作。

我在应用Book中有一个模型App1，在Student中有一个模型Employee和App2。学生可以发行最多3本书，但员工可以发行任意数量的书籍。

怎么能在django做到这一点？我想我应该在书模型中使用ForeignKey，类似这样：

class Book(models.Model):
    ...
    issued_to = models.ForeignKey(Student) # <-- Student or Employee
                                           #     how do I do that?

但是如何确保最多3个Book实例与单个Student实例相关。

Answer 1

在我看来，你应该研究django模型的clean()方法。当您尝试保存实例时调用它。对于您的情况，代码可能看起来像这样（未经测试）：

from django.core.exceptions import ValidationError


class Person(models.Model):
    max_books = 1

    def clean(self):
        books_count = self.books.all().count()
        if books_count >= max_books:
            raise ValidationError("This person has too much books !")


 class Book(models.Model):
    issued_to = models.ForeignKey(Person, related_name="books")

这样，您可以创建Person模型的子类，并设置自己的max_books限制：

class Student(Person)
    max_books = 3

class Employee(Person)
    max_books = 30

但是，要小心，因为通过使用此解决方案，您将依赖于django模型的具体继承，这可能会导致performance issues。

Answer 2

另一种解决方案可能是使用django信号。

Django会在保存任何模型之前发送pre_save信号，因此您可以挂钩函数以对其做出反应并执行检查。

（构建@Eliot Berriot解决方案）：

from django.db.models.signals import pre_save
from django.core.exceptions import ValidationError

def validate_num_books(sender, **kwargs):
    if isinstance(sender, Student):
        max_books = 3
    elif isinstance(sender, Employee):
        max_books = 30

    books_count = sender.books.all().count()
    if books_count >= max_books:
        raise ValidationError("This person has too much books !")

pre_save.connect(validate_num_books, sender=Employee, dispatch_uid='validate_num_books')
pre_save.connect(validate_num_books, sender=Student, dispatch_uid='validate_num_books')

注意：

我不确定ValidationError会产生预期的效果。希望它足以绕过被保存的模型，但它可能不会那样工作，或者它在某些情况下有效但在其他情况下却没有......请将此片段视为实验性伪代码。
正如您可能从上面的观点中猜到的那样，我不习惯发出信号，而且仍然有点厌倦了。这可能是完全正常的，但如果出现更清洁的解决方案，我可能会采用它并完全避免信号。更简单通常更好。
我可以看到这个解决方案超过艾略特答案的唯一优势是它避免了继承，正如他所说，这会导致性能问题。但是，如果您避免在父类上定义字段，我认为您应该没问题。

如果您决定尝试使用信号start here。

Django - 将模型X与模型Y的最大n个实例相关联

2 个答案: